From 73bd60b8431087bc4d527de3f40ae0178e23b52f Mon Sep 17 00:00:00 2001 From: Dmitri Gribenko Date: Fri, 20 May 2022 18:03:20 +0200 Subject: [PATCH] Adjust BUILD files for [MLIR][GPU] Add NvGpu mma.sync path to the VectorToGPU pass --- utils/bazel/llvm-project-overlay/mlir/BUILD.bazel | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel b/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel index 3bb09cb..bc250d1 100644 --- a/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel +++ b/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel @@ -7608,6 +7608,8 @@ cc_library( ":IR", ":LLVMDialect", ":MemRefDialect", + ":NVGPU", + ":NVVMDialect", ":Pass", ":SCFDialect", ":Support", @@ -8560,8 +8562,8 @@ cc_library( td_library( name = "MLProgramOpsTdFiles", srcs = [ - "include/mlir/Dialect/MLProgram/IR/MLProgramBase.td", "include/mlir/Dialect/MLProgram/IR/MLProgramAttributes.td", + "include/mlir/Dialect/MLProgram/IR/MLProgramBase.td", "include/mlir/Dialect/MLProgram/IR/MLProgramOps.td", ], includes = ["include"], -- 2.7.4