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> |
||
|---|---|---|
| .. | ||
| TMTensor | ||
| Torch | ||
| TorchConversion | ||
| CMakeLists.txt | ||