From 5a4cdbb3dfb2de82ed6864d38a4381c52d4dba4c Mon Sep 17 00:00:00 2001 From: dangqingqing Date: Thu, 2 Nov 2017 00:30:12 +0800 Subject: [PATCH] Fix check bug. --- paddle/operators/math/detail/lstm_gpu_kernel.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/operators/math/detail/lstm_gpu_kernel.h b/paddle/operators/math/detail/lstm_gpu_kernel.h index e07655eaac..1781460c35 100644 --- 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 -- GitLab