fix device context init error (PaddlePaddle#43910)

chenwhql · Aug 8, 2022 · 72a48da · 72a48da
1 parent 9336dd3
commit 72a48da
Show file tree

Hide file tree

Showing 3 changed files with 228 additions and 95 deletions.
diff --git a/paddle/fluid/platform/device_context.h b/paddle/fluid/platform/device_context.h
@@ -58,7 +58,7 @@ limitations under the License. */
 #endif
 
 #ifdef PADDLE_WITH_MKLDNN
-#include "dnnl.hpp"
+#include "dnnl.hpp"  // NOLINT
 #include "paddle/fluid/framework/data_layout.h"
 #endif
 
@@ -902,6 +902,8 @@ class DeviceContextPool {
     return *pool;
   }
 
+  static bool IsInitialized() { return pool != nullptr; }
+
   static void SetPool(DeviceContextPool* dev_pool) { pool = dev_pool; }
 
   /*! \brief  Return handle of single device context. */