diff --git a/mindspore/ccsrc/kernel/tbe/tbe_kernel_parallel_build.cc b/mindspore/ccsrc/kernel/tbe/tbe_kernel_parallel_build.cc
index bac2cae87119ba8398c797c92bf9048e31e2acae..577af45d5956c18936934d8f7ae19cbb29396b35 100644
--- a/mindspore/ccsrc/kernel/tbe/tbe_kernel_parallel_build.cc
+++ b/mindspore/ccsrc/kernel/tbe/tbe_kernel_parallel_build.cc
@@ -245,7 +245,10 @@ void ParallelBuildManager::ResetTaskInfo() {
   }
   task_map_.clear();
   same_op_list_.clear();
-  (void)PyObject_CallMethod(tbe_parallel_compiler_, kResetTaskInfo, "O");
+  if (tbe_parallel_compiler_ != nullptr) {
+    PyObject *pArg = Py_BuildValue("()");
+    (void)PyObject_CallMethod(tbe_parallel_compiler_, kResetTaskInfo, "O", pArg);
+  }
 }
 }  // namespace kernel
 }  // namespace mindspore