3pp-mirror/Meshroom
1
0
Fork 0
mirror of https://github.com/alicevision/Meshroom.git synced 2025-07-24 03:57:28 +02:00
Code Issues Projects Releases Packages Wiki Activity

Adding rangeBlocksCount to parallelization

Browse source
This commit is contained in:
Fabien Servant 2025-07-07 12:11:07 +02:00
parent b5eb9fdffa
commit 4066f3975c
1 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

8
meshroom/core/desc/computation.py
View file

@ -11,10 +11,11 @@ class Level(Enum):
class Range: class Range:
def __init__(self, iteration=0, blockSize=0, fullSize=0): def __init__(self, iteration=0, blockSize=0, fullSize=0, nbBlocks=0):
self.iteration = iteration self.iteration = iteration
self.blockSize = blockSize self.blockSize = blockSize
self.fullSize = fullSize self.fullSize = fullSize
self.nbBlocks = nbBlocks
@property @property
def start(self): def start(self):
@ -42,6 +43,7 @@ class Range:
"rangeBlockSize": self.blockSize, "rangeBlockSize": self.blockSize,
"rangeEffectiveBlockSize": self.effectiveBlockSize, "rangeEffectiveBlockSize": self.effectiveBlockSize,
"rangeFullSize": self.fullSize, "rangeFullSize": self.fullSize,
"rangeBlocksCount": self.nbBlocks
} }
@ -66,13 +68,13 @@ class Parallelization:
def getRange(self, node, iteration): def getRange(self, node, iteration):
blockSize, fullSize, nbBlocks = self.getSizes(node) blockSize, fullSize, nbBlocks = self.getSizes(node)
return Range(iteration=iteration, blockSize=blockSize, fullSize=fullSize) return Range(iteration=iteration, blockSize=blockSize, fullSize=fullSize, nbBlocks=nbBlocks)
def getRanges(self, node): def getRanges(self, node):
blockSize, fullSize, nbBlocks = self.getSizes(node) blockSize, fullSize, nbBlocks = self.getSizes(node)
ranges = [] ranges = []
for i in range(nbBlocks): for i in range(nbBlocks):
ranges.append(Range(iteration=i, blockSize=blockSize, fullSize=fullSize)) ranges.append(Range(iteration=i, blockSize=blockSize, fullSize=fullSize, nbBlocks=nbBlocks))
return ranges return ranges