提交 5a4cdbb3 编写于 作者: D dangqingqing

Fix check bug.

上级 03bfd761
...@@ -248,7 +248,7 @@ void gpu_lstm_backward(const platform::DeviceContext& context, Op op, ...@@ -248,7 +248,7 @@ void gpu_lstm_backward(const platform::DeviceContext& context, Op op,
cudaStreamSynchronize(stream); cudaStreamSynchronize(stream);
// TODO(qingqing): Add cuda error check for each kernel. // TODO(qingqing): Add cuda error check for each kernel.
cudaError_t err = cudaGetLastError(); cudaError_t err = cudaGetLastError();
PADDLE_ENFORCE_EQ(err, cudaGetErrorString(err)); PADDLE_ENFORCE(err, cudaGetErrorString(err));
} }
} // namespace detail } // namespace detail
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册