-
Notifications
You must be signed in to change notification settings - Fork 5.6k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Enable P2P memory copy #9190
Enable P2P memory copy #9190
Conversation
paddle/fluid/framework/init.cc
Outdated
PADDLE_ENFORCE(cudaDeviceCanAccessPeer(&can_acess, i, j), | ||
"Failed to test P2P access."); | ||
if (can_acess != 1) { | ||
VLOG(1) << "Cannot enable P2P access from " << i << " to " << j; |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
It may be better to use LOG(WARNING)
.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Agreed. VLOG for debugging only. #5181
LOG(WARNING) is better here.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Done
paddle/fluid/framework/init.cc
Outdated
VLOG(1) << "Cannot enable P2P access from " << i << " to " << j; | ||
} else { | ||
cudaSetDevice(i); | ||
cudaDeviceEnablePeerAccess(j, 0); |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Is P2P Access need any specific hardware? I vaguely remember that it needs the slim-connect(or something)?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
It needs to be tesla. So gforce titan x doesn't support it
paddle/fluid/framework/init.cc
Outdated
PADDLE_ENFORCE(cudaDeviceCanAccessPeer(&can_acess, i, j), | ||
"Failed to test P2P access."); | ||
if (can_acess != 1) { | ||
VLOG(1) << "Cannot enable P2P access from " << i << " to " << j; |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Agreed. VLOG for debugging only. #5181
LOG(WARNING) is better here.
On k40 with 4 devices, time reduces from ~4.0 to ~3.8+, should be more obvious on better hardware
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
On k40 with 4 devices, time reduces from ~4.0 to ~3.8+, should be
more obvious on better hardware