torch-mlir/lib/Conversion
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
..
TorchConversionToMLProgram Fix a few missing dependencies. (#2014) 2023-04-11 11:18:49 -07:00
TorchToArith [Torch Dialect] Add canonicalize pattern for aten.is_floating_point (#2194) 2023-06-07 17:05:31 +08:00
TorchToLinalg [MLIR][TORCH] Add support for negative index values for index.Tensor op (#2233) 2023-06-16 14:21:04 -05:00
TorchToSCF build: update llvm tag to 798fa4b4 (#1684) 2022-12-07 12:20:41 -08:00
TorchToStablehlo [Stablehlo] add conversion for AtenFlipOp (#2163) 2023-06-15 10:27:34 +08:00
TorchToTMTensor [TM_TENSOR] Add `aten.scatter.[src|value]` op 2023-05-29 12:35:53 +05:30
TorchToTosa TorchToTosa: Cast float constants to correct type to support bfloat16 (#2239) 2023-06-16 09:51:24 +02:00
Utils Update to LLVM 3157f03a349cfc852cdd994675eaa9652caa2e3a (#2060) 2023-04-25 08:52:46 -07:00
CMakeLists.txt mhlo: migrate conversion to stablehlo (#1840) 2023-02-02 07:29:47 -06:00
PassDetail.h Minor fixes for `ConvertTorchConversionToMLProgram`. (#1991) 2023-04-04 09:09:58 -07:00
Passes.cpp mhlo: migrate conversion to stablehlo (#1840) 2023-02-02 07:29:47 -06:00