llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/lib/Dialect/Torch/Utils
History
Nithin Meganathan 5ecc1d5c0d
Align softmax accumulation types with Torch's CUDA implementation (#2996) 		2024-03-12 15:07:45 -07:00
..
CMakeLists.txt mlir: add missing dependency on TableGen targets (#1537) 2022-11-01 14:59:11 -05:00
TorchUpstream.cpp [Torch Dialect] Scatter reduce lowering (#1884) 2023-02-21 23:05:55 +00:00
Utils.cpp Align softmax accumulation types with Torch's CUDA implementation (#2996) 2024-03-12 15:07:45 -07:00