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

[perf]hipMalloc perf optimization

Sfoglia il codice sorgente 
Change-Id: I9cd1d29205cb0c9938d26bb8e336fc9fe8a262ed
This commit is contained in:
Sarbojit Sarkar
2020-08-24 13:19:49 -04:00
committato da Sarbojit Sarkar
parent 8034467ec3
commit a3b44b5f75
1 ha cambiato i file con 10 aggiunte e 0 eliminazioni
Scarica il file Patch Scarica il file Diff Expand all files Collapse all files
hipamd/rocclr/hip_peer.cpp
+10
Vedi File
@@ -185,6 +185,11 @@ hipError_t hipDeviceDisablePeerAccess(int peerDeviceId) {
if ((hipSuccess != canAccessPeer(&canAccess, deviceId, peerDeviceId)) || (canAccess == 0)) {
HIP_RETURN(hipErrorInvalidDevice);
}
amd::Device* device = g_devices[deviceId]->devices()[0];
amd::Device* peer_device = g_devices[peerDeviceId]->devices()[0];
peer_device->disableP2P(device);
HIP_RETURN(hip::getCurrentDevice()->DisablePeerAccess(peerDeviceId));
}
@@ -198,6 +203,11 @@ hipError_t hipDeviceEnablePeerAccess(int peerDeviceId, unsigned int flags) {
if ((hipSuccess != canAccessPeer(&canAccess, deviceId, peerDeviceId)) || (canAccess == 0)) {
HIP_RETURN(hipErrorInvalidDevice);
}
amd::Device* device = g_devices[deviceId]->asContext()->devices()[0];
amd::Device* peer_device = g_devices[peerDeviceId]->asContext()->devices()[0];
peer_device->enableP2P(device);
HIP_RETURN(hip::getCurrentDevice()->EnablePeerAccess(peerDeviceId));
}