Added gpus_from_gres gres/gpu tests (#754)

* Added gpus_from_gres gres/gpu tests * added another count gpus test
OSC · Apr 7, 2022 · 6fcc24c · 6fcc24c
1 parent b311447
commit 6fcc24c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/spec/job/adapters/slurm_spec.rb b/spec/job/adapters/slurm_spec.rb
@@ -1215,7 +1215,11 @@ def job_info(opts = {})
         ["gres:third-thing:sub-thing:17,gres:pfsdir:1,gres:gpu:v100-32g:2", 2],
         ["gres:gpu:v30-12g:2,gres:gpu:v31-32g:1", 3],
         ["gres:gpu:1", 1],
-        ["gres:pfsdir:ess", 0]
+        ["gres:pfsdir:ess", 0],
+        ["gres/gpu:v100-32g=2", 2],
+        ["gres/gpu:v100-32g=2,gres/gpu:v100-32g=4", 6],
+        ["gres/gpu:v100-32g=2,gres:gpu:1,gres/gpu:v100-32g=4", 7],
+        ["gres/gpu:v100-32g=2,gres:pfsdir:1", 2]
       ]
       gres_cases.each do |gc| 
         it "does not return the correct number of gpus when gres=\"#{gc[0]}\"" do