Skip to content

Commit 72ad6cd

Browse files
committed
[mlir][bazel] Fix build.
1 parent 58679ea commit 72ad6cd

File tree

1 file changed

+85
-0
lines changed

1 file changed

+85
-0
lines changed

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

Lines changed: 85 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3121,6 +3121,38 @@ gentbl_cc_library(
31213121
],
31223122
)
31233123

3124+
gentbl_cc_library(
3125+
name = "MeshShardingInterfaceIncGen",
3126+
tbl_outs = [
3127+
(
3128+
["-gen-op-interface-decls"],
3129+
"include/mlir/Dialect/Mesh/Interfaces/ShardingInterface.h.inc",
3130+
),
3131+
(
3132+
["-gen-op-interface-defs"],
3133+
"include/mlir/Dialect/Mesh/Interfaces/ShardingInterface.cpp.inc",
3134+
),
3135+
],
3136+
tblgen = ":mlir-tblgen",
3137+
td_file = "include/mlir/Dialect/Mesh/Interfaces/ShardingInterface.td",
3138+
deps = [":OpBaseTdFiles"],
3139+
)
3140+
3141+
cc_library(
3142+
name = "MeshShardingInterface",
3143+
srcs = ["lib/Dialect/Mesh/Interfaces/ShardingInterface.cpp"],
3144+
hdrs = ["include/mlir/Dialect/Mesh/Interfaces/ShardingInterface.h"],
3145+
includes = ["include"],
3146+
deps = [
3147+
":DialectUtils",
3148+
":IR",
3149+
":MeshDialect",
3150+
":MeshShardingInterfaceIncGen",
3151+
":Support",
3152+
"//llvm:Support",
3153+
],
3154+
)
3155+
31243156
cc_library(
31253157
name = "MeshDialect",
31263158
srcs = ["lib/Dialect/Mesh/IR/MeshOps.cpp"],
@@ -3136,6 +3168,40 @@ cc_library(
31363168
],
31373169
)
31383170

3171+
gentbl_cc_library(
3172+
name = "MeshTransformsPassIncGen",
3173+
tbl_outs = [
3174+
(
3175+
[
3176+
"-gen-pass-decls",
3177+
"-name=Mesh",
3178+
],
3179+
"include/mlir/Dialect/Mesh/Transforms/Passes.h.inc",
3180+
),
3181+
],
3182+
tblgen = ":mlir-tblgen",
3183+
td_file = "include/mlir/Dialect/Mesh/Transforms/Passes.td",
3184+
deps = [":PassBaseTdFiles"],
3185+
)
3186+
3187+
cc_library(
3188+
name = "MeshTransforms",
3189+
srcs = glob([
3190+
"lib/Dialect/Mesh/Transforms/*.cpp",
3191+
"lib/Dialect/Mesh/Transforms/*.h",
3192+
]),
3193+
hdrs = glob(["include/mlir/Dialect/Mesh/Transforms/*.h"]),
3194+
includes = ["include"],
3195+
deps = [
3196+
":FuncDialect",
3197+
":MeshDialect",
3198+
":MeshShardingInterface",
3199+
":MeshTransformsPassIncGen",
3200+
":Pass",
3201+
"//llvm:Support",
3202+
],
3203+
)
3204+
31393205
##---------------------------------------------------------------------------##
31403206
# NVGPU dialect.
31413207
##---------------------------------------------------------------------------##
@@ -5183,6 +5249,8 @@ cc_library(
51835249
":ROCDLTarget",
51845250
":ROCDLToLLVMIRTranslation",
51855251
":SCFDialect",
5252+
":SPIRVDialect",
5253+
":SPIRVTarget",
51865254
":SerializeToCubin_stub",
51875255
":SideEffectInterfaces",
51885256
":Support",
@@ -5619,6 +5687,7 @@ cc_library(
56195687
deps = [
56205688
":ArithToSPIRV",
56215689
":ConversionPassIncGen",
5690+
":FuncDialect",
56225691
":FuncToSPIRV",
56235692
":GPUDialect",
56245693
":IR",
@@ -6438,6 +6507,7 @@ cc_library(
64386507
":CommonFolders",
64396508
":ControlFlowInterfaces",
64406509
":FunctionInterfaces",
6510+
":GPUDialect",
64416511
":IR",
64426512
":InferTypeOpInterface",
64436513
":Parser",
@@ -6455,6 +6525,17 @@ cc_library(
64556525
],
64566526
)
64576527

6528+
cc_library(
6529+
name = "SPIRVTarget",
6530+
srcs = ["lib/Target/SPIRV/Target.cpp"],
6531+
hdrs = glob(["include/mlir/Target/SPIRV/*.h"]),
6532+
includes = ["include"],
6533+
deps = [
6534+
":GPUDialect",
6535+
":SPIRVDialect",
6536+
],
6537+
)
6538+
64586539
gentbl_cc_library(
64596540
name = "SPIRVPassIncGen",
64606541
tbl_outs = [
@@ -8633,6 +8714,7 @@ cc_library(
86338714
":MemRefTransformOps",
86348715
":MemRefTransforms",
86358716
":MeshDialect",
8717+
":MeshTransforms",
86368718
":NVGPUDialect",
86378719
":NVGPUPassIncGen",
86388720
":NVGPUToNVVM",
@@ -8659,6 +8741,7 @@ cc_library(
86598741
":SDBM",
86608742
":SPIRVDialect",
86618743
":SPIRVPassIncGen",
8744+
":SPIRVTarget",
86628745
":SPIRVToLLVM",
86638746
":SPIRVTransforms",
86648747
":ShapeDialect",
@@ -11053,6 +11136,8 @@ cc_library(
1105311136
":IR",
1105411137
":InferTypeOpInterface",
1105511138
":LoopLikeInterface",
11139+
":MeshDialect",
11140+
":MeshShardingInterface",
1105611141
":Pass",
1105711142
":QuantOps",
1105811143
":Support",

0 commit comments

Comments
 (0)