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
History
Abhishek Varma 5337944ddb [MLIR][TORCH] Add e2e support for aten.randint 
-- This commit adds e2e support for aten.randint by decomposing it into
   an aten.randint.low by setting low=0.

Signed-off-by: Abhishek Varma <abhishek@nod-labs.com>
 		2023-04-07 00:13:56 +05:30
..
_torch_mlir_custom_op_example [custom op] Generalize shape library logic to work with dtypes (#1594) 2022-12-13 08:25:41 -08:00
cmake/modules Add initial LTC backend (#610) 2022-07-30 09:40:02 -04:00
csrc Update LLVM tag to 21f4b84c (#1918) 2023-03-06 22:53:26 -08:00
dialects [MLIR][TORCH] Add e2e support for aten.randint 2023-04-07 00:13:56 +05:30
__init__.py Add `extra_library` kwarg to `torch_mlir.compile` (#1986) 2023-03-30 09:20:19 -07:00
compiler_utils.py Update LLVM tag to 21f4b84c (#1918) 2023-03-06 22:53:26 -08:00
dynamo.py [TorchDynamo] Add aten.squeeze op to the decomposition list 2023-04-06 22:21:25 +05:30