Skip to content

P
Paddle

BaiXuePrincess / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
提交 d3e55fde 编写于 作者: Y Yu Yang
浏览文件
操作

Guard devctx

上级 a5ba704d
隐藏空白更改
内联 并排
Showing with 2 addition and 0 deletion
paddle/fluid/platform/device_context.cc
浏览文件 @ d3e55fde
...@@ -159,6 +159,7 @@ CUDADeviceContext::~CUDADeviceContext() { ...@@ -159,6 +159,7 @@ CUDADeviceContext::~CUDADeviceContext() {
Place CUDADeviceContext::GetPlace() const { return place_; } Place CUDADeviceContext::GetPlace() const { return place_; }
void CUDADeviceContext::Wait() const { void CUDADeviceContext::Wait() const {
std::lock_guard<std::mutex> guard(this->mutex_);
PADDLE_ENFORCE(cudaStreamSynchronize(stream_)); PADDLE_ENFORCE(cudaStreamSynchronize(stream_));
PADDLE_ENFORCE(cudaGetLastError()); PADDLE_ENFORCE(cudaGetLastError());
} }
......
paddle/fluid/platform/device_context.h
浏览文件 @ d3e55fde
...@@ -110,6 +110,7 @@ class CUDADeviceContext : public DeviceContext { ...@@ -110,6 +110,7 @@ class CUDADeviceContext : public DeviceContext {
int compute_capability; int compute_capability;
int multi_process; int multi_process;
int max_threads_per_mp; int max_threads_per_mp;
mutable std::mutex mutex_;
}; };
template <> template <>
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或