torch-mlir/python/torch_mlir_e2e_test/test_suite
Arham Khan 82456eefed
[MLIR][TORCH] add E2E support for aten.new_full (#2425)
* implement aten.new_full

* remove extraneous tests
2023-09-12 09:29:08 -05: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.new_full (#2425) 2023-09-12 09:29:08 -05: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 [torch-dialect] fix aten.type_as op's folder (#2283) 2023-07-20 09:51:58 +08: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