check for gpus per node arg

Author: luccabb

python/monarch/_src/job/slurm.py

@@ -137,7 +137,7 @@ def _submit_slurm_job(self, num_nodes: int) -> str:
             sbatch_directives.append(f"#SBATCH --partition={self._partition}")
 
         # add proportional cpu and memory args when not taking the full node
-        if not self._exclusive and self._partition:
+        if not self._exclusive and self._partition and self._gpus_per_node:
             gpus_per_task = self._gpus_per_node // self._ntasks_per_node
             slurm_args = clusterscope.job_gen_task_slurm(
                 partition=self._partition,