Skip to content

Commit

Permalink
Fix compiler detection
Browse files Browse the repository at this point in the history
  • Loading branch information
alekstheod committed Dec 2, 2024
1 parent 86d1742 commit c4672b6
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 1 deletion.
1 change: 1 addition & 0 deletions third_party/tsl/third_party/gpus/rocm_configure.bzl
Original file line number Diff line number Diff line change
Expand Up @@ -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)),
Expand Down
4 changes: 3 additions & 1 deletion xla/service/gpu/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -2518,7 +2518,9 @@ gpu_kernel_library(
deps = [
"//xla:shape_util",
"//xla:types",
],
] + if_rocm_is_configured([
"@local_config_rocm//rocm:rocm_headers"
]),
)

xla_test(
Expand Down

0 comments on commit c4672b6

Please sign in to comment.