PR #18296: [Rocm] fix arch

Imported from GitHub PR openxla/xla#18296 Copybara import of the project: -- d7396f02c524568bdf342ba3fae1860df45d8219 by Ruturaj4 <[email protected]>: [Rocm] fix arch Merging this change closes #18296 PiperOrigin-RevId: 688071318
google · Oct 21, 2024 · 05e85cf · 05e85cf
1 parent 8d98460
commit 05e85cf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/third_party/gpus/rocm_configure.bzl b/third_party/gpus/rocm_configure.bzl
@@ -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)