diff --git a/third_party/mlir/BUILD b/third_party/mlir/BUILD index 69d8c203849..f36745d920c 100644 --- a/third_party/mlir/BUILD +++ b/third_party/mlir/BUILD @@ -2081,6 +2081,7 @@ cc_library( hdrs = ["include/mlir/Conversion/GPUCommon/GPUCommonPass.h"], includes = ["include"], deps = [ + ":AsyncToLLVM", ":ConversionPassIncGen", ":GPUDialect", ":IR", @@ -3969,6 +3970,9 @@ cc_binary( srcs = ["tools/mlir-cuda-runner/mlir-cuda-runner.cpp"], deps = [ ":AllPassesAndDialectsNoRegistration", + ":Async", + ":AsyncToLLVM", + ":AsyncTransforms", ":ExecutionEngineUtils", ":GPUDialect", ":GPUToGPURuntimeTransforms",