Skip to content

Commit 5a9abe4

Browse files
Resource request and limits
1 parent 2b6916f commit 5a9abe4

File tree

3 files changed

+21
-19
lines changed

3 files changed

+21
-19
lines changed

src/codeflare_sdk/templates/base-template.yaml

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -16,21 +16,21 @@ spec:
1616
custompodresources:
1717
- replicas: 1
1818
requests:
19-
cpu: 2
20-
memory: 8G
19+
cpu: 1
20+
memory: 4G
2121
nvidia.com/gpu: 0
2222
limits:
23-
cpu: 2
24-
memory: 8G
23+
cpu: 1
24+
memory: 4G
2525
nvidia.com/gpu: 0
2626
- replicas: 3
2727
requests:
28-
cpu: 2
29-
memory: 12G
28+
cpu: 1
29+
memory: 4G
3030
nvidia.com/gpu: 1
3131
limits:
32-
cpu: 2
33-
memory: 12G
32+
cpu: 1
33+
memory: 4G
3434
nvidia.com/gpu: 1
3535
generictemplate:
3636
# This config demonstrates KubeRay's Ray autoscaler integration.
Binary file not shown.

tests/e2e/mnist_raycluster_sdk.py

Lines changed: 13 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -9,17 +9,19 @@
99

1010
namespace = sys.argv[1]
1111

12-
cluster = Cluster(ClusterConfiguration(
13-
name='mnist',
14-
namespace=namespace,
15-
num_workers=1,
16-
min_cpus='500m',
17-
max_cpus=1,
18-
min_memory=0.5,
19-
max_memory=1,
20-
num_gpus=0,
21-
instascale=False,
22-
))
12+
cluster = Cluster(
13+
ClusterConfiguration(
14+
name="mnist",
15+
namespace=namespace,
16+
num_workers=1,
17+
min_cpus="500m",
18+
max_cpus=1,
19+
min_memory=0.5,
20+
max_memory=1,
21+
num_gpus=0,
22+
instascale=False,
23+
)
24+
)
2325

2426
cluster.up()
2527

0 commit comments

Comments
 (0)