Skip to content

Commit

Permalink
PR #18296: [Rocm] fix arch
Browse files Browse the repository at this point in the history
Imported from GitHub PR #18296

Copybara import of the project:

--
d7396f0 by Ruturaj4 <ruturaj.vaidya@amd.com>:

[Rocm] fix arch

Merging this change closes #18296

COPYBARA_INTEGRATE_REVIEW=#18296 from ROCm:ci_arch_fix d7396f0
PiperOrigin-RevId: 688071318
  • Loading branch information
Ruturaj4 authored and Google-ML-Automation committed Oct 21, 2024
1 parent c5c29c1 commit 3bb521f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion third_party/tsl/third_party/gpus/rocm_configure.bzl
Original file line number Diff line number Diff line change
Expand Up @@ -572,7 +572,7 @@ def _tf_sysroot(repository_ctx):
return get_host_environ(repository_ctx, _TF_SYSROOT, "")

def _compute_rocm_extra_copts(repository_ctx, amdgpu_targets):
amdgpu_target_flags = ["--amdgpu-target=" +
amdgpu_target_flags = ["--offload-arch=" +
amdgpu_target for amdgpu_target in amdgpu_targets]
return str(amdgpu_target_flags)

Expand Down

0 comments on commit 3bb521f

Please sign in to comment.