Fix compiler detection

openxla · Dec 2, 2024 · 3002fbc · 3002fbc
1 parent 86d1742
commit 3002fbc
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/third_party/tsl/third_party/gpus/rocm_configure.bzl b/third_party/tsl/third_party/gpus/rocm_configure.bzl
@@ -756,6 +756,7 @@ def _create_local_rocm_repository(repository_ctx):
         tpl_paths["crosstool:clang/bin/crosstool_wrapper_driver_rocm"],
         {
             "%{cpu_compiler}": str(cc),
+            "%{compiler}": rocm_defines["%{compiler}"],
             "%{hipcc_path}": str(repository_ctx.path(rocm_config.rocm_toolkit_path + "/bin/hipcc")),
             "%{hipcc_env}": _hipcc_env(repository_ctx),
             "%{rocm_path}": str(repository_ctx.path(rocm_config.rocm_toolkit_path)),

diff --git a/xla/service/gpu/BUILD b/xla/service/gpu/BUILD
@@ -2518,7 +2518,9 @@ gpu_kernel_library(
     deps = [
         "//xla:shape_util",
         "//xla:types",
-    ],
+    ] + if_rocm_is_configured([
+        "@config_local_rocm//rocm:rocm_headers"
+    ]),
 )
 
 xla_test(