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> |
||
---|---|---|
.. | ||
CMakeLists.txt | ||
TorchToTMTensor.cpp |