[mlir][bazel] Fix build.
authorOleg Shyshkov <shyshkov@google.com>
Wed, 19 Jul 2023 09:10:53 +0000 (11:10 +0200)
committerOleg Shyshkov <shyshkov@google.com>
Wed, 19 Jul 2023 09:11:32 +0000 (11:11 +0200)
utils/bazel/llvm-project-overlay/mlir/BUILD.bazel
utils/bazel/llvm-project-overlay/mlir/test/BUILD.bazel

index c252efc..e6b24a1 100644 (file)
@@ -9023,6 +9023,7 @@ td_library(
     srcs = [
         "include/mlir/Dialect/Linalg/IR/LinalgBase.td",
         "include/mlir/Dialect/Linalg/IR/LinalgEnums.td",
+        "include/mlir/Dialect/Linalg/IR/LinalgInterfaces.td",
         "include/mlir/Dialect/Linalg/IR/LinalgOps.td",
     ],
     includes = ["include"],
index ac811b2..7a2e1c9 100644 (file)
@@ -3,6 +3,7 @@
 # SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 
 load("@bazel_skylib//rules:expand_template.bzl", "expand_template")
+load("//mlir:build_defs.bzl", "if_cuda_available")
 load("//mlir:tblgen.bzl", "gentbl_cc_library", "td_library")
 load("//llvm:lit_test.bzl", "package_path")
 
@@ -593,7 +594,9 @@ cc_library(
 cc_library(
     name = "TestGPU",
     srcs = glob(["lib/Dialect/GPU/*.cpp"]),
-    defines = ["MLIR_CUDA_CONVERSIONS_ENABLED"],
+    defines = ["MLIR_CUDA_CONVERSIONS_ENABLED"] + if_cuda_available([
+        "MLIR_GPU_TO_CUBIN_PASS_ENABLE",
+    ]),
     includes = ["lib/Dialect/Test"],
     deps = [
         "//llvm:NVPTXCodeGen",