diff --git a/third_party/mlir/BUILD b/third_party/mlir/BUILD index 5f3f0a4b99b..06e0ed8d4b4 100644 --- a/third_party/mlir/BUILD +++ b/third_party/mlir/BUILD @@ -756,6 +756,25 @@ cc_library( ], ) +cc_library( + name = "ShapeToSCF", + srcs = glob([ + "lib/Conversion/ShapeToSCF/*.cpp", + "lib/Conversion/ShapeToSCF/*.h", + ]) + ["lib/Conversion/PassDetail.h"], + hdrs = ["include/mlir/Conversion/ShapeToSCF/ShapeToSCF.h"], + includes = ["include"], + deps = [ + ":ConversionPassIncGen", + ":IR", + ":Pass", + ":SCFDialect", + ":Shape", + ":StandardOps", + ":Transforms", + ], +) + gentbl( name = "ShapeTransformsPassIncGen", strip_include_prefix = "include", @@ -2613,6 +2632,7 @@ cc_library( ":Parser", ":Pass", ":SCFTransforms", + ":ShapeToSCF", ":ShapeToStandard", ":ShapeTransforms", ":StandardOpsTransforms", @@ -2713,6 +2733,7 @@ cc_library( ":SPIRVPassIncGen", ":SPIRVToLLVM", ":Shape", + ":ShapeToSCF", ":ShapeToStandard", ":ShapeTransforms", ":ShapeTransformsPassIncGen", @@ -3282,6 +3303,7 @@ cc_library( ":LinalgTransforms", ":Pass", ":StandardOps", + ":StandardOpsTransforms", ":Support", ":Transforms", ":VectorToLLVM", diff --git a/third_party/mlir/test.BUILD b/third_party/mlir/test.BUILD index 23287ce28d6..14c2ba7778e 100644 --- a/third_party/mlir/test.BUILD +++ b/third_party/mlir/test.BUILD @@ -166,6 +166,7 @@ cc_library( "@llvm-project//mlir:Pass", "@llvm-project//mlir:SCFDialect", "@llvm-project//mlir:StandardOps", + "@llvm-project//mlir:StandardOpsTransforms", "@llvm-project//mlir:Support", "@llvm-project//mlir:TargetNVVMIR", "@llvm-project//mlir:TargetROCDLIR",