diff options
Diffstat (limited to 'utils/bazel/llvm-project-overlay/mlir/BUILD.bazel')
| -rw-r--r-- | utils/bazel/llvm-project-overlay/mlir/BUILD.bazel | 407 |
1 files changed, 204 insertions, 203 deletions
diff --git a/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel b/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel index 9ec7c51..f00c310 100644 --- a/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel +++ b/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel @@ -7,6 +7,7 @@ load("@bazel_skylib//rules:common_settings.bzl", "bool_flag") load("@bazel_skylib//rules:expand_template.bzl", "expand_template") +load("@rules_cc//cc:defs.bzl", "cc_binary", "cc_library") load( ":build_defs.bzl", "cc_headers_only", @@ -1003,6 +1004,9 @@ mlir_c_api_cc_library( ":BuiltinToLLVMIRTranslation", ":IR", ":LLVMToLLVMIRTranslation", + ":RegisterAllDialects", + ":RegisterAllExtensions", + ":RegisterAllPasses", ], ) @@ -1657,6 +1661,7 @@ cc_library( ":AMDGPUPassIncGen", ":AMDGPUUtils", ":AffineDialect", + ":AffineUtils", ":ArithDialect", ":ArithUtils", ":ControlFlowDialect", @@ -3056,14 +3061,14 @@ cc_library( ) ##---------------------------------------------------------------------------## -# Mesh Dialect +# Shard Dialect ##---------------------------------------------------------------------------## td_library( - name = "MeshTdFiles", + name = "ShardTdFiles", srcs = [ - "include/mlir/Dialect/Mesh/IR/MeshBase.td", - "include/mlir/Dialect/Mesh/IR/MeshOps.td", + "include/mlir/Dialect/Shard/IR/ShardBase.td", + "include/mlir/Dialect/Shard/IR/ShardOps.td", ], includes = ["include"], deps = [ @@ -3075,92 +3080,92 @@ td_library( ) gentbl_cc_library( - name = "MeshIncGen", + name = "ShardIncGen", tbl_outs = { - "include/mlir/Dialect/Mesh/IR/MeshOps.h.inc": [ + "include/mlir/Dialect/Shard/IR/ShardOps.h.inc": [ "-gen-op-decls", - "-dialect=mesh", + "-dialect=shard", ], - "include/mlir/Dialect/Mesh/IR/MeshOps.cpp.inc": [ + "include/mlir/Dialect/Shard/IR/ShardOps.cpp.inc": [ "-gen-op-defs", - "-dialect=mesh", + "-dialect=shard", ], - "include/mlir/Dialect/Mesh/IR/MeshDialect.h.inc": [ + "include/mlir/Dialect/Shard/IR/ShardDialect.h.inc": [ "-gen-dialect-decls", - "-dialect=mesh", + "-dialect=shard", ], - "include/mlir/Dialect/Mesh/IR/MeshDialect.cpp.inc": [ + "include/mlir/Dialect/Shard/IR/ShardDialect.cpp.inc": [ "-gen-dialect-defs", - "-dialect=mesh", + "-dialect=shard", ], - "include/mlir/Dialect/Mesh/IR/MeshEnums.h.inc": [ + "include/mlir/Dialect/Shard/IR/ShardEnums.h.inc": [ "-gen-enum-decls", - "-dialect=mesh", + "-dialect=shard", ], - "include/mlir/Dialect/Mesh/IR/MeshEnums.cpp.inc": [ + "include/mlir/Dialect/Shard/IR/ShardEnums.cpp.inc": [ "-gen-enum-defs", - "-dialect=mesh", + "-dialect=shard", ], - "include/mlir/Dialect/Mesh/IR/MeshAttributes.h.inc": [ + "include/mlir/Dialect/Shard/IR/ShardAttributes.h.inc": [ "-gen-attrdef-decls", - "-dialect=mesh", + "-dialect=shard", ], - "include/mlir/Dialect/Mesh/IR/MeshAttributes.cpp.inc": [ + "include/mlir/Dialect/Shard/IR/ShardAttributes.cpp.inc": [ "-gen-attrdef-defs", - "-dialect=mesh", + "-dialect=shard", ], - "include/mlir/Dialect/Mesh/IR/MeshTypes.h.inc": [ + "include/mlir/Dialect/Shard/IR/ShardTypes.h.inc": [ "-gen-typedef-decls", - "-typedefs-dialect=mesh", + "-typedefs-dialect=shard", ], - "include/mlir/Dialect/Mesh/IR/MeshTypes.cpp.inc": [ + "include/mlir/Dialect/Shard/IR/ShardTypes.cpp.inc": [ "-gen-typedef-defs", - "-typedefs-dialect=mesh", + "-typedefs-dialect=shard", ], }, tblgen = ":mlir-tblgen", - td_file = "include/mlir/Dialect/Mesh/IR/MeshOps.td", + td_file = "include/mlir/Dialect/Shard/IR/ShardOps.td", deps = [ - ":MeshTdFiles", ":ShapeOpsTdFiles", + ":ShardTdFiles", ], ) gentbl_cc_library( - name = "MeshShardingInterfaceIncGen", + name = "ShardingInterfaceIncGen", tbl_outs = { - "include/mlir/Dialect/Mesh/Interfaces/ShardingInterface.h.inc": ["-gen-op-interface-decls"], - "include/mlir/Dialect/Mesh/Interfaces/ShardingInterface.cpp.inc": ["-gen-op-interface-defs"], + "include/mlir/Dialect/Shard/Interfaces/ShardingInterface.h.inc": ["-gen-op-interface-decls"], + "include/mlir/Dialect/Shard/Interfaces/ShardingInterface.cpp.inc": ["-gen-op-interface-defs"], }, tblgen = ":mlir-tblgen", - td_file = "include/mlir/Dialect/Mesh/Interfaces/ShardingInterface.td", + td_file = "include/mlir/Dialect/Shard/Interfaces/ShardingInterface.td", deps = [":OpBaseTdFiles"], ) cc_library( - name = "MeshShardingInterface", - srcs = ["lib/Dialect/Mesh/Interfaces/ShardingInterface.cpp"], + name = "ShardingInterface", + srcs = ["lib/Dialect/Shard/Interfaces/ShardingInterface.cpp"], hdrs = [ - "include/mlir/Dialect/Mesh/Interfaces/ShardingInterface.h", - "include/mlir/Dialect/Mesh/Interfaces/ShardingInterfaceImpl.h", + "include/mlir/Dialect/Shard/Interfaces/ShardingInterface.h", + "include/mlir/Dialect/Shard/Interfaces/ShardingInterfaceImpl.h", ], includes = ["include"], deps = [ ":DialectUtils", ":IR", - ":MeshDialect", - ":MeshShardingInterfaceIncGen", + ":ShardDialect", + ":ShardingInterfaceIncGen", ":Support", "//llvm:Support", ], ) cc_library( - name = "MeshDialect", - srcs = ["lib/Dialect/Mesh/IR/MeshOps.cpp"], + name = "ShardDialect", + srcs = ["lib/Dialect/Shard/IR/ShardOps.cpp"], hdrs = [ - "include/mlir/Dialect/Mesh/IR/MeshDialect.h", - "include/mlir/Dialect/Mesh/IR/MeshOps.h", + "include/mlir/Dialect/Shard/IR/ShardDialect.h", + "include/mlir/Dialect/Shard/IR/ShardOps.h", ], includes = ["include"], deps = [ @@ -3171,7 +3176,7 @@ cc_library( ":IR", ":InferTypeOpInterface", ":InliningUtils", - ":MeshIncGen", + ":ShardIncGen", ":SideEffectInterfaces", ":Support", ":ViewLikeInterface", @@ -3180,23 +3185,23 @@ cc_library( ) gentbl_cc_library( - name = "MeshTransformsPassIncGen", - tbl_outs = {"include/mlir/Dialect/Mesh/Transforms/Passes.h.inc": [ + name = "ShardTransformsPassIncGen", + tbl_outs = {"include/mlir/Dialect/Shard/Transforms/Passes.h.inc": [ "-gen-pass-decls", - "-name=Mesh", + "-name=Shard", ]}, tblgen = ":mlir-tblgen", - td_file = "include/mlir/Dialect/Mesh/Transforms/Passes.td", + td_file = "include/mlir/Dialect/Shard/Transforms/Passes.td", deps = [":PassBaseTdFiles"], ) cc_library( - name = "MeshTransforms", + name = "ShardTransforms", srcs = glob([ - "lib/Dialect/Mesh/Transforms/*.cpp", - "lib/Dialect/Mesh/Transforms/*.h", + "lib/Dialect/Shard/Transforms/*.cpp", + "lib/Dialect/Shard/Transforms/*.h", ]), - hdrs = glob(["include/mlir/Dialect/Mesh/Transforms/*.h"]), + hdrs = glob(["include/mlir/Dialect/Shard/Transforms/*.h"]), includes = ["include"], deps = [ ":AffineDialect", @@ -3209,10 +3214,10 @@ cc_library( ":FuncDialect", ":FunctionInterfaces", ":IR", - ":MeshDialect", - ":MeshShardingInterface", - ":MeshTransformsPassIncGen", ":Pass", + ":ShardDialect", + ":ShardTransformsPassIncGen", + ":ShardingInterface", ":Support", ":TensorDialect", ":Transforms", @@ -3221,11 +3226,11 @@ cc_library( ) cc_library( - name = "MeshToMPIConversion", + name = "ShardToMPIConversion", srcs = glob([ - "lib/Conversion/MeshToMPI/*.cpp", + "lib/Conversion/ShardToMPI/*.cpp", ]), - hdrs = glob(["include/mlir/Conversion/MeshToMPI/*.h"]), + hdrs = glob(["include/mlir/Conversion/ShardToMPI/*.h"]), includes = ["include"], deps = [ ":AffineDialect", @@ -3240,10 +3245,10 @@ cc_library( ":LinalgDialect", ":MPIDialect", ":MemRefDialect", - ":MeshDialect", - ":MeshTransforms", ":Pass", ":SCFDialect", + ":ShardDialect", + ":ShardTransforms", ":Support", ":TensorDialect", ":TransformUtils", @@ -3988,7 +3993,6 @@ cc_library( ":MemRefToEmitC", ":MemRefToLLVM", ":MemRefToSPIRV", - ":MeshToMPIConversion", ":NVGPUToNVVM", ":NVVMToLLVM", ":OpenACCToSCF", @@ -4002,6 +4006,7 @@ cc_library( ":SCFToSPIRV", ":SPIRVToLLVM", ":ShapeToStandard", + ":ShardToMPIConversion", ":TensorToLinalg", ":TensorToSPIRV", ":TosaToArith", @@ -4522,7 +4527,7 @@ cc_library( ":FuncDialect", ":IR", ":InliningUtils", - ":MeshShardingInterface", + ":ShardingInterface", ], ) @@ -4593,57 +4598,6 @@ cc_library( cc_library( name = "AllExtensions", hdrs = ["include/mlir/InitAllExtensions.h"], - deps = [ - ":AMXTransforms", - ":AffineTransformOps", - ":ArithToEmitC", - ":ArithToLLVM", - ":ArmNeonTransformOps", - ":ArmSVEVectorTransformOps", - ":BufferizationTransformOps", - ":BuiltinToLLVMIRTranslation", - ":ComplexToLLVM", - ":ControlFlowToLLVM", - ":DLTITransformOps", - ":FuncExtensions", - ":FuncToEmitC", - ":FuncToLLVM", - ":FuncTransformOps", - ":GPUToGPURuntimeTransforms", - ":GPUToLLVMIRTranslation", - ":GPUToNVVMTransforms", - ":GPUTransformOps", - ":IndexToLLVM", - ":LLVMToLLVMIRTranslation", - ":LinalgTransformOps", - ":MPIToLLVM", - ":MathToLLVM", - ":MemRefToEmitC", - ":MemRefToLLVM", - ":MemRefTransformOps", - ":MeshDialect", - ":NVGPUTransformOps", - ":NVVMTarget", - ":NVVMToLLVM", - ":OpenMPToLLVM", - ":ROCDLTarget", - ":ROCDLToLLVMIRTranslation", - ":SCFToEmitC", - ":SCFTransformOps", - ":SparseTensorTransformOps", - ":TensorExtensions", - ":TensorTransformOps", - ":TransformDebugExtension", - ":TransformIRDLExtension", - ":TransformLoopExtension", - ":TransformPDLExtension", - ":TransformTuneExtension", - ":UBToLLVM", - ":VCIXToLLVMIRTranslation", - ":VectorToLLVM", - ":VectorTransformOps", - ":XeVMToLLVM", - ], ) # TODO(zinenko): remove this after updating users. @@ -5118,10 +5072,7 @@ cc_library( ":SideEffectInterfaces", ":Support", ":ViewLikeInterface", - "//llvm:AsmParser", "//llvm:BinaryFormat", - "//llvm:BitReader", - "//llvm:BitWriter", "//llvm:Core", "//llvm:Support", ], @@ -7194,7 +7145,7 @@ cc_library( includes = ["include"], deps = [ ":IR", - ":MeshShardingInterface", + ":ShardingInterface", ":TensorDialect", "//llvm:Support", ], @@ -8718,6 +8669,23 @@ cc_library( ) cc_library( + name = "XeVMToLLVMIRTranslation", + srcs = glob(["lib/Target/LLVMIR/Dialect/XeVM/*.cpp"]), + hdrs = glob(["include/mlir/Target/LLVMIR/Dialect/XeVM/*.h"]), + includes = ["include"], + deps = [ + ":DialectUtils", + ":IR", + ":LLVMDialect", + ":Support", + ":ToLLVMIRTranslation", + ":XeVMDialect", + "//llvm:Core", + "//llvm:Support", + ], +) + +cc_library( name = "AllToLLVMIRTranslations", hdrs = ["include/mlir/Target/LLVMIR/Dialect/All.h"], includes = ["include"], @@ -8736,6 +8704,7 @@ cc_library( ":ROCDLToLLVMIRTranslation", ":SPIRVToLLVMIRTranslation", ":VCIXToLLVMIRTranslation", + ":XeVMToLLVMIRTranslation", ], ) @@ -8936,130 +8905,69 @@ cc_library( "include/mlir/InitAllDialects.h", "include/mlir/InitAllPasses.h", ], +) + +cc_library( + name = "RegisterAllDialects", + srcs = ["lib/RegisterAllDialects.cpp"], deps = [ ":AMDGPUDialect", - ":AMDGPUToROCDL", - ":AMDGPUTransforms", ":AMXDialect", - ":AMXTransforms", ":AffineDialect", - ":AffinePassIncGen", - ":AffineToStandard", - ":AffineTransformOps", - ":AffineTransforms", + ":AllPassesAndDialects", ":ArithDialect", - ":ArithToAMDGPU", - ":ArithToLLVM", - ":ArithToSPIRV", ":ArithTransforms", ":ArithValueBoundsOpInterfaceImpl", ":ArmNeonDialect", ":ArmSMEDialect", ":ArmSMETransforms", ":ArmSVEDialect", - ":ArmSVETransforms", ":AsyncDialect", - ":AsyncPassIncGen", - ":AsyncToLLVM", - ":AsyncTransforms", ":BufferizationDialect", - ":BufferizationPipelines", - ":BufferizationTransformOps", ":BufferizationTransforms", ":CastInterfaces", ":ComplexDialect", - ":ComplexToLLVM", - ":ComplexToLibm", - ":ComplexToSPIRV", ":ControlFlowDialect", ":ControlFlowTransforms", - ":ConversionPasses", - ":ConvertToEmitC", - ":ConvertToLLVM", ":DLTIDialect", ":EmitCDialect", - ":EmitCTransforms", ":FuncDialect", - ":FuncToLLVM", - ":FuncToSPIRV", - ":FuncTransformOps", - ":FuncTransforms", - ":FuncTransformsPassIncGen", ":GPUDialect", - ":GPUPassIncGen", - ":GPUPipelines", - ":GPUToGPURuntimeTransforms", - ":GPUToNVVMTransforms", - ":GPUToROCDLTransforms", - ":GPUToSPIRV", - ":GPUTransformOps", ":GPUTransforms", ":IR", ":IRDLDialect", ":IndexDialect", ":LLVMDialect", ":LLVMIRTransforms", - ":LLVMPassIncGen", ":LinalgDialect", - ":LinalgPassIncGen", - ":LinalgToStandard", - ":LinalgTransformOps", ":LinalgTransforms", ":MLProgramDialect", ":MLProgramTransforms", ":MPIDialect", ":MathDialect", - ":MathToFuncs", - ":MathToLLVM", - ":MathToLibm", - ":MathToSPIRV", - ":MathTransforms", ":MemRefDialect", - ":MemRefToLLVM", - ":MemRefToSPIRV", - ":MemRefTransformOps", ":MemRefTransforms", - ":MeshDialect", - ":MeshTransforms", ":NVGPUDialect", - ":NVGPUPassIncGen", - ":NVGPUToNVVM", - ":NVGPUTransformOps", - ":NVGPUTransforms", ":NVVMDialect", ":NVVMTarget", ":OpenACCDialect", - ":OpenACCTransforms", ":OpenMPDialect", - ":OpenMPToLLVM", ":PDLDialect", ":PDLInterpDialect", - ":PDLToPDLInterp", ":PtrDialect", ":QuantOps", - ":QuantTransforms", ":ROCDLDialect", ":ROCDLTarget", - ":ReconcileUnrealizedCasts", ":SCFDialect", - ":SCFPassIncGen", - ":SCFToControlFlow", - ":SCFToGPU", ":SCFTransformOps", ":SCFTransforms", ":SMTDialect", ":SPIRVDialect", - ":SPIRVPassIncGen", ":SPIRVTarget", - ":SPIRVToLLVM", - ":SPIRVTransforms", ":ShapeDialect", - ":ShapeToStandard", ":ShapeTransforms", - ":ShapeTransformsPassIncGen", + ":ShardDialect", ":SparseTensorDialect", - ":SparseTensorPipelines", - ":SparseTensorTransformOps", ":SparseTensorTransforms", ":TensorDialect", ":TensorInferTypeOpInterfaceImpl", @@ -9067,29 +8975,112 @@ cc_library( ":TensorTransformOps", ":TensorTransforms", ":TosaDialect", - ":TosaToLinalg", - ":TransformDebugExtension", ":TransformDialect", ":TransformDialectTransforms", - ":TransformLoopExtension", ":TransformPDLExtension", - ":Transforms", - ":TransformsPassIncGen", ":UBDialect", ":VectorDialect", - ":VectorToLLVM", - ":VectorToSCF", - ":VectorToSPIRV", - ":VectorTransformOps", ":VectorTransforms", ":X86VectorDialect", - ":X86VectorTransforms", ":XeGPUDialect", - ":XeGPUTransforms", ":XeVMDialect", ], ) +cc_library( + name = "RegisterAllPasses", + srcs = ["lib/RegisterAllPasses.cpp"], + deps = [ + ":AMDGPUTransforms", + ":AffineTransforms", + ":AllPassesAndDialects", + ":ArithTransforms", + ":ArmSMETransforms", + ":ArmSVETransforms", + ":AsyncTransforms", + ":BufferizationPipelines", + ":BufferizationTransforms", + ":ConversionPasses", + ":EmitCTransforms", + ":FuncTransforms", + ":GPUPipelines", + ":GPUTransforms", + ":LLVMIRTransforms", + ":LinalgTransforms", + ":MLProgramTransforms", + ":MathTransforms", + ":MemRefTransforms", + ":NVGPUTransforms", + ":OpenACCTransforms", + ":QuantTransforms", + ":SCFTransforms", + ":SPIRVTransforms", + ":ShapeTransforms", + ":ShardTransforms", + ":SparseTensorPipelines", + ":SparseTensorTransforms", + ":TensorTransforms", + ":TosaDialect", + ":TransformDialectTransforms", + ":Transforms", + ":VectorTransforms", + ":XeGPUTransforms", + ], +) + +cc_library( + name = "RegisterAllExtensions", + srcs = ["lib/RegisterAllExtensions.cpp"], + deps = [ + ":AMXTransforms", + ":AffineTransformOps", + ":AllExtensions", + ":ArithToEmitC", + ":ArithToLLVM", + ":ArmNeonTransformOps", + ":ArmSVEVectorTransformOps", + ":BufferizationTransformOps", + ":BuiltinToLLVMIRTranslation", + ":ComplexToLLVM", + ":ControlFlowToLLVM", + ":DLTITransformOps", + ":FuncExtensions", + ":FuncToEmitC", + ":FuncToLLVM", + ":FuncTransformOps", + ":GPUToGPURuntimeTransforms", + ":GPUToLLVMIRTranslation", + ":GPUToNVVMTransforms", + ":GPUTransformOps", + ":IndexToLLVM", + ":LLVMToLLVMIRTranslation", + ":LinalgTransformOps", + ":MPIToLLVM", + ":MathToLLVM", + ":MemRefToEmitC", + ":MemRefToLLVM", + ":MemRefTransformOps", + ":NVGPUTransformOps", + ":NVVMToLLVM", + ":OpenMPToLLVM", + ":ROCDLToLLVMIRTranslation", + ":SCFToEmitC", + ":SCFTransformOps", + ":SparseTensorTransformOps", + ":TensorExtensions", + ":TensorTransformOps", + ":TransformDebugExtension", + ":TransformIRDLExtension", + ":TransformLoopExtension", + ":TransformPDLExtension", + ":TransformTuneExtension", + ":UBToLLVM", + ":VectorToLLVM", + ":VectorTransformOps", + ":XeVMToLLVM", + ], +) + cc_binary( name = "mlir-lsp-server", srcs = ["tools/mlir-lsp-server/mlir-lsp-server.cpp"], @@ -9101,6 +9092,8 @@ cc_binary( ":IR", ":MlirLspServerLib", ":MlirLspServerSupportLib", + ":RegisterAllDialects", + ":RegisterAllExtensions", "//mlir/test:TestAffine", "//mlir/test:TestAnalysis", "//mlir/test:TestArith", @@ -9120,7 +9113,6 @@ cc_binary( "//mlir/test:TestMath", "//mlir/test:TestMathToVCIX", "//mlir/test:TestMemRef", - "//mlir/test:TestMesh", "//mlir/test:TestNVGPU", "//mlir/test:TestPDLL", "//mlir/test:TestPass", @@ -9129,6 +9121,7 @@ cc_binary( "//mlir/test:TestSCF", "//mlir/test:TestSPIRV", "//mlir/test:TestShapeDialect", + "//mlir/test:TestShard", "//mlir/test:TestTensor", "//mlir/test:TestTestDynDialect", "//mlir/test:TestTilingInterface", @@ -9155,6 +9148,9 @@ cc_binary( ":OpenMPDialect", ":Pass", ":QuantOps", + ":RegisterAllDialects", + ":RegisterAllExtensions", + ":RegisterAllPasses", ":SCFToGPU", ":Support", ":Transforms", @@ -9182,7 +9178,6 @@ cc_binary( "//mlir/test:TestMathToVCIX", "//mlir/test:TestMemRef", "//mlir/test:TestMemRefToLLVMWithTransforms", - "//mlir/test:TestMesh", "//mlir/test:TestNVGPU", "//mlir/test:TestPDLL", "//mlir/test:TestPass", @@ -9191,6 +9186,7 @@ cc_binary( "//mlir/test:TestSCF", "//mlir/test:TestSPIRV", "//mlir/test:TestShapeDialect", + "//mlir/test:TestShard", "//mlir/test:TestTensor", "//mlir/test:TestTestDynDialect", "//mlir/test:TestTilingInterface", @@ -9218,6 +9214,7 @@ cc_binary( ":ParseUtilities", ":Parser", ":Pass", + ":RegisterAllDialects", ":Support", ":TransformUtils", ":Transforms", @@ -10548,9 +10545,9 @@ cc_library( ":LinalgStructuredOpsIncGen", ":MathDialect", ":MemRefDialect", - ":MeshShardingInterface", ":Parser", ":SCFDialect", + ":ShardingInterface", ":SideEffectInterfaces", ":SparseTensorDialect", ":SubsetOpInterface", @@ -10699,14 +10696,14 @@ cc_library( ":MathDialect", ":MemRefDialect", ":MemRefTransforms", - ":MeshDialect", - ":MeshShardingInterface", - ":MeshTransforms", ":Pass", ":RuntimeVerifiableOpInterface", ":SCFDialect", ":SCFTransforms", ":SCFUtils", + ":ShardDialect", + ":ShardTransforms", + ":ShardingInterface", ":SparseTensorDialect", ":SubsetOpInterface", ":Support", @@ -10988,6 +10985,7 @@ cc_library( ":VectorTransforms", ":X86VectorDialect", ":X86VectorTransforms", + "//llvm:Core", "//llvm:Support", ], ) @@ -11198,10 +11196,10 @@ cc_library( ":InferTypeOpInterface", ":InliningUtils", ":LoopLikeInterface", - ":MeshDialect", - ":MeshShardingInterface", ":Pass", ":QuantOps", + ":ShardDialect", + ":ShardingInterface", ":SideEffectInterfaces", ":Support", ":TensorDialect", @@ -12141,8 +12139,8 @@ cc_library( ":FuncTransforms", ":IR", ":MemRefDialect", - ":MeshShardingInterface", ":Pass", + ":ShardingInterface", ":SideEffectInterfaces", ":TensorDialect", ":TransformUtils", @@ -13229,6 +13227,8 @@ cc_binary( ":AllPassesAndDialects", ":IR", ":MlirReduceLib", + ":RegisterAllDialects", + ":RegisterAllPasses", "//mlir/test:TestDialect", ], ) @@ -13287,6 +13287,7 @@ cc_binary( ":IR", ":MlirQueryLib", ":MlirQueryUtils", + ":RegisterAllDialects", "//mlir/test:TestDialect", ], ) |