Updating stream caching (#614)

- Adding non-captured hipStream for use in setup
이 커밋은 다음에 포함됨:
gilbertlee-amd 작성하고 GitHub가 2022-09-09 16:30:15 -06:00를 커밋함
1 부모 65d78e9a1d
커밋 dd56135a9a
10 파일이 변경됨
+30 -34

검색과 일치하는 파일 없음

+2 -2
파일 보기
@@ -245,7 +245,7 @@ private:
printf("GDR Read %s\n", use_gdr_read ? "enabled" : "disabled");
if (use_gdr_read) {
NCCLCHECK(ncclCudaCalloc(&sendDevBuffer, sendBuffSize, 1));
NCCLCHECK(ncclCudaCalloc(&sendDevBuffer, sendBuffSize, nullptr, 1));
printf("Allocated sendDevBuffer %p of %d bytes, sliceSteps %d\n",
sendDevBuffer, sendBuffSize, sliceSteps);
}
@@ -397,7 +397,7 @@ private:
printf("GDR Write %s\n", use_gdr_write ? "enabled" : "disabled");
if (use_gdr_write) {
NCCLCHECK(ncclCudaCalloc(&recvDevBuffer, recvBuffSize, 1));
NCCLCHECK(ncclCudaCalloc(&recvDevBuffer, recvBuffSize, nullptr, 1));
printf("Allocated recvDevBuffer %p of %d bytes, sliceSteps %d\n",
recvDevBuffer, recvBuffSize, sliceSteps);
}