Index: utils/bazel/llvm-project-overlay/mlir/BUILD.bazel =================================================================== --- utils/bazel/llvm-project-overlay/mlir/BUILD.bazel +++ utils/bazel/llvm-project-overlay/mlir/BUILD.bazel @@ -2391,6 +2391,21 @@ ], ) +cc_library( + name = "NVGPUUtils", + srcs = ["lib/Dialect/NVGPU/Utils/MMAUtils.cpp"], + hdrs = ["include/mlir/Dialect/NVGPU/Utils/MMAUtils.h"], + includes = ["include"], + deps = [ + ":AffineDialect", + ":ArithDialect", + ":IR", + ":NVGPUDialect", + ":NVVMDialect", + ":VectorDialect", + ], +) + cc_library( name = "NVGPUTransforms", srcs = [ @@ -8695,8 +8710,6 @@ srcs = glob([ "lib/Conversion/VectorToGPU/*.cpp", "lib/Conversion/VectorToGPU/*.h", - "lib/Dialect/NVGPU/Utils/*.cpp", - "include/mlir/Dialect/NVGPU/Utils/*.h", ]), hdrs = glob([ "include/mlir/Conversion/VectorToGPU/*.h", @@ -8715,6 +8728,7 @@ ":LLVMDialect", ":MemRefDialect", ":NVGPUDialect", + ":NVGPUUtils", ":NVVMDialect", ":Pass", ":SCFDialect",