From 24d6cd946f73c9af636bc3717485c80c7167901f Mon Sep 17 00:00:00 2001
From: Zaccharie Ramzi <zaccharie.ramzi@gmail.com>
Date: Thu, 18 Aug 2022 16:38:19 +0100
Subject: [PATCH] used only cpus per task in base config (#11)

---
 README.md                                                      | 2 +-
 .../jz_hydra_submitit_launcher/hydra/launcher/base.yaml        | 3 +--
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index a830e2d..6382a82 100644
--- a/README.md
+++ b/README.md
@@ -42,7 +42,7 @@ The parameters you can override are defined in the `hydra-submitit-launcher` plu
 
 For example, if you want to use the gpu_p2 partition, you would need to do:
 ```
-hydra-submitit-launch my_app.py dev hydra.launcher.setup=null hydra.launcher.partition=gpu_p2
+hydra-submitit-launch my_app.py dev hydra.launcher.setup=null hydra.launcher.partition=gpu_p2 hydra.launcher.cpus_per_task=3
 ```
 
 In order to change the timeout on the SLURM job to for example 10 hours, you would need to do:
diff --git a/hydra_plugins/jz_hydra_submitit_launcher/hydra/launcher/base.yaml b/hydra_plugins/jz_hydra_submitit_launcher/hydra/launcher/base.yaml
index 89a7ed7..9f67032 100644
--- a/hydra_plugins/jz_hydra_submitit_launcher/hydra/launcher/base.yaml
+++ b/hydra_plugins/jz_hydra_submitit_launcher/hydra/launcher/base.yaml
@@ -6,8 +6,7 @@ gpus_per_node: 1
 tasks_per_node: 1
 gres: "gpu:${hydra.launcher.gpus_per_node}"
 qos: ${qos_from_hours:${hours}}
-cpus_per_gpu: 10
-cpus_per_task: ${cpu_from_gpu:${hydra.launcher.gpus_per_node},${hydra.launcher.cpus_per_gpu}}
+cpus_per_task: ${cpu_from_gpu:${hydra.launcher.gpus_per_node},10}
 gpus_per_task: ${hydra.launcher.gpus_per_node}
 additional_parameters:
   account: ${oc.env:IDRPROJ}@gpu