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
Tanyo Kwok d4f1f41435
[MLIR][TORCH] Add decomposition of aten.repeat (#932) 
* [MLIR][TORCH] Add decomposition of aten.repeat

* refine & rebase

* refine static shapes

* add e2e test

* Rebase and Refine naming style
 		2022-07-01 13:02:31 +08:00
..
__init__.py Temporarily revert support for custom op extensions. (#944) 2022-06-14 18:24:40 -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 [MLIR][TORCH] Add decomposition of aten.repeat (#932) 2022-07-01 13:02:31 +08: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 2D case for convolution (#693) 2022-04-08 00:47:57 -04:00
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 Move e2e test definitions into the `torch_mlir_e2e_test` package 2022-03-25 13:56:41 -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 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
stats.py [LINALG] Add E2E support for `aten.mean.dim` op 2022-05-04 20:11: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