[bazel build] pull in cubin only when needed

Reviewed By: Peiming Differential Revision: https://reviews.llvm.org/D147826
llvm · Apr 8, 2023 · 7fff3bf · 7fff3bf
1 parent f733b4f
commit 7fff3bf
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel b/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel
@@ -2386,14 +2386,15 @@ cc_library(
         ":MemRefTransforms",
         ":NVVMDialect",
         ":Pass",
-        ":SerializeToCubin",
         ":SparseTensorDialect",
         ":SparseTensorTransforms",
         ":TensorTransforms",
         ":Transforms",
         ":VectorToLLVM",
         ":VectorTransforms",
-    ],
+    ] + if_cuda_available([
+        ":SerializeToCubin",
+    ]),
 )
 
 ##---------------------------------------------------------------------------##