llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/test
History
Yi Zhang a459e09ab7 E2e support for aten.softmax.int and aten.embedding 
- Added a DecomposeComplexOps pass to decompose complex torchOps.
- Refactored `visitAtenArgmaxOp` and `visitAtenAnyDimOp` to
`visitReductionAlongDimIntOp`.
- Moved some helper functions into
torch-mlir/Dialect/Torch/Utils/Utils.h to be shared by multiple files.
- Added support for f64 tensor as argument and return types.
 		2021-10-18 17:57:45 -04:00
..
Conversion Update llvm-project to a54f4eae0e1d0ef5adccdcf9f6c2b518dc1101aa 2021-10-18 13:36:42 -04:00
Dialect E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00
RefBackend add i64 support to refbackend 2021-10-05 15:12:44 -04:00
python Bring some more order to the e2e error reporting situation. 2021-10-08 13:00:12 -07:00
CMakeLists.txt Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
lit.cfg.py Add tool substitutions to support out-of-tree builds 2021-10-07 21:16:43 -07:00
lit.site.cfg.py.in Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00