diff --git a/dags/connection-check-default.py b/dags/connection-check-default.py index 596ad13..fbfe0f4 100644 --- a/dags/connection-check-default.py +++ b/dags/connection-check-default.py @@ -2,7 +2,7 @@ from airflow import DAG from airflow.decorators import task import pendulum -TARGET_HOST = "airflow-k8s-demo-0163-worker-default-0.airflow-k8s-demo-0163-worker-default" +TARGET_HOST = "airflow-k8s-demo-0164-worker-default-0.airflow-k8s-demo-0164-worker-default" with DAG( dag_id='debug_network_check', diff --git a/dags/connection-check.py b/dags/connection-check.py index 09d3660..e5d4a63 100644 --- a/dags/connection-check.py +++ b/dags/connection-check.py @@ -2,7 +2,7 @@ from airflow import DAG from airflow.decorators import task import pendulum -TARGET_HOST = "airflow-k8s-demo-0163-worker-gpu-0.airflow-k8s-demo-0163-worker-gpu" +TARGET_HOST = "airflow-k8s-demo-0164-worker-gpu-0.airflow-k8s-demo-0164-worker-gpu" with DAG( dag_id='debug_gpu_network_check', diff --git a/dags/test-train-pytorch.py b/dags/test-train-pytorch.py index 15619ee..5c73d3e 100644 --- a/dags/test-train-pytorch.py +++ b/dags/test-train-pytorch.py @@ -7,7 +7,7 @@ import json # --- CONFIGURATION --- WORLD_SIZE = 2 -MASTER_ADDR = "airflow-k8s-demo-0163-worker-gpu-0.airflow-k8s-demo-0163-worker-gpu" +MASTER_ADDR = "airflow-k8s-demo-0164-worker-gpu-0.airflow-k8s-demo-0164-worker-gpu" MASTER_PORT = "29500" NCCL_TIMEOUT = 1800