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 afdaa60dd4
Fix typo in `inputRank` check of `AtenBatchNormOp` (#1046) 
The original conversion pattern for `AtenBatchNormOp` required that
the input rank be greater than 2; however, the only
expectation in the conversion pattern and in Pytorch is that the input
rank is greater than 1, since the second dimension of the input must
match the size of the `weight`, `bias`, `runningMean`, and
`runningVar` inputs. This commit fixes the `inputRank` check.
 		2022-07-15 09:35:59 -07:00
..
__init__.py Add lowering for _convolution 2022-07-11 11:03:03 +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 Add E2E support for aten.len.str (#969) 2022-07-07 10:41:55 -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 [LINALG] Add support for preserve memory format in aten_empty_like op. 2022-05-10 09:37:55 +05:30
control_flow.py [TORCH][MLIR]Add lowering for control flow operations. 2022-04-29 16:25:58 +05:30
conv.py Add lowering for _convolution 2022-07-11 11:03:03 +05:30
custom_op_example.py Allow torch-mlir to support PyTorch extensions. (#895) 2022-06-13 14:51:30 -07:00
elementwise.py linalg: lower `aten.triu` op to `linalg.generic` (#965) 2022-06-23 22:45:48 -07:00
elementwise_comparison.py Add E2E support for AtenAllBoolOp (#874) 2022-06-01 18:20:25 -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 Make e2e testing parallel 2022-05-05 21:27:58 -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 [LINALG] Lower `aten.Matmul` to `linalg.BatchMatmul` 2022-06-25 10:58:06 +05:30
mlp.py Handle `nn.Linear(..., bias=False)` case for TorchToLinalg (#919) 2022-06-08 21:13:43 -05: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 Fix typo in `inputRank` check of `AtenBatchNormOp` (#1046) 2022-07-15 09:35:59 -07:00
pooling.py [MLIR][TORCH] Add decomposition of aten.adaptive_avg_pool2d op 2022-05-27 07:56:37 +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 torch,linalg: add support for translating aten.linalg.vector_norm (#839) 2022-05-19 15:48:15 -07:00
reshape_like.py [LINALG] Fix handling of size-1 dims in `aten.view` again. (#992) 2022-06-30 16:39:25 -07:00
return_types.py [REFBACKEND] Add support for returning multiple different return types. 2022-04-21 09:02:30 +05:30
rng.py Add shape info to `rand_like` + support for `dtype` flag (#851) 2022-05-12 16:00:59 -07:00
scalar.py [MLIR][TORCH] Add E2E support for aten.Bool.[float|int] op 2022-05-24 21:18:34 +05:30
scalar_comparison.py [MLIR][TORCH] E2E support for [ge|ceil].float, [ge|ne|gt].float_int op 2022-05-05 21:48:35 +05:30
slice_like.py [TORCH][MLIR] Add lowering of `aten.slice_scatter` and 2022-07-11 14:07:21 +05:30
squeeze.py Move e2e test definitions into the `torch_mlir_e2e_test` package 2022-03-25 13:56:41 -07:00
stats.py [MLIR][TORCH] Add decomposition for aten.var.dim op 2022-07-15 09:53:42 +05:30
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 [MLIR][TORCH] Add support for bool type in convertScalarToDtype utility 2022-06-16 00:00:47 +05:30
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