torch-mlir/python/torch_mlir_e2e_test
武家伟 ab7aa01b1e
[MHLO] Add torch-to-mhlo e2e support for aten.gather op (#1410)
* Add torch-to-mhlo e2e support for aten.gather op 

* Add more e2e tests for torch.aten.gather op
2022-09-25 22:07:46 +08:00
..
configs Remove "torchscript" association from the e2e framework. 2022-08-29 14:10:03 -07:00
eager_backends add ci tests (#754) 2022-05-25 14:59:59 -05:00
linalg_on_tensors_backends add e2e support for aten.atan2 (#1117) 2022-08-02 11:39:41 -04:00
mhlo_backends [MLIR][MHLO] Add AtenFrobeniusNormDimOp and add its conversion pattern to MHLO and linalg (#1306) 2022-09-08 10:15:36 +08:00
test_suite [MHLO] Add torch-to-mhlo e2e support for aten.gather op (#1410) 2022-09-25 22:07:46 +08:00
tosa_backends [tosa] Add "-tosa-to-tensor" pass in the lowering pipeline 2022-09-24 10:03:07 +05:30
CMakeLists.txt Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
__init__.py Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
annotations.py Remove "torchscript" association from the e2e framework. 2022-08-29 14:10:03 -07:00
framework.py Sort tests before anything else. 2022-08-29 16:23:56 -07:00
registry.py Ensure that tests have unique names 2022-08-29 16:25:23 -07:00
reporting.py Remove "torchscript" association from the e2e framework. 2022-08-29 14:10:03 -07:00
serialization.py Remove "torchscript" association from the e2e framework. 2022-08-29 14:10:03 -07:00
utils.py Remove "torchscript" association from the e2e framework. 2022-08-29 14:10:03 -07:00