FIX: throw_on_error on cuda

7010a5da · liaogang · 90c4cd83 · 7010a5da
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddle/platform/enforce.h paddle/platform/enforce.h +1 -1

未找到文件。
--- 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__);