llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/e2e_testing/torchscript
History
Vivek Khandelwal 88c216da13 [MLIR][TORCH] Add support for same input and output shapes for view op 
This commit adds support for the cases of view op where the rank and
the shapes of the input and result are equal.

Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
 		2022-03-25 22:26:10 +05:30
..
arange.py [MLIR][TORCH] Add E2E support for torch.arange op 2021-12-27 22:45:48 +05:30
argmax.py add argmax lowering 2021-10-13 14:31:16 -04:00
backprop.py Add e2e test for aten.log_softmax_back_data op 2021-11-19 00:08:28 +05:30
basic.py [LINALG] Add decomposition of `aten.dropout` op 2022-03-22 13:14:49 +05:30
cast.py [LINALG] Add E2E support for `aten.[Bool.Tensor|Float.Tensor]` op 2022-02-14 23:09:20 +05:30
constant_alloc.py [LINALG] Add E2E support for `aten.zero_` op 2022-03-25 12:46:50 +05:30
conv.py Refine static shapes for conv2d and maxpool2d 2022-01-03 11:09:23 -06:00
elementwise.py Add non-default dtype support for a few elementwise math ops. (#687) 2022-03-23 13:35:43 -07:00
elementwise_comparison.py [LINALG] Add E2E support for `aten.[le|ge].Scalar` ops 2022-02-15 12:21:09 +05:30
histogram_binning_calibration.py Adding an e2e test for histogram binning calibration 2022-01-25 18:27:20 -05:00
index_put.py [MLIR][TORCH] Add E2E support for aten.index_put op 2022-03-16 22:02:02 +05:30
index_select.py [MLIR][TORCH] Add E2E support for aten.index_select op 2021-12-09 23:13:36 +05:30
main.py This PR implements an eager mode backend for PyTorch through the torch-mlir framework. This is accomplished by overriding the `__torch_dispatch__` class method on wrapper subclass `TorchMLIRTensor(torch.Tensor)`. 2022-03-22 14:42:57 -07:00
matmul.py Add lowering of aten.matmul op. 2021-10-26 12:45:09 -04:00
mlp.py Support aten::linear with rank 3 inputs 2021-11-18 22:15:04 +05:30
nll_loss.py Add `reduction` support to `torch.nll_loss_forward` (#624) 2022-02-28 11:01:23 -08:00
norm_like.py [TORCH][MLIR] Fix the return types of `aten.native_layer_norm`. 2022-03-17 12:08:32 +05:30
quantized_models.py Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
reduction.py [MLIR][TORCH]Add support for integer-type inputs for sum and max op 2022-03-08 22:52:34 +05:30
reshape_like.py [MLIR][TORCH] Add support for same input and output shapes for view op 2022-03-25 22:26:10 +05:30
rng.py This PR implements an eager mode backend for PyTorch through the torch-mlir framework. This is accomplished by overriding the `__torch_dispatch__` class method on wrapper subclass `TorchMLIRTensor(torch.Tensor)`. 2022-03-22 14:42:57 -07:00
scalar.py [LINALG] Add decomposition of `aten.dropout` op 2022-03-22 13:14:49 +05:30
scalar_comparison.py [LINALG] Add E2E support for `aten.eq.int` op 2022-02-15 01:37:35 +05:30
slice_like.py Introduce new shape library design. 2022-03-15 12:41:58 -07:00
squeeze.py Introduce new shape library design. 2022-03-15 12:41:58 -07:00
table_batch_embedding.py [TBE] Add a test module for table batch embedding 2022-01-28 02:24:28 +05:30
threshold.py [tosa] Support for some ops and fix for Issue #532 (#575) 2022-02-11 12:30:02 -08:00
type_conversion.py Convert bool to float or integer type. 2022-02-07 21:22:22 +05:30
type_promotion.py Add scalar type promotion for mul and div (#454) 2021-12-03 13:51:25 -06:00
vision_models.py Add aten.hardtanh e2e support. 2022-03-02 12:28:06 -05:00
xfail_sets.py Add non-default dtype support for a few elementwise math ops. (#687) 2022-03-23 13:35:43 -07:00