Skip to content

Commit

Permalink
[MHLO] Init MHLO matmul op patterns
Browse files Browse the repository at this point in the history
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
  • Loading branch information
Tanyo Kwok committed Aug 2, 2022
1 parent 3772e0b commit ce88297
Show file tree
Hide file tree
Showing 5 changed files with 1,024 additions and 2 deletions.
3 changes: 3 additions & 0 deletions lib/Conversion/TorchToMhlo/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -3,13 +3,15 @@ add_mlir_conversion_library(TorchMLIRTorchToMhlo
MhloLegalizeUtils.cpp
BasicOp.cpp
GatherOp.cpp
MatmulOp.cpp
ViewLikeOps.cpp

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

DEPENDS
MhloDialect
ChloDialect
TorchMLIRConversionPassIncGen

LINK_COMPONENTS
Expand All @@ -19,6 +21,7 @@ add_mlir_conversion_library(TorchMLIRTorchToMhlo
MLIRIR
MLIRPass
MhloDialect
ChloDialect
TorchMLIRTorchDialect
)

Expand Down
Loading

0 comments on commit ce88297

Please sign in to comment.