Skip to content

Commit 2a8ed6b

Browse files
committed
change region and device type
1 parent 13d0a37 commit 2a8ed6b

1 file changed

Lines changed: 8 additions & 8 deletions

File tree

.github/workflows/integration_gpu_cluster_create.yaml

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ jobs:
2727
group: nightly-test-cluster-group-gpu
2828
cancel-in-progress: false
2929
env:
30-
GPU_CLUSTER_NAME: nightly-xpk-b200
30+
GPU_CLUSTER_NAME: nightly-xpk-h100
3131
WORKLOAD_NAME: xpktest-gpu-nightly-${{ github.run_attempt }}
3232
steps:
3333
- uses: actions/download-artifact@v4
@@ -55,21 +55,21 @@ jobs:
5555
5656
# 4. Set Env Var for the host (GitHub Runner)
5757
echo "GOOGLE_APPLICATION_CREDENTIALS=$HOME/.config/gcloud/application_default_credentials.json" >> $GITHUB_ENV
58-
- name: Create an XPK Cluster with 1 x b200 GPU
59-
run: xpk cluster create --cluster $GPU_CLUSTER_NAME --device-type=b200-8 --zone=us-east7-b --default-pool-cpu-machine-type=e2-standard-8 --spot
58+
- name: Create an XPK Cluster with 1 x h100 GPU
59+
run: xpk cluster create --cluster $GPU_CLUSTER_NAME --device-type=h100-mega-80gb-8 --zone=asia-southeast1-b --default-pool-cpu-machine-type=e2-standard-8 --spot
6060
- name: Authenticate Docker
6161
run: gcloud auth configure-docker --quiet
6262
- name: Run a base-docker-image workload
63-
run: xpk workload create --cluster $GPU_CLUSTER_NAME --workload $WORKLOAD_NAME --docker-image='nvidia/cuda:12.1.0-base-ubuntu22.04' --command "nvidia-smi" --zone=us-east7-b --device-type=b200-8
63+
run: xpk workload create --cluster $GPU_CLUSTER_NAME --workload $WORKLOAD_NAME --docker-image='nvidia/cuda:12.1.0-base-ubuntu22.04' --command "nvidia-smi" --zone=asia-southeast1-b --device-type=h100-mega-80gb-8
6464
- name: List out the workloads on the cluster
65-
run: xpk workload list --cluster $GPU_CLUSTER_NAME --zone=us-east7-b
65+
run: xpk workload list --cluster $GPU_CLUSTER_NAME --zone=asia-southeast1-b
6666
- name: Wait for workload completion and confirm it succeeded
67-
run: xpk workload list --cluster $GPU_CLUSTER_NAME --zone=us-east7-b --wait-for-job-completion $WORKLOAD_NAME --timeout 600
67+
run: xpk workload list --cluster $GPU_CLUSTER_NAME --zone=asia-southeast1-b --wait-for-job-completion $WORKLOAD_NAME --timeout 600
6868
- name: Delete the workload on the cluster
69-
run: xpk workload delete --workload $WORKLOAD_NAME --cluster $GPU_CLUSTER_NAME --zone=us-east7-b
69+
run: xpk workload delete --workload $WORKLOAD_NAME --cluster $GPU_CLUSTER_NAME --zone=asia-southeast1-b
7070
- name: Delete the cluster created
7171
if: always()
72-
run: xpk cluster delete --cluster $GPU_CLUSTER_NAME --zone=us-east7-b --force
72+
run: xpk cluster delete --cluster $GPU_CLUSTER_NAME --zone=asia-southeast1-b --force
7373
- name: Upload cluster nodepool creation log
7474
if: always()
7575
uses: actions/upload-artifact@v4

0 commit comments

Comments
 (0)