diff options
Diffstat (limited to 'utils/bazel/llvm-project-overlay/mlir/BUILD.bazel')
-rw-r--r-- | utils/bazel/llvm-project-overlay/mlir/BUILD.bazel | 260 |
1 files changed, 131 insertions, 129 deletions
diff --git a/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel b/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel index 178c3ce..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", ], ) @@ -4594,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", - ":NVGPUTransformOps", - ":NVVMTarget", - ":NVVMToLLVM", - ":OpenMPToLLVM", - ":ROCDLTarget", - ":ROCDLToLLVMIRTranslation", - ":SCFToEmitC", - ":SCFTransformOps", - ":ShardDialect", - ":SparseTensorTransformOps", - ":TensorExtensions", - ":TensorTransformOps", - ":TransformDebugExtension", - ":TransformIRDLExtension", - ":TransformLoopExtension", - ":TransformPDLExtension", - ":TransformTuneExtension", - ":UBToLLVM", - ":VCIXToLLVMIRTranslation", - ":VectorToLLVM", - ":VectorTransformOps", - ":XeVMToLLVM", - ], ) # TODO(zinenko): remove this after updating users. @@ -8716,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"], @@ -8734,6 +8704,7 @@ cc_library( ":ROCDLToLLVMIRTranslation", ":SPIRVToLLVMIRTranslation", ":VCIXToLLVMIRTranslation", + ":XeVMToLLVMIRTranslation", ], ) @@ -8934,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", ":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", - ":ShardTransforms", ":SparseTensorDialect", - ":SparseTensorPipelines", - ":SparseTensorTransformOps", ":SparseTensorTransforms", ":TensorDialect", ":TensorInferTypeOpInterfaceImpl", @@ -9065,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"], @@ -9099,6 +9092,8 @@ cc_binary( ":IR", ":MlirLspServerLib", ":MlirLspServerSupportLib", + ":RegisterAllDialects", + ":RegisterAllExtensions", "//mlir/test:TestAffine", "//mlir/test:TestAnalysis", "//mlir/test:TestArith", @@ -9153,6 +9148,9 @@ cc_binary( ":OpenMPDialect", ":Pass", ":QuantOps", + ":RegisterAllDialects", + ":RegisterAllExtensions", + ":RegisterAllPasses", ":SCFToGPU", ":Support", ":Transforms", @@ -9216,6 +9214,7 @@ cc_binary( ":ParseUtilities", ":Parser", ":Pass", + ":RegisterAllDialects", ":Support", ":TransformUtils", ":Transforms", @@ -13228,6 +13227,8 @@ cc_binary( ":AllPassesAndDialects", ":IR", ":MlirReduceLib", + ":RegisterAllDialects", + ":RegisterAllPasses", "//mlir/test:TestDialect", ], ) @@ -13286,6 +13287,7 @@ cc_binary( ":IR", ":MlirQueryLib", ":MlirQueryUtils", + ":RegisterAllDialects", "//mlir/test:TestDialect", ], ) |