Merge pull request #2803 from ROCm/r2.18-rocm-enhanced-spack

change label to find_rocm_config.py
ROCm · Jan 22, 2025 · 7bb83d5 · 7bb83d5
2 parents 9a0f384 + 33c9329
commit 7bb83d5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/third_party/gpus/rocm_configure.bzl b/third_party/gpus/rocm_configure.bzl
@@ -892,7 +892,7 @@ remote_rocm_configure = repository_rule(
     attrs = {
         "environ": attr.string_dict(),
         "_find_rocm_config": attr.label(
-            default = Label("@local_tsl//third_party/gpus:find_rocm_config.py"),
+            default = Label("@org_tensorflow//third_party/gpus:find_rocm_config.py"),
         ),
     },
 )
@@ -902,7 +902,7 @@ rocm_configure = repository_rule(
     environ = _ENVIRONS + [_TF_ROCM_CONFIG_REPO],
     attrs = {
         "_find_rocm_config": attr.label(
-            default = Label("@local_tsl//third_party/gpus:find_rocm_config.py"),
+            default = Label("@org_tensorflow//third_party/gpus:find_rocm_config.py"),
         ),
     },
 )