ClusterCockpit/cc-condor-sync
1
0
Fork 0
mirror of https://gitlab.cs.uni-saarland.de/hpc/cc-condor-sync.git synced 2025-07-27 15:16:09 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix AssignedGPUs parsing.

Browse Source
This commit is contained in:
Joachim Meyer
2022-12-16 13:42:04 +01:00
parent 80619b6154
commit 6128b58cbd
2 changed files with 13 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/condor_status_to_gpu_map.py
View File

@@ -17,10 +17,14 @@ if __name__ == "__main__":
slot_gpu_map = {}
for slot in condor_status:
machine = slot["Machine"]
machine = slot["Machine"].split('.')[0]
gpu_map = {}
if machine in slot_gpu_map:
gpu_map = slot_gpu_map[machine]
else:
slot_gpu_map[machine] = {}
if not "AssignedGPUs" in slot:
continue
gpus = slot["AssignedGPUs"].split(',')
for gpu_id in gpus:
gpu = slot["GPUs_" + gpu_id.strip().replace("-", "_")]