3pp-mirror/Meshroom
1
0
Fork 0
mirror of https://github.com/alicevision/Meshroom.git synced 2025-06-08 13:51:57 +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) print(node.name)
command = [ command = [
"sbatch", "sbatch",
"--parsable", "--parsable",
@ -52,6 +51,10 @@ class SlurmSubmitter(BaseSubmitter):
dependenciesString = "%s:%d" % (dependenciesString, dependency.pid) dependenciesString = "%s:%d" % (dependenciesString, dependency.pid)
command.append(dependenciesString) command.append(dependenciesString)
if node.nodeDesc.gpu.name != "NONE":
command.append("--partition=gpu")
command.append("--gres=gpu:1")
countCores=24 countCores=24
if node.isParallelized: if node.isParallelized:
blockSize, fullSize, nbBlocks = node.nodeDesc.parallelization.getSizes(node) blockSize, fullSize, nbBlocks = node.nodeDesc.parallelization.getSizes(node)