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/Transforms
History
Yi Zhang 05c4dd8e39 Add convertScalarToDtype helper. 
This is to facilitate scalar type conversion in the TorchToLinalg. As
part of adding the helper, this PR also:
- Updated `AtenAddTensorOp`, `AtenSubTensorOp` to use the helpers to
support more type variants.
- Added e2e type promotion testing.
- Added i32 memref return/arg type to support e2e testing.
 		2021-11-08 17:50:52 -05:00
..
AdjustCallingConventions.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00
DecomposeComplexOps.cpp [TORCH][MLIR] Add E2E support for aten.expand 2021-11-03 23:58:59 +05:30
GlobalizeObjectGraph.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
InlineGlobalSlots.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
MaximizeValueSemantics.cpp [TORCH][MLIR] Add E2E support for aten::to.dtype. 2021-11-08 12:56:03 -05:00
PassDetail.h Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
Passes.cpp E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00
PrepareForGlobalizeObjectGraph.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
ReduceOpVariants.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
RefinePublicReturn.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
RefineTypes.cpp Add convertScalarToDtype helper. 2021-11-08 17:50:52 -05:00