hotfix #127

proper handling of array size < nthreads
MatthewRHermes · Nov 13, 2024 · 2274fa1 · 2274fa1
1 parent 01c24c4
commit 2274fa1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/my_pyscf/lassi/op_o1/utilities.py b/my_pyscf/lassi/op_o1/utilities.py
@@ -133,6 +133,7 @@ def get_contig_blks (mask):
 
 def split_contig_array (arrlen, nthreads):
     '''Divide a contiguous array into chunks to be handled by each thread'''
+    nthreads = min (arrlen, nthreads)
     blklen, rem = divmod (arrlen, nthreads);
     blklen = np.array ([blklen,]*nthreads)
     blklen[:rem] += 1