1 files changed, 13 insertions, 0 deletions
diff --git a/third_party/gpus/rocm/build_defs.bzl.tpl b/third_party/gpus/rocm/build_defs.bzl.tpl
index 306f57551f..08c59f95a0 100644
--- a/third_party/gpus/rocm/build_defs.bzl.tpl
+++ b/third_party/gpus/rocm/build_defs.bzl.tpl
@@ -16,6 +16,19 @@ def rocm_default_copts():
     """Default options for all ROCm compilations."""
     return if_rocm(["-x", "rocm"] + %{rocm_extra_copts})
 
+def rocm_copts(opts = []):
+    """Gets the appropriate set of copts for (maybe) ROCm compilation.
+
+      If we're doing ROCm compilation, returns copts for our particular ROCm
+      compiler.  If we're not doing ROCm compilation, returns an empty list.
+
+      """
+    return rocm_default_copts() + select({
+        "//conditions:default": [],
+        "@local_config_rocm//rocm:using_hipcc": ([
+            "",
+        ]),
+    }) + if_rocm_is_configured(opts)
 
 def rocm_is_configured():
     """Returns true if ROCm was enabled during the configure process."""