KJ135
/
ISCE_INSAR
mirror of http://172.16.0.12:5000/WBJY/ISCE_INSAR.git
4
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

topsStack/topo: use OMP_NUM_THREADS if available 

for the parallel processing with multiprocessing.Pool()
LT1AB
Zhang Yunjun 2020-09-02 12:33:13 -07:00 committed by piyushrpt
parent 6a02f79277
commit f06d22e02c
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
contrib/stack/topsStack/topo.py
View File

@ -97,7 +97,12 @@ def main(iargs=None):
for ind in range(reference.numberOfBursts): for ind in range(reference.numberOfBursts):
inputs.append((dirname, demImage, reference, ind)) inputs.append((dirname, demImage, reference, ind))
pool = mp.Pool(mp.cpu_count()) # parallel processing
numThread = int(os.environ.get('OMP_NUM_THREADS', mp.cpu_count()))
numThread = min(numThread, mp.cpu_count())
print('running in parallel with {} processes'.format(numThread))
pool = mp.Pool(numThread)
results = pool.map(call_topo, inputs) results = pool.map(call_topo, inputs)
pool.close() pool.close()