torch-mlir/python/torch_mlir_e2e_test
Vivek Khandelwal f6a6cfea4e
[MLIR][TORCH] Add support for negative index values for index.Tensor op (#2233)
This commit adds the support for index.Tensor op when the index values
are negative. This commit wraps around the index values by checking
their values at run time.

Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2023-06-16 14:21:04 -05: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 [MLIR][TORCH] Add support for negative index values for index.Tensor op (#2233) 2023-06-16 14:21:04 -05: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 Print name of the backend when tests fail to help debugging issues in CI (#2210) 2023-06-09 10:47:07 +02:00
utils.py Remove "torchscript" association from the e2e framework. 2022-08-29 14:10:03 -07:00