Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

DRAFT : Reservation with groups #168

Draft
wants to merge 6 commits into
base: main
Choose a base branch
from

Conversation

jeffnvidia
Copy link
Contributor

Summary

This PR is intended to enable reservation with group node selection (previously, reservation only worked when specifying node names)
This PR is based on this one which cleaned the slurm_system file : #167

Test Plan

Tested with
extra_srun_args = "--reservation reservation_name" in system.toml

test scenario :
name = "gpt_175B"

[Tests]
[Tests.1]
name = "gpt_175B"
nodes = ['cluster:SU2:2', 'cluster:SU3:3', 'cluster:SU4:3']

$ python ./cloudaix.py --mode run --system-config ... --test-templates-dir conf/v0.6/general/test_template --tests-dir conf/v0.6/general/test --test-scenario conf/v0.6/general/test_scenario/gpt/gpt_175B.toml

...
Section Name: Tests.1
Test Name: gpt_175B
Description: gpt
No dependencies
[INFO] Initializing Runner
[INFO] Creating SlurmRunner
[INFO] Starting test scenario execution.
[INFO] Starting test: Tests.1
[INFO] Running test: Tests.1

@jeffnvidia jeffnvidia force-pushed the reservation_with_groups branch 2 times, most recently from 6d680f7 to e16ca71 Compare August 8, 2024 13:13
@srinivas212 srinivas212 added the enhancement New feature or request label Aug 11, 2024
@jeffnvidia jeffnvidia changed the title Reservation with groups DRAFT : Reservation with groups Aug 11, 2024
@TaekyungHeo TaekyungHeo added feature and removed enhancement New feature or request labels Aug 29, 2024
@TaekyungHeo TaekyungHeo marked this pull request as draft September 5, 2024 20:31
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants