llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/e2e_testing/torchscript
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
..
argmax.py add argmax lowering 2021-10-13 14:31:16 -04:00
basic.py Add NumToTensor 2021-11-08 15:56:52 -05:00
batchnorm.py E2e support for layernorm. 2021-10-04 14:15:13 -04:00
conv.py Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
elementwise.py Add log2 2021-11-08 16:19:59 -05:00
main.py Add convertScalarToDtype helper. 2021-11-08 17:50:52 -05:00
matmul.py Add lowering of aten.matmul op. 2021-10-26 12:45:09 -04:00
mlp.py Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
quantized_models.py Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
reduction.py Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
type_conversion.py Add convertScalarToDtype helper. 2021-11-08 17:50:52 -05:00
type_promotion.py Add convertScalarToDtype helper. 2021-11-08 17:50:52 -05:00
view.py [MLIR][TORCH] Add E2E support for `torch.aten.view` 2021-10-29 22:33:10 +05:30
vision_models.py Add min/max/clamp support. 2021-10-27 13:29:21 -07:00
xfail_sets.py [tosa] Add Torch.sigmoid fp32 to TOSA (#386) 2021-10-28 10:09:12 -07:00