jupyter/spawner
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add SLURM gres extraction from resources

Browse source
This commit is contained in:
Benoît Seignovert 2024-02-20 11:16:53 +01:00
parent 21935f15f5
commit 05be86da07
Signed by: Benoît Seignovert
GPG key ID: F5D8895227D18A0B
2 changed files with 23 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

11
tests/test_slurm.py
View file

@ -8,6 +8,7 @@ from glicid_spawner.slurm import (
SlurmGpu,
SlurmNode,
SlurmPartition,
gres,
sinfo,
sinfo_filter,
sinfo_from_file,
@ -242,3 +243,13 @@ def test_slurm_sinfo_resources(monkeypatch):
assert isinstance(cnode, SlurmNode)
assert cnode == 'cnode001'
def test_slurm_gres():
"""Test SLURM GPU resources extraction."""
resources = sinfo_from_file(SINFO_FILE, with_states=('idle', 'idle~', 'mixed', 'allocated'))
gpus = gres(resources)
# Sorted and without duplicates
assert gpus == ['A100', 'A40', 'None', 'P100', 'T4']