mirror of https://github.com/llvm/torch-mlir
0a86deb59a
Set PyTorch and TorchVision version to nightly release 2024-08-18. This commit also updates the `scaled_dot_product_attention` op. A new attribute `enable_gqa` has been added. As of now, only the default value for the same is supported. Signed-Off By: Vivek Khandelwal <vivekkhandelwal1424@gmail.com> |
||
|---|---|---|
| .. | ||
| _torch_mlir_custom_op_example | ||
| csrc | ||
| jit_ir_importer | ||
| _dynamo_fx_importer.py | ||
| _version.py | ||
| dynamo.py | ||
| torchscript.py | ||