Updating stream caching (#614)

- Adding non-captured hipStream for use in setup
このコミットが含まれているのは:
gilbertlee-amd authored and GitHub committed 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);
}