Fix check bug.

8 years ago · 5a4cdbb3df
parent 03bfd7611a
commit 5a4cdbb3df
1 changed files with 1 additions and 1 deletions
--- a/paddle/operators/math/detail/lstm_gpu_kernel.h
+++ b/paddle/operators/math/detail/lstm_gpu_kernel.h
@ -248,7 +248,7 @@ void gpu_lstm_backward(const platform::DeviceContext& context, Op op,
  cudaStreamSynchronize(stream);
  // TODO(qingqing): Add cuda error check for each kernel.
  cudaError_t err = cudaGetLastError();
-  PADDLE_ENFORCE_EQ(err, cudaGetErrorString(err));
+  PADDLE_ENFORCE(err, cudaGetErrorString(err));
 }

 }  // namespace detail