limit number of openmp threads

lizhenjie2015 · Jul 21, 2015 · b54a847 · b54a847
1 parent 6ca2980
commit b54a847
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ann_benchmarks/__init__.py b/ann_benchmarks/__init__.py
@@ -12,6 +12,7 @@
     print('resetting memory limit from', soft, 'to', memory_limit)
     resource.setrlimit(resource.RLIMIT_DATA, (memory_limit, hard))
 
+os.environ['OMP_THREAD_LIMIT'] = '1' # just to limit number of processors                                                                                                                                                                    
 
 class BaseANN(object):
     pass