torch-mlir/python/torch_mlir_e2e_test/test_suite
Bruce Kim a520d39f84
[MLIR][TORCH] Add device "cpu" support for aten.to.dtype_layout op (#2481)
This PR adds device="cpu" support for `aten.to_dtypeLayout` op and
corresponding e2e test suit.
(refer:  PR https://github.com/llvm/torch-mlir/pull/812/)
2023-09-25 10:00:19 -04:00
..
__init__.py Fix version comparison against stable (#2209) 2023-06-07 10:19:38 +02:00
arange.py Revert "[MLIR][TORCH] Fix indentation and spacing for E2E tests" 2022-11-25 12:41:56 +05:30
argmax.py Revert "build: update llvm tag to 147fe9de" 2022-11-25 12:41:56 +05:30
backprop.py [MLIR][TORCH] Add support for non-unit stride for conv backward 2023-04-04 17:53:27 +05:30
basic.py [MLIR][TORCH] Add different dtype support for aten.bmm op 2023-09-12 12:38:46 +05:30
cast.py Replace `torch.rand` and `torch.randn` in e2e tests with `tu.rand` (#1890) 2023-02-21 14:30:05 -08:00
constant_alloc.py [MLIR][TORCH] Add E2E support for aten.empty_strided decomposition op (redo PR) (#2459) 2023-09-13 10:04:31 -07:00
control_flow.py Revert "build: update llvm tag to 147fe9de" 2022-11-25 12:41:56 +05:30
conv.py [TOSA] Fix conversion for depthwise convolutions (#2398) 2023-08-18 08:15:54 -07:00
custom_op_example.py Revert "build: update llvm tag to 147fe9de" 2022-11-25 12:41:56 +05:30
elementwise.py [Torch Dialect] add canonicalize for aten.min.other (#2452) 2023-09-11 17:28:22 +08:00
elementwise_comparison.py lib/Conversion/TorchToTosa/TorchToTosa.cpp: Fix legalization of comparions where the input type is bool (#2304) 2023-07-17 09:49:04 +02:00
histogram_binning_calibration.py Replace `torch.rand` and `torch.randn` in e2e tests with `tu.rand` (#1890) 2023-02-21 14:30:05 -08:00
index_select.py [MLIR][TORCH] Patch up Ops and their lowerings to deal with +ve `dim` 2023-04-14 13:12:56 +05:30
matmul.py Revert "build: update llvm tag to 147fe9de" 2022-11-25 12:41:56 +05:30
mlp.py Revert "build: update llvm tag to 147fe9de" 2022-11-25 12:41:56 +05:30
nll_loss.py [MLIR][TORCH] Add support for the total_weight for aten.nll_loss_forward op 2023-05-30 20:29:27 +05:30
norm_like.py [Stablehlo] add e2e test for aten.batch_norm (#2129) 2023-05-17 09:04:40 -07:00
pooling.py [Torch Dialect] add support for adaptive_avgpool_1d (#2342) 2023-08-05 07:48:09 +08:00
quantized_models.py Revert "[MLIR][TORCH] Fix indentation and spacing for E2E tests" 2022-11-25 12:41:56 +05:30
reduction.py [MLIR][TORCH] Add E2E `torch.aten.prod_dim_int` (#2423) 2023-09-05 13:38:51 -07:00
reshape_like.py [TOSA] Add aten.view shape -1 support (#1815) 2023-01-20 11:56:26 -08:00
return_types.py Add float16 support in the refbackend. 2022-12-15 21:19:52 +05:30
rng.py [Torch Dialect][stablehlo] emit aten.rand op and add converter to stablehlo (#2413) 2023-08-27 21:56:36 +08:00
scalar.py Replace `torch.rand` and `torch.randn` in e2e tests with `tu.rand` (#1890) 2023-02-21 14:30:05 -08:00
scalar_comparison.py Replace `torch.rand` and `torch.randn` in e2e tests with `tu.rand` (#1890) 2023-02-21 14:30:05 -08:00
scatter.py [MLIR][TORCH] Add support for aten._unsafe_index_put.hacked_twin op 2023-08-11 08:57:01 +05:30
slice_like.py [Torch Dialect] support aten.split_with_sizes (#2431) 2023-09-04 09:59:26 +08:00
squeeze.py [MLIR][TORCH] Add decomposition for prims::squeeze op 2023-04-01 21:45:58 +05:30
stats.py [MLIR][TORCH] Add E2E support for aten.var_mean.dim op 2023-04-27 22:00:44 +05:30
threshold.py Replace `torch.rand` and `torch.randn` in e2e tests with `tu.rand` (#1890) 2023-02-21 14:30:05 -08:00
type_conversion.py [MLIR][TORCH] Add device "cpu" support for aten.to.dtype_layout op (#2481) 2023-09-25 10:00:19 -04:00
type_promotion.py Replace `torch.rand` and `torch.randn` in e2e tests with `tu.rand` (#1890) 2023-02-21 14:30:05 -08:00
vision_models.py [cleanup] Remove disabled e2e test 2022-12-07 06:36:48 -08:00