Fix merge error and replace inline asm (#838)

This commit is contained in:
Wenkai Du
2023-08-03 13:46:40 -07:00
committed by GitHub
parent c8085eb704
commit 60efe26549
+1 -5
View File
@@ -292,10 +292,6 @@ private:
needReload &= (0 == checkAbort(spins, i, 0));
} while (__any(needReload));
#pragma unroll
for (int u=0; u<ELEMS_PER_THREAD; u+=2)
load128(ptr+u*WARP_SIZE, vr[u], vr[u+1]);
#pragma unroll
for (int u=0; u<ELEMS_PER_THREAD; u+=2) {
v[u] = applyReduce(redOp, vr[u], v[u]);
@@ -314,7 +310,7 @@ private:
}
#if RCCL_USE_WBINVL1_VOL
if (tid == 0) __asm__ __volatile__("buffer_wbinvl1_vol");
if (tid == 0) __builtin_amdgcn_buffer_wbinvl1();
#endif
/************************ Send **************************/
if (SEND) {