Skip to content

Commit 847cd21

Browse files
committed
Fix thread indexing in contract_blocks! to account for interactive nthreads
1 parent 2c7552d commit 847cd21

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

NDTensors/src/blocksparse/contract_folds.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ function contract_blocks!(
1515
Folds.foreach(eachnzblock(boffs1).values, ThreadedEx()) do block1
1616
for block2 in eachnzblock(boffs2)
1717
maybe_contract_blocks!(
18-
contraction_plans[threadid()],
18+
contraction_plans[threadid() - nthreads(:interactive)],
1919
block1,
2020
block2,
2121
labels1_to_labels2,
@@ -29,7 +29,7 @@ function contract_blocks!(
2929
Folds.foreach(eachnzblock(boffs2).values, ThreadedEx()) do block2
3030
for block1 in eachnzblock(boffs1)
3131
maybe_contract_blocks!(
32-
contraction_plans[threadid()],
32+
contraction_plans[threadid() - nthreads(:interactive)],
3333
block1,
3434
block2,
3535
labels1_to_labels2,

0 commit comments

Comments
 (0)