Skip to content

GitLab

Unverified Commit bda67c67 authored by Peter Eastman's avatar Peter Eastman Committed by GitHub
Browse files

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

parent 862e6ac7
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
platforms/cuda/src/CudaContext.cpp
View file @ bda67c67
......@@ -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) {
......
platforms/hip/src/HipContext.cpp
View file @ bda67c67
......@@ -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) {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment