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 51d4d55f8a
Add support for multi-dim input to `index_put_impl` (#722) 
This commit adds support for multi-dimensional tensors as input to the
`_index_put_impl_` op. The support was to some degree already there,
since `ScatterOp` already supports multi-dimensional tensors. This
commit also adds a bit more error checking to `index_put` and
refactors the code for creating `ScatterOp`s to mimic the way one
would make a `Linalg::GenericOp`.
 		2022-03-31 09:27:21 -07:00
..
__init__.py Move e2e test definitions into the `torch_mlir_e2e_test` package 2022-03-25 13:56:41 -07:00
arange.py Move e2e test definitions into the `torch_mlir_e2e_test` package 2022-03-25 13:56:41 -07:00
argmax.py Move e2e test definitions into the `torch_mlir_e2e_test` package 2022-03-25 13:56:41 -07:00
backprop.py Move e2e test definitions into the `torch_mlir_e2e_test` package 2022-03-25 13:56:41 -07:00
basic.py Move e2e test definitions into the `torch_mlir_e2e_test` package 2022-03-25 13:56:41 -07:00
cast.py Move e2e test definitions into the `torch_mlir_e2e_test` package 2022-03-25 13:56:41 -07:00
constant_alloc.py [MLIR][TORCH] Add E2E support for aten.new_empty op 2022-03-30 13:21:01 +05:30
conv.py Move e2e test definitions into the `torch_mlir_e2e_test` package 2022-03-25 13:56:41 -07:00
elementwise.py [LINALG] Add E2E support for `aten.where.[Scalar|ScalarSelf|ScalarOther]` ops 2022-03-30 20:36:48 +05:30
elementwise_comparison.py Move e2e test definitions into the `torch_mlir_e2e_test` package 2022-03-25 13:56:41 -07:00
histogram_binning_calibration.py Move e2e test definitions into the `torch_mlir_e2e_test` package 2022-03-25 13:56:41 -07:00
index_put.py Add support for multi-dim input to `index_put_impl` (#722) 2022-03-31 09:27:21 -07:00
index_select.py Move e2e test definitions into the `torch_mlir_e2e_test` package 2022-03-25 13:56:41 -07:00
matmul.py Move e2e test definitions into the `torch_mlir_e2e_test` package 2022-03-25 13:56:41 -07:00
mlp.py Move e2e test definitions into the `torch_mlir_e2e_test` package 2022-03-25 13:56:41 -07:00
nll_loss.py Move e2e test definitions into the `torch_mlir_e2e_test` package 2022-03-25 13:56:41 -07:00
norm_like.py Move e2e test definitions into the `torch_mlir_e2e_test` package 2022-03-25 13:56:41 -07:00
quantized_models.py Move e2e test definitions into the `torch_mlir_e2e_test` package 2022-03-25 13:56:41 -07:00
reduction.py Move e2e test definitions into the `torch_mlir_e2e_test` package 2022-03-25 13:56:41 -07:00
reshape_like.py This commit decomposes aten._reshape_alias op into aten.view op. (#690) 2022-03-28 23:54:28 -05:00
rng.py Move e2e test definitions into the `torch_mlir_e2e_test` package 2022-03-25 13:56:41 -07:00
scalar.py Move e2e test definitions into the `torch_mlir_e2e_test` package 2022-03-25 13:56:41 -07:00
scalar_comparison.py Move e2e test definitions into the `torch_mlir_e2e_test` package 2022-03-25 13:56:41 -07:00
slice_like.py Move e2e test definitions into the `torch_mlir_e2e_test` package 2022-03-25 13:56:41 -07:00
squeeze.py Move e2e test definitions into the `torch_mlir_e2e_test` package 2022-03-25 13:56:41 -07:00
table_batch_embedding.py Move e2e test definitions into the `torch_mlir_e2e_test` package 2022-03-25 13:56:41 -07:00
threshold.py Move e2e test definitions into the `torch_mlir_e2e_test` package 2022-03-25 13:56:41 -07:00
type_conversion.py Move e2e test definitions into the `torch_mlir_e2e_test` package 2022-03-25 13:56:41 -07:00
type_promotion.py Move e2e test definitions into the `torch_mlir_e2e_test` package 2022-03-25 13:56:41 -07:00
vision_models.py Move e2e test definitions into the `torch_mlir_e2e_test` package 2022-03-25 13:56:41 -07:00