msccl: fix data corruption with MTYPE_RW (#1014)

2023-12-11 20:33:15 -08:00
parent fca459baaf
commit 81602814a7
1 changed files with 1 additions and 1 deletions
@@ -460,7 +460,7 @@ __device__ __forceinline__ void mscclRunInterpreter(
          return;
      }
      if (t->hasDependence && tid == nthreads-1)
-        __atomic_store_n(&mscclFlags[bid].flag, (uint64_t) COMPUTE_FLAG(workIndex, iter, step), __ATOMIC_RELAXED);
+        __atomic_store_n(&mscclFlags[bid].flag, (uint64_t) COMPUTE_FLAG(workIndex, iter, step), t->type == MSCCL_REDUCE ? __ATOMIC_RELEASE : __ATOMIC_RELAXED);
      step++;
    }
  }