From f6f9bcac38eaf21c363c4fef0b51f9246006a3e5 Mon Sep 17 00:00:00 2001
From: Fabien Servant <fabien.servant@interdigital.com>
Date: Fri, 6 Nov 2020 18:53:31 +0100
Subject: [PATCH] add gpu

---
 meshroom/submitters/slurmSubmitter.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/meshroom/submitters/slurmSubmitter.py b/meshroom/submitters/slurmSubmitter.py
index 86ebd1a9..6fe53f77 100644
--- a/meshroom/submitters/slurmSubmitter.py
+++ b/meshroom/submitters/slurmSubmitter.py
@@ -39,7 +39,6 @@ class SlurmSubmitter(BaseSubmitter):
 
         print(node.name)
         
-    
         command = [
             "sbatch",
             "--parsable",
@@ -52,6 +51,10 @@ class SlurmSubmitter(BaseSubmitter):
                 dependenciesString = "%s:%d" % (dependenciesString, dependency.pid)
             command.append(dependenciesString)
 
+        if node.nodeDesc.gpu.name != "NONE":
+            command.append("--partition=gpu")
+            command.append("--gres=gpu:1")
+
         countCores=24
         if node.isParallelized:
             blockSize, fullSize, nbBlocks = node.nodeDesc.parallelization.getSizes(node)