Add wckey parameter for Slurm

diff --git a/src/Core/Constants.cxx b/src/Core/Constants.cxx
index d2d0e74d270eb3598478bfc4781c4faa485f60c4..607b53606c9744b544f441a46ce0ebba76bc2563 100644 (file)
--- a/src/Core/Constants.cxx
+++ b/src/Core/Constants.cxx
@@ -50,6 +50,7 @@ namespace Batch {
   def_Constant(WORKDIR);
   def_Constant(EXCLUSIVE);
   def_Constant(MEMPERCPU);
+  def_Constant(WCKEY);
 
   // Parameters for COORM
   def_Constant(LAUNCHER_FILE);

diff --git a/src/Core/Constants.hxx b/src/Core/Constants.hxx
index 8a080984f18fe3cfc809c5cd351db9cf17780a23..cf2b0cac5f7f4e320aa320c95053a2df0c207bd1 100644 (file)
--- a/src/Core/Constants.hxx
+++ b/src/Core/Constants.hxx
@@ -61,6 +61,7 @@ namespace Batch {
   decl_extern_Constant(WORKDIR);
   decl_extern_Constant(EXCLUSIVE);
   decl_extern_Constant(MEMPERCPU);
+  decl_extern_Constant(WCKEY);
 
   // Parameters for COORM
   decl_extern_Constant(LAUNCHER_FILE);

diff --git a/src/Core/ParameterTypeMap.cxx b/src/Core/ParameterTypeMap.cxx
index c20ad7177571e99537a623582e2986f5ad0a64a1..60c14348553dcc83ae6b70e3526bf4d0bca56127 100644 (file)
--- a/src/Core/ParameterTypeMap.cxx
+++ b/src/Core/ParameterTypeMap.cxx
@@ -63,6 +63,7 @@ namespace Batch {
     addParameter("WORKDIR", STRING, 1);
     addParameter("EXCLUSIVE", BOOL, 1);
     addParameter("MEMPERCPU", LONG, 1);
+    addParameter("WCKEY", STRING, 1);
 
        // Parameters for COORM
     addParameter("LAUNCHER_FILE", STRING, 1);

diff --git a/src/Slurm/BatchManager_Slurm.cxx b/src/Slurm/BatchManager_Slurm.cxx
index 9d62defc038c2774872b6589f6ceec54569df347..0fe20d03ba36424b8bb6c48efc5d9d9e37c4abd8 100644 (file)
--- a/src/Slurm/BatchManager_Slurm.cxx
+++ b/src/Slurm/BatchManager_Slurm.cxx
@@ -152,6 +152,8 @@ namespace Batch {
       tempOutputFile << "#SBATCH --mem-per-cpu=" << params[MEMPERCPU] << endl;
     if (params.find(QUEUE) != params.end())
       tempOutputFile << "#SBATCH --partition=" << params[QUEUE] << endl;
+    if (params.find(WCKEY) != params.end())
+      tempOutputFile << "#SBATCH --wckey=" << params[WCKEY] << endl;
 
     // Define environment for the job
     Environnement env = job.getEnvironnement();