Skip to content

Commit 69599c5

Browse files
bangtianliubangtianliu
committed
[Codegen][GPU] Add DistributeArgCompare pattern
Signed-off-by: Bangtian Liu <liubangtian@gmail.com>
1 parent cdf0367 commit 69599c5

File tree

6 files changed

+1611
-6
lines changed

6 files changed

+1611
-6
lines changed

‎compiler/src/iree/compiler/Codegen/Common/GPU/BUILD.bazel‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -158,6 +158,7 @@ iree_compiler_cc_library(
158158
"@llvm-project//mlir:LinalgTransforms",
159159
"@llvm-project//mlir:LinalgUtils",
160160
"@llvm-project//mlir:LoopLikeInterface",
161+
"@llvm-project//mlir:MathDialect",
161162
"@llvm-project//mlir:MemRefDialect",
162163
"@llvm-project//mlir:MemRefTransforms",
163164
"@llvm-project//mlir:MemRefUtils",

‎compiler/src/iree/compiler/Codegen/Common/GPU/CMakeLists.txt‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -122,6 +122,7 @@ iree_cc_library(
122122
MLIRLinalgTransforms
123123
MLIRLinalgUtils
124124
MLIRLoopLikeInterface
125+
MLIRMathDialect
125126
MLIRMemRefDialect
126127
MLIRMemRefTransforms
127128
MLIRMemRefUtils

0 commit comments

Comments
 (0)