Skip to content

Commit db7b240

Browse files
committed
fix(ncp-local): size fake GPUs for one-GPU workers
1 parent 65546d5 commit db7b240

3 files changed

Lines changed: 11 additions & 11 deletions

File tree

tools/ncp-local-cluster/apps/fake-gpu-operator/static/fake-gpu-operator.yaml

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ data:
6363
nodePoolLabelKey: run.ai/simulated-gpu-node-pool
6464
nodePools:
6565
default:
66-
gpuCount: 8
66+
gpuCount: 2
6767
gpuMemory: 81559
6868
gpuProduct: NVIDIA-H100-SXM4-80GB
6969
kind: ConfigMap
@@ -418,7 +418,7 @@ spec:
418418
metadata:
419419

420420
annotations:
421-
checksum/topology: 4e5af8859971722ac4a3aa64e8f38f255630bad818277aec4825b6c1d1f79cb1
421+
checksum/topology: d4293b3607dec297d688bfb0dfe5ffa78eb892015c926bf1d5b61500788468c4
422422
labels:
423423
app: device-plugin
424424
component: device-plugin
@@ -496,7 +496,7 @@ spec:
496496
template:
497497
metadata:
498498
annotations:
499-
checksum/topology: 4e5af8859971722ac4a3aa64e8f38f255630bad818277aec4825b6c1d1f79cb1
499+
checksum/topology: d4293b3607dec297d688bfb0dfe5ffa78eb892015c926bf1d5b61500788468c4
500500
labels:
501501
app: mig-faker
502502
component: mig-faker
@@ -652,7 +652,7 @@ kind: Deployment
652652
metadata:
653653
name: kwok-gpu-device-plugin
654654
annotations:
655-
checksum/topology: 4e5af8859971722ac4a3aa64e8f38f255630bad818277aec4825b6c1d1f79cb1
655+
checksum/topology: d4293b3607dec297d688bfb0dfe5ffa78eb892015c926bf1d5b61500788468c4
656656
labels:
657657
app: kwok-gpu-device-plugin
658658
spec:
@@ -664,7 +664,7 @@ spec:
664664
template:
665665
metadata:
666666
annotations:
667-
checksum/topology: 4e5af8859971722ac4a3aa64e8f38f255630bad818277aec4825b6c1d1f79cb1
667+
checksum/topology: d4293b3607dec297d688bfb0dfe5ffa78eb892015c926bf1d5b61500788468c4
668668
labels:
669669
app: kwok-gpu-device-plugin
670670
component: kwok-gpu-device-plugin
@@ -770,7 +770,7 @@ kind: Deployment
770770
metadata:
771771
name: status-updater
772772
annotations:
773-
checksum/topology: 4e5af8859971722ac4a3aa64e8f38f255630bad818277aec4825b6c1d1f79cb1
773+
checksum/topology: d4293b3607dec297d688bfb0dfe5ffa78eb892015c926bf1d5b61500788468c4
774774
labels:
775775
app: status-updater
776776
spec:
@@ -782,7 +782,7 @@ spec:
782782
template:
783783
metadata:
784784
annotations:
785-
checksum/topology: 4e5af8859971722ac4a3aa64e8f38f255630bad818277aec4825b6c1d1f79cb1
785+
checksum/topology: d4293b3607dec297d688bfb0dfe5ffa78eb892015c926bf1d5b61500788468c4
786786
labels:
787787
app: status-updater
788788
component: status-updater

tools/ncp-local-cluster/apps/fake-gpu-operator/values.yaml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,9 +6,9 @@ topology:
66
default:
77
gpuProduct: NVIDIA-H100-SXM4-80GB
88
# Keep a one-GPU local instance schedulable after NVCA infrastructure
9-
# overhead. A 128-way slice on k3d nodes is too small and suppresses
10-
# the required *_1x registration type.
11-
gpuCount: 8
9+
# overhead. Higher-density slices on k3d nodes suppress the required
10+
# *_1x registration type.
11+
gpuCount: 2
1212
gpuMemory: 81559
1313
nodePoolLabelKey: run.ai/simulated-gpu-node-pool
1414
migStrategy: mixed

tools/ncp-local-cluster/tests/test-multicluster-make.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,7 @@ assert_missing_docker_config_blocks_target start
8383
assert_missing_docker_config_blocks_target build-and-deploy-cluster
8484

8585
fake_gpu_count="$(yq -r '.topology.nodePools.default.gpuCount' "$ROOT_DIR/apps/fake-gpu-operator/values.yaml")"
86-
assert_eq "8" "$fake_gpu_count" "fake GPU count"
86+
assert_eq "2" "$fake_gpu_count" "fake GPU count"
8787

8888
print_directory_clusters="$(MAKEFLAGS=--print-directory; export MAKEFLAGS; run_make print-compute-clusters)"
8989
assert_eq "ncp-local-compute-1" "$print_directory_clusters" "print-directory compute cluster output"

0 commit comments

Comments
 (0)