torch-mlir/python/torch_mlir_e2e_test
Zhekun Zhang aa97c8383e
[Torch Op] Add unbind.int support with ListUnpack (#2058)
* add unbind int

* reformat

* use unpack canonicalize

* address comments

* Empty commit, trigger test

* add ltc blacklist

* clean up

* address comments

* check permute list

* erase in recompose

---------

Co-authored-by: zhekun.zhang <zhekun.zhang@bytedance.com>
2023-05-18 19:07:58 -07:00
..
configs Pure-Python FX importer. (#2098) 2023-05-12 00:46:33 -05:00
debug [torchdynamo] Add "lockstep" numerical accuracy debugger. 2022-12-06 07:57:45 -08:00
linalg_on_tensors_backends [Stablehlo] add e2e test for aten.batch_norm (#2129) 2023-05-17 09:04:40 -07:00
stablehlo_backends [Stablehlo] add e2e test for aten.batch_norm (#2129) 2023-05-17 09:04:40 -07:00
test_suite [Torch Op] Add unbind.int support with ListUnpack (#2058) 2023-05-18 19:07:58 -07:00
tosa_backends Update to LLVM 029313cc979ae71877b65794b1063d4e51184cc8 2023-03-21 04:16:20 -07:00
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 [Torch Dialect] Scatter reduce lowering (#1884) 2023-02-21 23:05:55 +00: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
utils.py Remove "torchscript" association from the e2e framework. 2022-08-29 14:10:03 -07:00