gpu: test not providing shape file

Signed-off-by: vsoch <[email protected]>

Author: vsoch

fluxbind/scripts/run_mapping.sh

@@ -21,7 +21,7 @@ if [ -z "$JOB_SHAPE_FILE" ]; then
     exit 1
 fi
 
-gpus_per_task=${GPUS_PER_TASK:-1}
+gpus_per_task=${GPUS_PER_TASK:-0}
 
 # Call the fluxbind helper script to get the target location string (e.g., "core:0" or "UNBOUND")
 # It ALWAYS returns a single line in the format: BIND_LOCATION,CUDA_DEVICE_ID
@@ -42,6 +42,8 @@ CUDA_DEVICES="${BIND_INFO#*;}"
 
 if [[ "$CUDA_DEVICES" != "NONE" ]]; then
     export CUDA_VISIBLE_DEVICES=$CUDA_DEVICES
+    export ROCR_VISIBLE_DEVICES=$CUDA_DEVICES
+    export HIP_VISIBLE_DEVICES=$CUDA_DEVICES
 fi
 
 if [[ "${BIND_LOCATION}" == "UNBOUND" ]]; then

fluxbind/shape/shape.py

@@ -19,7 +19,7 @@ def __init__(self, filepath, machine="machine:0"):
         """
         self.machine = machine
         self.machine_cpuset = commands.hwloc_calc.get_cpuset(self.machine)
-        self.data = self.load_file(filepath)
+        self.data = self.load_file(filepath) or []
         # This discovers and cache hardware properties on init
         # The expectation is that this is running from the node (task)
         self.num_cores = commands.hwloc_calc.count("core", within=self.machine)
@@ -74,11 +74,12 @@ def discover_gpus(self):
 
         return gpus_by_numa
 
-    def load_file(self, filepath):
+    def load_file(self, filepath=None):
         """
         Loads and parses the YAML shape file.
         """
-        return utils.read_yaml(filepath)
+        if filepath is not None:
+            return utils.read_yaml(filepath)
 
     @staticmethod
     def parse_range(range_str: str) -> set: