gstykalin/test-dags
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update dags

Browse Source
This commit is contained in:
George Stykalin
2026-01-26 15:49:45 +03:00
parent 4049bddbef
commit 2edf537df6
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
dags/connection-check.py
View File

@@ -2,7 +2,7 @@ from airflow import DAG
from airflow.decorators import task
import pendulum
TARGET_HOST = "airflow-k8s-demo-0164-worker-gpu-0.airflow-k8s-demo-0164-worker-gpu"
TARGET_HOST = "airflow-k8s-demo-0168-worker-gpu-0.airflow-k8s-demo-0168-worker-gpu"
with DAG(
dag_id='debug_gpu_network_check',

2
dags/test-train-pytorch.py
View File

@@ -7,7 +7,7 @@ import json
# --- CONFIGURATION ---
WORLD_SIZE = 2
MASTER_ADDR = "airflow-k8s-demo-0167-worker-gpu-0.airflow-k8s-demo-0167-worker-gpu"
MASTER_ADDR = "airflow-k8s-demo-0168-worker-gpu-0.airflow-k8s-demo-0168-worker-gpu"
MASTER_PORT = "29500"
NCCL_TIMEOUT = 1800