diff --git a/paddle/fluid/imperative/layer.h b/paddle/fluid/imperative/layer.h
index 64429af547d0cf06148030fc2b506080578f9cd6..86c2dc3fa4a7d03aa8f0a89a25c17656e1cd708c 100644
--- a/paddle/fluid/imperative/layer.h
+++ b/paddle/fluid/imperative/layer.h
@@ -22,6 +22,7 @@
 #include "paddle/fluid/framework/operator.h"
 #include "paddle/fluid/framework/var_desc.h"
 #include "paddle/fluid/platform/enforce.h"
+#include "pybind11/pybind11.h"
 
 #include "paddle/fluid/imperative/type_defs.h"
 
diff --git a/paddle/fluid/imperative/tracer.h b/paddle/fluid/imperative/tracer.h
index f225d8abe6c0635d2bdd8dba0b12c7fc3a4110db..664cb5ec1c96f4d7ef3d09c515e6ff2d5263c5fd 100644
--- a/paddle/fluid/imperative/tracer.h
+++ b/paddle/fluid/imperative/tracer.h
@@ -14,7 +14,6 @@
 
 #pragma once
 
-#include <map>
 #include <string>
 #include <vector>
 
@@ -40,10 +39,11 @@ class Tracer {
 
   virtual ~Tracer() {}
 
-  void Trace(OpBase* op,
-             const std::map<std::string, std::vector<VarBase*>>& inputs,
-             const std::map<std::string, std::vector<VarBase*>>& outputs,
-             framework::BlockDesc* block, const bool stop_gradient = false);
+  void Trace(
+      OpBase* op,
+      const std::map<std::string, std::vector<VarBase*>>& inputs,   // NOLINT
+      const std::map<std::string, std::vector<VarBase*>>& outputs,  // NOLINT
+      framework::BlockDesc* block, const bool stop_gradient = false);
 
   std::vector<VarBase*> PyTrace(OpBase* op, const std::vector<VarBase*>& inputs,
                                 bool stop_gradient = false);