llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/projects/pt1/python
History
Vivek Khandelwal 0a86deb59a
build: manually update PyTorch version (#3627) 
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>
 		2024-08-19 12:03:56 +05:30
..
test [NFC reformat] Applies pre-commit formatting to Python files. (#3244) 2024-04-27 14:16:31 -07:00
torch_mlir build: manually update PyTorch version (#3627) 2024-08-19 12:03:56 +05:30
torch_mlir_e2e_test Implement lowering of aten.atleast_2d (#3546) 2024-08-14 18:52:31 +05:30
CMakeLists.txt [NFC reformat] Run pre-commit on all files and format misc. 2024-04-27 14:08:09 -07:00