Merge pull request #175 from wenkaidu/ll_abort

Fix abort handling in LL primitives

[ROCm/rccl commit: 9bf1edb554]
Этот коммит содержится в:
Wenkai Du
2020-02-25 14:12:33 -08:00
коммит произвёл GitHub
родитель 81af08cfde 5747c3cac1
Коммит a11aba0bfa
+1
Просмотреть файл
@@ -130,6 +130,7 @@ class ncclLLPrimitives {
asm volatile ("flat_load_dwordx4 %0, %1, glc\n"
"s_waitcnt vmcnt(0)\n"
"buffer_wbinvl1_vol\n" : "=v"(i4) : "v"(src));
if (checkAbort(i, 0)) break;
} while ((i4[1] != flag) || (i4[3] != flag));
uint64_t val64 = (uint64_t)(i4[0]) + (((uint64_t)i4[2]) << 32);
#else