diff --git a/paddle/fluid/operators/jit/benchmark.cc b/paddle/fluid/operators/jit/benchmark.cc
index ef7ccc64adf2a93c1ac8e2a00ffb3299c10d3647..5cc82b69f8b8ee098a704913b66f141eea10df45 100644
--- a/paddle/fluid/operators/jit/benchmark.cc
+++ b/paddle/fluid/operators/jit/benchmark.cc
@@ -12,9 +12,8 @@
  * See the License for the specific language governing permissions and
  * limitations under the License. */
 
-// #include <cstring>  // for memcpy
-// #include <random>
 #include <iostream>
+#include <random>
 #include <string>
 #include <vector>
 #include "gflags/gflags.h"
diff --git a/paddle/fluid/operators/jit/helper.h b/paddle/fluid/operators/jit/helper.h
index 11bbd6a56cf4d1503f9739ee55971bf70b7ee883..d1bbe103814323e803d9684bf0d901b28cb982e3 100644
--- a/paddle/fluid/operators/jit/helper.h
+++ b/paddle/fluid/operators/jit/helper.h
@@ -49,15 +49,17 @@ inline typename KernelTuples::func_type GetJitCode(
     // pool: (KernelKey(type, place), vector<GenCreatorPtr>)
     auto& creator_map = JitCodeCreatorPool().Instance().AllCreators();
     auto iter = creator_map.find(kkey);
-    auto& creators = iter->second;
-    for (auto& cur : creators) {
-      auto i = dynamic_cast<const JitCodeCreator<Attr>*>(cur.get());
-      if (i && i->UseMe(attr)) {
-        auto p = i->CreateJitCode(attr);
-        if (p) {
-          auto f = p->template getCode<Func>();
-          codes.Insert(key, std::move(p));
-          return f;
+    if (iter != creator_map.end()) {
+      auto& creators = iter->second;
+      for (auto& cur : creators) {
+        auto i = dynamic_cast<const JitCodeCreator<Attr>*>(cur.get());
+        if (i && i->UseMe(attr)) {
+          auto p = i->CreateJitCode(attr);
+          if (p) {
+            auto f = p->template getCode<Func>();
+            codes.Insert(key, std::move(p));
+            return f;
+          }
         }
       }
     }