FIX: throw_on_error on cuda

8 years ago · 7010a5da1a
parent 90c4cd8323
commit 7010a5da1a
1 changed files with 1 additions and 1 deletions
--- a/paddle/platform/enforce.h
+++ b/paddle/platform/enforce.h
@ -78,7 +78,7 @@ class EnforceNotMet : public std::exception {

 template <typename... Args>
 inline void throw_on_error(cudaError_t e, const Args&... args) {
-  if (UNLIKELY(!(e))) {
+  if (e) {
    std::stringstream ss;
    ss << ::paddle::string::Sprintf(args...);
    ss << ::paddle::string::Sprintf(" at [%s:%s];", __FILE__, __LINE__);