AI/rocm-systems
2
0
Forka 0
Codice Problemi Pull Request Actions Pacchetti Progetti Rilasci Wiki Attività

Updating to use hipDeviceMallocUncached (#95)

Sfoglia il codice sorgente 
Use hipDeviceMallocUncached instead of hipDeviceMallocFinegrained on newer ROCm versions.
This commit is contained in:
saurabhAMD
2025-01-11 19:23:04 -06:00
committato da nileshnegi
parent f2a48983ae
commit fc9917e0da
1 ha cambiato i file con 10 aggiunte e 3 eliminazioni
Scarica il file Patch Scarica il file Diff Expand all files Collapse all files
src/common.cu
+10 -3
Vedi File
@@ -880,9 +880,16 @@ testResult_t AllocateBuffs(void **sendbuff, size_t sendBytes, void **recvbuff, s
nbytes = nbytes + cache_bytes;
}
if (memorytype == ncclFine) {
CUDACHECK(hipExtMallocWithFlags(sendbuff, nbytes, hipDeviceMallocFinegrained));
CUDACHECK(hipExtMallocWithFlags(recvbuff, nbytes, hipDeviceMallocFinegrained));
if (datacheck) CUDACHECK(hipExtMallocWithFlags(expected, recvBytes, hipDeviceMallocFinegrained));
if(HIP_VERSION >= 50700000) {
CUDACHECK(hipExtMallocWithFlags(sendbuff, nbytes, hipDeviceMallocUncached));
CUDACHECK(hipExtMallocWithFlags(recvbuff, nbytes, hipDeviceMallocUncached));
if (datacheck) CUDACHECK(hipExtMallocWithFlags(expected, recvBytes, hipDeviceMallocUncached));
}
else {
CUDACHECK(hipExtMallocWithFlags(sendbuff, nbytes, hipDeviceMallocFinegrained));
CUDACHECK(hipExtMallocWithFlags(recvbuff, nbytes, hipDeviceMallocFinegrained));
if (datacheck) CUDACHECK(hipExtMallocWithFlags(expected, recvBytes, hipDeviceMallocFinegrained));
}
}
else if (memorytype == ncclHost) {
CUDACHECK(hipHostMalloc(sendbuff, nbytes));