[bazel] Port #151150: Move InitAll*** implementation into static library (#151183)

And prune deps when splitting

Author: rupprecht

utils/bazel/llvm-project-overlay/mlir/BUILD.bazel

@@ -4594,57 +4594,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.
@@ -8934,160 +8883,182 @@ 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",
         ":TensorTilingInterfaceImpl",
         ":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 +9070,8 @@ cc_binary(
         ":IR",
         ":MlirLspServerLib",
         ":MlirLspServerSupportLib",
+        ":RegisterAllDialects",
+        ":RegisterAllExtensions",
         "//mlir/test:TestAffine",
         "//mlir/test:TestAnalysis",
         "//mlir/test:TestArith",
@@ -9153,6 +9126,9 @@ cc_binary(
         ":OpenMPDialect",
         ":Pass",
         ":QuantOps",
+        ":RegisterAllDialects",
+        ":RegisterAllExtensions",
+        ":RegisterAllPasses",
         ":SCFToGPU",
         ":Support",
         ":Transforms",
@@ -9216,6 +9192,7 @@ cc_binary(
         ":ParseUtilities",
         ":Parser",
         ":Pass",
+        ":RegisterAllDialects",
         ":Support",
         ":TransformUtils",
         ":Transforms",
@@ -13228,6 +13205,8 @@ cc_binary(
         ":AllPassesAndDialects",
         ":IR",
         ":MlirReduceLib",
+        ":RegisterAllDialects",
+        ":RegisterAllPasses",
         "//mlir/test:TestDialect",
     ],
 )
@@ -13286,6 +13265,7 @@ cc_binary(
         ":IR",
         ":MlirQueryLib",
         ":MlirQueryUtils",
+        ":RegisterAllDialects",
         "//mlir/test:TestDialect",
     ],
 )

utils/bazel/llvm-project-overlay/mlir/unittests/BUILD.bazel

@@ -419,6 +419,7 @@ cc_test(
         "//mlir:Parser",
         "//mlir:Pass",
         "//mlir:ReconcileUnrealizedCasts",
+        "//mlir:RegisterAllDialects",
         "//mlir:ToLLVMIRTranslation",
         "//mlir:VectorToLLVM",
         "//mlir:VectorToSCF",