AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Updating stream caching (#614)

Browse Source 
- Adding non-captured hipStream for use in setup
This commit is contained in:
gilbertlee-amd
2022-09-09 16:30:15 -06:00
committed by GitHub
parent 65d78e9a1d
commit dd56135a9a
10 changed files with 30 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tools/ib-test/ib_test.cpp
+2 -2
View File
@@ -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);
}