llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/lib/Conversion/TorchToLinalg
History
JianzheXiao e4f8fb1b8c
[Torch Dialect] add support for AtenIsnanOp (#2170) 
* add support for mhlo

* Add Test for torch.ne

* fix torch.ne shape/add static test case

* add support for static torch.ne

---------

Co-authored-by: root <root@n31-177-039.byted.org>
 		2023-06-07 10:06:27 +08:00
..
CMakeLists.txt Re-enable custom op support 2022-08-16 22:49:08 +05:30
DataMovement.cpp Fix torchdynamo fail test. 2023-05-11 21:29:07 +05:30
IndirectDataMovement.cpp [MLIR][TORCH] Patch up Ops and their lowerings to deal with +ve `dim` 2023-04-14 13:12:56 +05:30
Linear.cpp [MLIR][TORCH] Add support for non-unit stride for conv backward 2023-04-04 17:53:27 +05:30
Pooling.cpp [Linalg] fix lowering reduce max with -inf (#2097) 2023-05-08 09:17:49 -07:00
PopulatePatterns.h Re-enable custom op support 2022-08-16 22:49:08 +05:30
Random.cpp build: update llvm tag to 798fa4b4 (#1684) 2022-12-07 12:20:41 -08:00
Reduction.cpp [Linalg] fix lowering reduce max with -inf (#2097) 2023-05-08 09:17:49 -07:00
TensorConstructors.cpp Fix creation of empty tensor in decomposition for randn ops (#2043) 2023-04-19 08:25:39 -07:00
TensorScalarInterop.cpp build: update llvm tag to 798fa4b4 (#1684) 2022-12-07 12:20:41 -08:00
TorchToLinalg.cpp Add complex types support with basic complex ops. 2023-05-11 21:29:07 +05:30
Uncategorized.cpp [Torch Dialect] add support for AtenIsnanOp (#2170) 2023-06-07 10:06:27 +08:00
Utils.cpp [MLIR][TORCH] Add support for non-unit stride for conv backward 2023-04-04 17:53:27 +05:30
Utils.h [MLIR][TORCH] Add support for non-unit stride for conv backward 2023-04-04 17:53:27 +05:30