Skip to content

Commit 49973ee

Browse files
Resource request and limits
1 parent 441d68f commit 49973ee

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
@@ -18,21 +18,21 @@ spec:
1818
custompodresources:
1919
- replicas: 1
2020
requests:
21-
cpu: 2
22-
memory: 8G
21+
cpu: 1
22+
memory: 4G
2323
nvidia.com/gpu: 0
2424
limits:
25-
cpu: 2
26-
memory: 8G
25+
cpu: 1
26+
memory: 4G
2727
nvidia.com/gpu: 0
2828
- replicas: 3
2929
requests:
30-
cpu: 2
31-
memory: 12G
30+
cpu: 1
31+
memory: 4G
3232
nvidia.com/gpu: 1
3333
limits:
34-
cpu: 2
35-
memory: 12G
34+
cpu: 1
35+
memory: 4G
3636
nvidia.com/gpu: 1
3737
generictemplate:
3838
# 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)