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 yibairen.byron@bytedance.com Co-authored-by: Jiawei Wu xremold@gmail.com Co-authored-by: Tianyou Guo tianyou.gty@alibaba-inc.com Co-authored-by: Xu Yan yancey.yx@alibaba-inc.com Co-authored-by: Ziheng Jiang ziheng.jiang@bytedance.com
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)