@@ -3121,6 +3121,38 @@ gentbl_cc_library(
3121
3121
],
3122
3122
)
3123
3123
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
+
3124
3156
cc_library (
3125
3157
name = "MeshDialect" ,
3126
3158
srcs = ["lib/Dialect/Mesh/IR/MeshOps.cpp" ],
@@ -3136,6 +3168,40 @@ cc_library(
3136
3168
],
3137
3169
)
3138
3170
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
+
3139
3205
##---------------------------------------------------------------------------##
3140
3206
# NVGPU dialect.
3141
3207
##---------------------------------------------------------------------------##
@@ -5183,6 +5249,8 @@ cc_library(
5183
5249
":ROCDLTarget" ,
5184
5250
":ROCDLToLLVMIRTranslation" ,
5185
5251
":SCFDialect" ,
5252
+ ":SPIRVDialect" ,
5253
+ ":SPIRVTarget" ,
5186
5254
":SerializeToCubin_stub" ,
5187
5255
":SideEffectInterfaces" ,
5188
5256
":Support" ,
@@ -5619,6 +5687,7 @@ cc_library(
5619
5687
deps = [
5620
5688
":ArithToSPIRV" ,
5621
5689
":ConversionPassIncGen" ,
5690
+ ":FuncDialect" ,
5622
5691
":FuncToSPIRV" ,
5623
5692
":GPUDialect" ,
5624
5693
":IR" ,
@@ -6438,6 +6507,7 @@ cc_library(
6438
6507
":CommonFolders" ,
6439
6508
":ControlFlowInterfaces" ,
6440
6509
":FunctionInterfaces" ,
6510
+ ":GPUDialect" ,
6441
6511
":IR" ,
6442
6512
":InferTypeOpInterface" ,
6443
6513
":Parser" ,
@@ -6455,6 +6525,17 @@ cc_library(
6455
6525
],
6456
6526
)
6457
6527
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
+
6458
6539
gentbl_cc_library (
6459
6540
name = "SPIRVPassIncGen" ,
6460
6541
tbl_outs = [
@@ -8633,6 +8714,7 @@ cc_library(
8633
8714
":MemRefTransformOps" ,
8634
8715
":MemRefTransforms" ,
8635
8716
":MeshDialect" ,
8717
+ ":MeshTransforms" ,
8636
8718
":NVGPUDialect" ,
8637
8719
":NVGPUPassIncGen" ,
8638
8720
":NVGPUToNVVM" ,
@@ -8659,6 +8741,7 @@ cc_library(
8659
8741
":SDBM" ,
8660
8742
":SPIRVDialect" ,
8661
8743
":SPIRVPassIncGen" ,
8744
+ ":SPIRVTarget" ,
8662
8745
":SPIRVToLLVM" ,
8663
8746
":SPIRVTransforms" ,
8664
8747
":ShapeDialect" ,
@@ -11053,6 +11136,8 @@ cc_library(
11053
11136
":IR" ,
11054
11137
":InferTypeOpInterface" ,
11055
11138
":LoopLikeInterface" ,
11139
+ ":MeshDialect" ,
11140
+ ":MeshShardingInterface" ,
11056
11141
":Pass" ,
11057
11142
":QuantOps" ,
11058
11143
":Support" ,
0 commit comments