3pp-mirror/Meshroom
1
0
Fork 0
mirror of https://github.com/alicevision/Meshroom.git synced 2025-06-01 18:31:58 +02:00
Code Issues Projects Releases Packages Wiki Activity

add gpu

Browse source
This commit is contained in:
Fabien Servant 2020-11-06 18:53:31 +01:00
parent d0fe3f4318
commit f6f9bcac38
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
meshroom/submitters/slurmSubmitter.py
View file

@ -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)