Fixed exception from re-enabling peer-to-peer access (#4825)

bda67c67 · Peter Eastman · GitHub · 862e6ac7 · bda67c67 · bda67c67
Unverified Commit bda67c67 authored Mar 04, 2025 by Peter Eastman Committed by GitHub Mar 04, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

platforms/cuda/src/CudaContext.cpp platforms/cuda/src/CudaContext.cpp +1 -1

platforms/hip/src/HipContext.cpp platforms/hip/src/HipContext.cpp +1 -1

No files found.
--- a/platforms/cuda/src/CudaContext.cpp
+++ b/platforms/cuda/src/CudaContext.cpp
@@ -189,7 +189,7 @@ CudaContext::CudaContext(const System& system, int deviceIndex, bool useBlocking
    contextIsValid = true;
    ContextSelector selector(*this);
    CHECK_RESULT(cuCtxSetCacheConfig(CU_FUNC_CACHE_PREFER_SHARED));
-    if (contextIndex > 0) {
+    if (contextIndex > 0 && originalContext == NULL) {
        int canAccess;
        cuDeviceCanAccessPeer(&canAccess, getDevice(), platformData.contexts[0]->getDevice());
        if (canAccess) {

--- a/platforms/hip/src/HipContext.cpp
+++ b/platforms/hip/src/HipContext.cpp
@@ -181,7 +181,7 @@ HipContext::HipContext(const System& system, int deviceIndex, bool useBlockingSy
    contextIsValid = true;
    ContextSelector selector(*this);
-    if (contextIndex > 0) {
+    if (contextIndex > 0 && originalContext == NULL) {
        int canAccess;
        CHECK_RESULT(hipDeviceCanAccessPeer(&canAccess, getDevice(), platformData.contexts[0]->getDevice()));
        if (canAccess) {