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
Prateek Gupta e1db318a3c [TORCH][MLIR]Add lowering for control flow operations. 
1. This commit adds lowering of "while-like" prim loop to scf.while
operation.
2. Adds lowering of "for-like" prim loops to scf.for operation.

Signed-Off-By: Prateek Gupta <prateek@nod-labs.com>
 		2022-04-29 16:25:58 +05:30
..
__init__.py [TORCH][MLIR]Add lowering for control flow operations. 2022-04-29 16:25:58 +05:30
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 [LINALG] Bug fix i64 vs i32 type comparison. 2022-04-22 08:09:58 +05:30
cast.py Move e2e test definitions into the `torch_mlir_e2e_test` package 2022-03-25 13:56:41 -07:00
constant_alloc.py [LINALG] Add torch.layout information 2022-04-07 20:47:49 +05:30
control_flow.py [TORCH][MLIR]Add lowering for control flow operations. 2022-04-29 16:25:58 +05:30
conv.py Add 2D case for convolution (#693) 2022-04-08 00:47:57 -04:00
elementwise.py test: add end-to-end test for aten.neg (#760) 2022-04-15 12:37:57 -07:00
elementwise_comparison.py [LINALG] Add the lowering of `aten.ne.Scalar` op 2022-04-05 21:07:28 +05:30
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 Fix the input tensors inplace update issue for e2e tests 2022-04-28 11:43:54 -04: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 Add 1D, weight, and reduction support to nll_loss_backward (#729) 2022-04-04 10:57:49 -07:00
norm_like.py Move e2e test definitions into the `torch_mlir_e2e_test` package 2022-03-25 13:56:41 -07:00
pooling.py [MLIR][TORCH] Add E2E support for max_pool2d_with_indices op 2022-04-18 21:05:19 +05:30
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
return_types.py [REFBACKEND] Add support for returning multiple different return types. 2022-04-21 09:02:30 +05:30
rng.py Move e2e test definitions into the `torch_mlir_e2e_test` package 2022-03-25 13:56:41 -07:00
scalar.py Revert "[MLIR][TORCH] Add E2E support for aten.ceil.float op" 2022-04-28 07:49:58 -07:00
scalar_comparison.py Revert "[MLIR][TORCH] Add E2E support for aten.ge.float op" 2022-04-28 07:49:58 -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