Merge pull request #1079 from AI-Hypercomputer:yooh/gpu_coordinator_a…

…ddress PiperOrigin-RevId: 703142347
AI-Hypercomputer · Dec 5, 2024 · 1c09a2e · 1c09a2e
2 parents faa6131 + 6423d99
commit 1c09a2e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gpu_multi_process_run.sh b/gpu_multi_process_run.sh
@@ -124,7 +124,7 @@ resolve_coordinator_ip() {
   echo "Coordinator Address $JAX_COORDINATOR_ADDRESS"
 
   while [[ "$coordinator_found" = false && $lookup_attempt -le $max_coordinator_lookups ]]; do
-    coordinator_ip_address=$(nslookup "$JAX_COORDINATOR_ADDRESS" 2>/dev/null | awk '/^Address: / { print $2 }' | head -n 1)
+    coordinator_ip_address=$(nslookup "$JAX_COORDINATOR_ADDRESS" 2>/dev/null | awk '/Address: / { print $2 }' | head -n 1)
     if [[ -n "$coordinator_ip_address" ]]; then
       coordinator_found=true
       echo "Coordinator IP address: $coordinator_ip_address"