Skip to content

Commit f0b8e73

Browse files
committed
Fix logic error in mutexflow SPMM
Darn difference between device and non-device tasks, needs to be resolved asap! Signed-off-by: Joseph Schuchart <joseph.schuchart@stonybrook.edu>
1 parent 2e0c34c commit f0b8e73

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

examples/spmm/spmm.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -737,7 +737,7 @@ class SpMM25D {
737737

738738
// pass the running total to the next flow, if needed
739739
// otherwise write to the result flow
740-
if (enable_mutexflows_) {
740+
if (!enable_mutexflows_) {
741741
if (have_next_k) {
742742
co_await ttg::device::forward(ttg::device::send<1>(
743743
Key<3>({i, j, next_k}),
@@ -751,7 +751,7 @@ class SpMM25D {
751751
result));
752752
}
753753
} else {
754-
// no mutex flows, just send the result
754+
// mutex flows, just send the result
755755
co_await ttg::device::forward(ttg::device::send<1>(
756756
ijk, // reuse the key, it will be mapped to the same bucket
757757
std::move(C),

0 commit comments

Comments
 (0)