@@ -3268,6 +3268,20 @@ gentbl_cc_library(
3268
3268
],
3269
3269
"include/mlir/Dialect/Mesh/IR/MeshAttributes.cpp.inc" ,
3270
3270
),
3271
+ (
3272
+ [
3273
+ "-gen-typedef-decls" ,
3274
+ "-typedefs-dialect=mesh" ,
3275
+ ],
3276
+ "include/mlir/Dialect/Mesh/IR/MeshTypes.h.inc" ,
3277
+ ),
3278
+ (
3279
+ [
3280
+ "-gen-typedef-defs" ,
3281
+ "-typedefs-dialect=mesh" ,
3282
+ ],
3283
+ "include/mlir/Dialect/Mesh/IR/MeshTypes.cpp.inc" ,
3284
+ ),
3271
3285
],
3272
3286
tblgen = ":mlir-tblgen" ,
3273
3287
td_file = "include/mlir/Dialect/Mesh/IR/MeshOps.td" ,
@@ -3308,6 +3322,26 @@ cc_library(
3308
3322
":MeshDialect" ,
3309
3323
":MeshShardingInterfaceIncGen" ,
3310
3324
":Support" ,
3325
+ ":TensorDialect" ,
3326
+ "//llvm:Support" ,
3327
+ ],
3328
+ )
3329
+
3330
+ cc_library (
3331
+ name = "TensorShardingInterfaceImpl" ,
3332
+ srcs = ["lib/Dialect/Mesh/Interfaces/TensorShardingInterfaceImpl.cpp" ],
3333
+ hdrs = [
3334
+ "include/mlir/Dialect/Mesh/IR/TensorShardingInterfaceImpl.h" ,
3335
+ ],
3336
+ includes = ["include" ],
3337
+ deps = [
3338
+ ":DialectUtils" ,
3339
+ ":IR" ,
3340
+ ":MeshDialect" ,
3341
+ ":MeshShardingInterfaceIncGen" ,
3342
+ ":MeshShardingInterface" ,
3343
+ ":Support" ,
3344
+ ":TensorDialect" ,
3311
3345
"//llvm:Support" ,
3312
3346
],
3313
3347
)
@@ -4826,6 +4860,7 @@ cc_library(
4826
4860
":BuiltinToLLVMIRTranslation" ,
4827
4861
":ComplexToLLVM" ,
4828
4862
":ControlFlowToLLVM" ,
4863
+ ":DLTITransformOps" ,
4829
4864
":FuncExtensions" ,
4830
4865
":FuncToLLVM" ,
4831
4866
":FuncTransformOps" ,
@@ -8217,6 +8252,7 @@ cc_library(
8217
8252
":ControlFlowDialect" ,
8218
8253
":ConversionPassIncGen" ,
8219
8254
":IR" ,
8255
+ ":LLVMDialect" ,
8220
8256
":SCFDialect" ,
8221
8257
":SCFTransforms" ,
8222
8258
":TransformUtils" ,
@@ -9524,6 +9560,7 @@ cc_library(
9524
9560
":SparseTensorTransforms" ,
9525
9561
":TensorDialect" ,
9526
9562
":TensorInferTypeOpInterfaceImpl" ,
9563
+ ":TensorShardingInterfaceImpl" ,
9527
9564
":TensorTilingInterfaceImpl" ,
9528
9565
":TensorTransformOps" ,
9529
9566
":TensorTransforms" ,
@@ -11767,6 +11804,7 @@ cc_library(
11767
11804
":ArithDialect" ,
11768
11805
":ConversionPassIncGen" ,
11769
11806
":IR" ,
11807
+ ":LLVMDialect" ,
11770
11808
":MemRefDialect" ,
11771
11809
":Pass" ,
11772
11810
":SCFDialect" ,
@@ -11775,6 +11813,7 @@ cc_library(
11775
11813
":Transforms" ,
11776
11814
":VectorDialect" ,
11777
11815
":VectorTransforms" ,
11816
+ ":VectorUtils" ,
11778
11817
],
11779
11818
)
11780
11819
@@ -13988,6 +14027,55 @@ cc_library(
13988
14027
],
13989
14028
)
13990
14029
14030
+ td_library (
14031
+ name = "DLTITransformTdFiles" ,
14032
+ srcs = [
14033
+ "include/mlir/Dialect/DLTI/TransformOps/DLTITransformOps.td" ,
14034
+ ],
14035
+ includes = ["include" ],
14036
+ deps = [":TransformDialectTdFiles" ],
14037
+ )
14038
+
14039
+ gentbl_cc_library (
14040
+ name = "DLTITransformOpsIncGen" ,
14041
+ tbl_outs = [
14042
+ (
14043
+ [
14044
+ "-gen-op-decls" ,
14045
+ "-dialect=dlti" ,
14046
+ ],
14047
+ "include/mlir/Dialect/DLTI/TransformOps/DLTITransformOps.h.inc" ,
14048
+ ),
14049
+ (
14050
+ [
14051
+ "-gen-op-defs" ,
14052
+ "-dialect=dlti" ,
14053
+ ],
14054
+ "include/mlir/Dialect/DLTI/TransformOps/DLTITransformOps.cpp.inc" ,
14055
+ ),
14056
+ ],
14057
+ tblgen = ":mlir-tblgen" ,
14058
+ td_file = "include/mlir/Dialect/DLTI/TransformOps/DLTITransformOps.td" ,
14059
+ deps = [
14060
+ ":DLTITransformTdFiles" ,
14061
+ ],
14062
+ )
14063
+
14064
+ cc_library (
14065
+ name = "DLTITransformOps" ,
14066
+ srcs = glob (["lib/Dialect/DLTI/TransformOps/*.cpp" ]),
14067
+ hdrs = glob (["include/mlir/Dialect/DLTI/TransformOps/*.h" ]),
14068
+ includes = ["include" ],
14069
+ deps = [
14070
+ ":DataLayoutInterfaces" ,
14071
+ ":DLTIDialect" ,
14072
+ ":DLTITransformOpsIncGen" ,
14073
+ ":TransformDialect" ,
14074
+ ":TransformDialectInterfaces" ,
14075
+ ":TransformDialectUtils" ,
14076
+ ],
14077
+ )
14078
+
13991
14079
gentbl_cc_library (
13992
14080
name = "ReducerIncGen" ,
13993
14081
tbl_outs = [
0 commit comments