Skip to content

Commit a258544

Browse files
committed
[bazel][mesh] Add deps for 1D device mesh resharding spmdization
Added in 1a8fb88
1 parent ab43cf2 commit a258544

File tree

2 files changed

+8
-0
lines changed

2 files changed

+8
-0
lines changed

utils/bazel/llvm-project-overlay/mlir/BUILD.bazel

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3197,6 +3197,7 @@ gentbl_cc_library(
31973197
td_file = "include/mlir/Dialect/Mesh/IR/MeshOps.td",
31983198
deps = [
31993199
":MeshTdFiles",
3200+
":ShapeOpsTdFiles",
32003201
],
32013202
)
32023203

@@ -3275,12 +3276,16 @@ cc_library(
32753276
includes = ["include"],
32763277
deps = [
32773278
":ArithDialect",
3279+
":ControlFlowDialect",
3280+
":DialectUtils",
32783281
":FuncDialect",
32793282
":IR",
32803283
":MeshDialect",
32813284
":MeshShardingInterface",
32823285
":MeshTransformsPassIncGen",
32833286
":Pass",
3287+
":Support",
3288+
":TensorDialect",
32843289
":TransformUtils",
32853290
"//llvm:Support",
32863291
],

utils/bazel/llvm-project-overlay/mlir/test/BUILD.bazel

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -789,9 +789,12 @@ cc_library(
789789
":TestDialect",
790790
"//mlir:ArithDialect",
791791
"//mlir:FuncDialect",
792+
"//mlir:IR",
792793
"//mlir:MeshDialect",
793794
"//mlir:MeshTransforms",
794795
"//mlir:Pass",
796+
"//mlir:SPIRVDialect",
797+
"//mlir:Support",
795798
"//mlir:Transforms",
796799
],
797800
)

0 commit comments

Comments
 (0)