llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/python/torch_mlir_e2e_test/test_suite
History
Ramiro Leal-Cavazos 96d662647f
Fix import of constant bool tensor parameters (#2047) 
Bool tensors are represented in TorchScript as an array of
`int8_t`s. However, when importing them into Torch-MLIR, the importer
was assuming the array had `int32_t` elements, leading to the importer
reading into memory that was out of bounds. This commit fixes the
casting of the bool tensor.
 		2023-04-20 18:38:48 -07:00
..
__init__.py [Torch Dialect] Scatter reduce lowering (#1884) 2023-02-21 23:05:55 +00: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 Fix import of constant bool tensor parameters (#2047) 2023-04-20 18:38:48 -07:00
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 [TOSA] Add support for AtenZerosOp 0/strided layout (#1983) 2023-03-30 07:08:20 -07:00
control_flow.py Revert "build: update llvm tag to 147fe9de" 2022-11-25 12:41:56 +05:30
conv.py [MLIR][TORCH] Add support for non-unit stride for conv backward 2023-04-04 17:53:27 +05:30
custom_op_example.py Revert "build: update llvm tag to 147fe9de" 2022-11-25 12:41:56 +05:30
elementwise.py [Torch Dialect] add RSub, ScalarImplicit canonicalize (#1899) 2023-03-06 17:38:27 -08:00
elementwise_comparison.py Add aten.bucketize op and its decomposition (#1834) 2023-02-03 10:20:47 +08: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 Replace `torch.rand` and `torch.randn` in e2e tests with `tu.rand` (#1890) 2023-02-21 14:30:05 -08:00
norm_like.py Added support for aten::norm.ScalarOpt_dim (#1774) 2023-01-10 13:08:25 -05:00
pooling.py Max pool2d ceil mode to tosa (#1957) 2023-03-21 10:17:39 -07: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 support for cross entropy lowering 2023-04-18 08:00:20 +05:30
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 Fix creation of empty tensor in decomposition for randn ops (#2043) 2023-04-19 08:25:39 -07: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] Extend implementation of aten._index_put_impl op. 2023-04-05 14:04:30 +05:30
slice_like.py [StableHLO] Support for slice_scatter (#1960) 2023-03-22 13:41:04 -07:00
squeeze.py [MLIR][TORCH] Add decomposition for prims::squeeze op 2023-04-01 21:45:58 +05:30
stats.py Add aten.std.correction op and its decomposition (#1731) 2022-12-21 21:02:40 -08:00
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 Revert "build: update llvm tag to 147fe9de" 2022-11-25 12:41:56 +05:30
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