@@ -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+
31243156cc_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+
64586539gentbl_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