提交 e9e27e0f 编写于 作者: Y Yu Yang

Revert

上级 7289e758
...@@ -204,8 +204,8 @@ void ThreadedSSAGraphExecutor::RunOp( ...@@ -204,8 +204,8 @@ void ThreadedSSAGraphExecutor::RunOp(
VLOG(10) << op << " " << op->Name() << "Signal posted"; VLOG(10) << op << " " << op->Name() << "Signal posted";
} catch (platform::EnforceNotMet ex) { } catch (platform::EnforceNotMet ex) {
exception_.reset(new platform::EnforceNotMet(ex)); exception_.reset(new platform::EnforceNotMet(ex));
} catch (std::exception &exp) { } catch (...) {
LOG(FATAL) << exp.what(); LOG(FATAL) << "Unknown exception catched";
} }
}; };
if (pool_) { if (pool_) {
......
...@@ -190,7 +190,6 @@ void PyCUDATensorSetFromArray( ...@@ -190,7 +190,6 @@ void PyCUDATensorSetFromArray(
static_cast<const platform::CUDADeviceContext *>(pool.Get(place)); static_cast<const platform::CUDADeviceContext *>(pool.Get(place));
paddle::platform::GpuMemcpyAsync(dst, array.data(), sizeof(T) * array.size(), paddle::platform::GpuMemcpyAsync(dst, array.data(), sizeof(T) * array.size(),
cudaMemcpyHostToDevice, dev_ctx->stream()); cudaMemcpyHostToDevice, dev_ctx->stream());
dev_ctx->Wait();
} }
template <> template <>
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册