Force sync dev

helinwang-patch-1
Yu Yang 7 years ago
parent dfb8680018
commit 52dd8ff09a

@ -199,7 +199,7 @@ void ThreadedSSAGraphExecutor::RunOp(
auto op_run = [ready_buffer, op, this] { auto op_run = [ready_buffer, op, this] {
try { try {
VLOG(10) << op->DebugString(); VLOG(10) << op->Name() << " : " << op->DebugString();
op->Run(use_event_); op->Run(use_event_);
for (auto *ready : *ready_buffer) { for (auto *ready : *ready_buffer) {
@ -211,6 +211,7 @@ void ThreadedSSAGraphExecutor::RunOp(
} catch (...) { } catch (...) {
LOG(FATAL) << "Unknown exception catched"; LOG(FATAL) << "Unknown exception catched";
} }
PADDLE_ENFORCE(cudaDeviceSynchronize());
}; };
if (pool_) { if (pool_) {
pool_->enqueue(op_run); pool_->enqueue(op_run);

Loading…
Cancel
Save