diff --git a/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel b/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel --- a/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel +++ b/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel @@ -2187,6 +2187,7 @@ cc_library( name = "NVGPUTransforms", srcs = [ + "lib/Dialect/NVGPU/Transforms/MmaSyncTF32Transforms.cpp", "lib/Dialect/NVGPU/Transforms/OptimizeSharedMemory.cpp", "lib/Dialect/NVGPU/Transforms/PassDetail.h", ], @@ -6344,6 +6345,7 @@ "//mlir/test:TestLinalg", "//mlir/test:TestMath", "//mlir/test:TestMemRef", + "//mlir/test:TestNVGPU", "//mlir/test:TestPDLL", "//mlir/test:TestPass", "//mlir/test:TestReducer", diff --git a/utils/bazel/llvm-project-overlay/mlir/test/BUILD.bazel b/utils/bazel/llvm-project-overlay/mlir/test/BUILD.bazel --- a/utils/bazel/llvm-project-overlay/mlir/test/BUILD.bazel +++ b/utils/bazel/llvm-project-overlay/mlir/test/BUILD.bazel @@ -520,6 +520,27 @@ ], ) +cc_library( + name = "TestNVGPU", + srcs = glob(["lib/Dialect/NVGPU/*.cpp"]), + defines = ["MLIR_CUDA_CONVERSIONS_ENABLED"], + includes = ["lib/Dialect/Test"], + deps = [ + "//mlir:AffineDialect", + "//mlir:Analysis", + "//mlir:FuncDialect", + "//mlir:GPUDialect", + "//mlir:IR", + "//mlir:LLVMDialect", + "//mlir:MemRefDialect", + "//mlir:NVGPUDialect", + "//mlir:NVGPUTransforms", + "//mlir:Pass", + "//mlir:SCFDialect", + "//mlir:TransformUtils", + ], +) + cc_library( name = "TestSCF", srcs = glob(["lib/Dialect/SCF/*.cpp"]),