device: optimize threadfence for ll64 protocol (#1858)

* device: optimize threadfence for ll64 protocol

* device: use __atomic_signal_fence()

---------

Co-authored-by: Nusrat Islam <nusislam@useocpslog-003.amd.com>

[ROCm/rccl commit: 6ade5065b4]
이 커밋은 다음에 포함됨:
Nusrat Islam그리고Nusrat Islam 작성하고 GitHub가 2025-08-18 09:16:41 -05:00를 커밋함
1 부모 377160e0c9
커밋 e4c025e5cd
1 파일이 변경됨
+4 -2
+4 -2
파일 보기
@@ -138,11 +138,13 @@ private:
if (recvConnHeadPtr) STORE(recvConnHeadPtr, recvConnHead += 1);
}
inline __device__ void postSend() {
__atomic_signal_fence(__ATOMIC_SEQ_CST);
asm volatile("s_waitcnt lgkmcnt(0) vmcnt(0)");
__atomic_signal_fence(__ATOMIC_SEQ_CST);
if (sendConnTailPtr) {
#if __CUDA_ARCH__ >= 900
__threadfence_system();
#else
__threadfence();
#endif
STORE((unsigned long long *)sendConnTailPtr, sendConnTail += 1);
}