Skip to content

Commit ce88297

Browse files
author
Tanyo Kwok
committed
[MHLO] Init MHLO matmul op patterns
See RFC #999 Co-authored-by: Bairen Yi [email protected] Co-authored-by: Jiawei Wu [email protected] Co-authored-by: Tianyou Guo [email protected] Co-authored-by: Xu Yan [email protected] Co-authored-by: Ziheng Jiang [email protected]
1 parent 3772e0b commit ce88297

File tree

5 files changed

+1024
-2
lines changed

5 files changed

+1024
-2
lines changed

lib/Conversion/TorchToMhlo/CMakeLists.txt

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,13 +3,15 @@ add_mlir_conversion_library(TorchMLIRTorchToMhlo
33
MhloLegalizeUtils.cpp
44
BasicOp.cpp
55
GatherOp.cpp
6+
MatmulOp.cpp
67
ViewLikeOps.cpp
78

89
ADDITIONAL_HEADER_DIRS
910
${PROJECT_SOURCE_DIR}/include/torch-mlir/Conversion/TorchToMhlo
1011

1112
DEPENDS
1213
MhloDialect
14+
ChloDialect
1315
TorchMLIRConversionPassIncGen
1416

1517
LINK_COMPONENTS
@@ -19,6 +21,7 @@ add_mlir_conversion_library(TorchMLIRTorchToMhlo
1921
MLIRIR
2022
MLIRPass
2123
MhloDialect
24+
ChloDialect
2225
TorchMLIRTorchDialect
2326
)
2427

0 commit comments

Comments
 (0)