optional nvrtc static

Author: wsmoses

common/BUILD

@@ -109,6 +109,20 @@ config_setting(
     },
 )
 
+
+# Flag for linking static CUDA NVRTC libs
+bool_flag(
+    name = "link_nvrtc_static_libs",
+    build_setting_default = False,
+)
+
+config_setting(
+    name = "is_nvrtc_static_linking_enabled",
+    flag_values = {
+        ":link_nvrtc_static_libs": "True",
+    },
+)
+
 #######################################################
 # Enable SYCL support flags
 

third_party/gpus/cuda/build_defs.bzl.tpl

@@ -36,6 +36,19 @@ def if_static_cuda(if_true, if_false = []):
         "//conditions:default": if_false,
     })
 
+# Macros for building NVRTC static code.
+def if_static_nvrtc(if_true, if_false = []):
+    """Shorthand for select()'ing on whether we're building with static NVRTC libs.
+
+    Returns a select statement which evaluates to if_true if we're building
+    with static NVRTC enabled.  Otherwise, the select statement evaluates to if_false.
+    """
+    return select({
+        "@rules_ml_toolchain//common:is_nvrtc_static_linking_enabled": if_true,
+        "//conditions:default": if_false,
+    })
+
+
 def if_cuda_clang(if_true, if_false = []):
    """Shorthand for select()'ing on wheteher we're building with cuda-clang.
 

third_party/gpus/cuda/hermetic/cuda_nvrtc.BUILD.tpl

@@ -44,7 +44,7 @@ cc_import(
 %{multiline_comment}
 cc_library(
     name = "nvrtc",
-    %{comment}deps = if_static_cuda([
+    %{comment}deps = if_static_nvrtc([
         %{comment}":nvrtc_static",
         %{comment}":nvrtc_builtins_static",
         %{comment}"@cuda_nvcc//:nvptxcompiler",