diff --git a/ge/graph/load/model_manager/task_info/kernel_task_info.cc b/ge/graph/load/model_manager/task_info/kernel_task_info.cc
index b60529d5..f3213308 100755
--- a/ge/graph/load/model_manager/task_info/kernel_task_info.cc
+++ b/ge/graph/load/model_manager/task_info/kernel_task_info.cc
@@ -414,8 +414,7 @@ Status KernelTaskInfo::SuperKernelDistribute() {
   return SUCCESS;
 }
 
-Status KernelTaskInfo::Distribute() {
-  GELOGD("KernelTaskInfo Distribute Start.");
+void KernelTaskInfo::SetArgs() {
   if (davinci_model_->IsKnownNode()) {
     if (kernel_type_ == ccKernelType::TE) {
       args_ = l2_buffer_on_ ? davinci_model_->GetCurrentHybridArgsAddr(hybrid_args_offset_)
@@ -425,6 +424,11 @@ Status KernelTaskInfo::Distribute() {
     }
     GELOGI("Known node %s args addr %p, offset %u.", op_desc_->GetName().c_str(), args_, args_offset_);
   }
+}
+
+Status KernelTaskInfo::Distribute() {
+  GELOGD("KernelTaskInfo Distribute Start.");
+  SetArgs();
   rtError_t rt_ret = RT_ERROR_NONE;
   char skt_enable_env[MMPA_MAX_PATH] = { 0x00 };
   INT32 res = mmGetEnv("SKT_ENABLE", skt_enable_env, MMPA_MAX_PATH);
@@ -1075,7 +1079,7 @@ Status KernelTaskInfo::InitAicpuTask(uint32_t op_index, const domi::KernelDef &k
     return RT_ERROR_TO_GE_STATUS(rt_ret);
   }
   InitDumpTask(sizeof(aicpu::AicpuParamHead));
-  
+
   if (kernel_type_ == ccKernelType::CUST_AI_CPU) {
     dump_flag_ |= RT_KERNEL_CUSTOM_AICPU;
   }
diff --git a/ge/graph/load/model_manager/task_info/kernel_task_info.h b/ge/graph/load/model_manager/task_info/kernel_task_info.h
index 83df8736..79347255 100644
--- a/ge/graph/load/model_manager/task_info/kernel_task_info.h
+++ b/ge/graph/load/model_manager/task_info/kernel_task_info.h
@@ -145,6 +145,7 @@ class KernelTaskInfo : public TaskInfo {
   bool IsMarkedFirstNode();
   bool FirstCallSKTLaunchCheck();
   bool DoubleCallSKTSaveCheck();
+  void SetArgs();
 
   void *stub_func_;
   void *args_;