diff --git a/mindspore/ccsrc/backend/optimizer/mem_reuse/mem_reuse_allocator.cc b/mindspore/ccsrc/backend/optimizer/mem_reuse/mem_reuse_allocator.cc
index 427b6b776d..c3eebaad38 100644
--- a/mindspore/ccsrc/backend/optimizer/mem_reuse/mem_reuse_allocator.cc
+++ b/mindspore/ccsrc/backend/optimizer/mem_reuse/mem_reuse_allocator.cc
@@ -178,7 +178,6 @@ void BestFitMemReuse::AssignCommonNodeOutputOffset() {
 
 void BestFitMemReuse::AssignCommunicationNodeOutputOffset() {
   size_t total_kernel_output_size = 0;
-  size_t output_num = 0;
   // get all output size
   MS_EXCEPTION_IF_NULL(current_kernel_);
   for (const auto &tensor_idx : current_kernel_->GetOutputRefIndexs()) {
@@ -205,7 +204,7 @@ void BestFitMemReuse::AssignCommunicationNodeOutputOffset() {
     size_t index = GetTensorIndex(tensor_idx);
     auto tensor_desc = tensor_ptr_list_[index];
     MS_EXCEPTION_IF_NULL(tensor_desc);
-    if (output_index == 0 || output_index == output_num - 1) {
+    if (output_index == 0 || output_index == output_ref_indexes.size() - 1) {
       tensor_desc->size_ += kDefaultMemAlignSize;
     }
 
diff --git a/mindspore/ccsrc/runtime/device/kernel_runtime.cc b/mindspore/ccsrc/runtime/device/kernel_runtime.cc
index fa51e21c40..5dfffbaa39 100644
--- a/mindspore/ccsrc/runtime/device/kernel_runtime.cc
+++ b/mindspore/ccsrc/runtime/device/kernel_runtime.cc
@@ -339,7 +339,7 @@ void KernelRuntime::AssignStaticMemoryInput(const session::KernelGraph *graph) {
   }
 }
 
-void KernelRuntime::AssignStaticMemoryOutput(session::KernelGraph *graph) {
+void KernelRuntime::AssignStaticMemoryOutput(const session::KernelGraph *graph) {
   MS_EXCEPTION_IF_NULL(graph);
   auto nodes = AnfAlgo::GetAllOutput(graph->output(), {prim::kPrimTupleGetItem});
   std::vector<session::KernelWithIndex> non_communication_op;
diff --git a/mindspore/ccsrc/runtime/device/kernel_runtime.h b/mindspore/ccsrc/runtime/device/kernel_runtime.h
index 7cd8427e57..145fa00c50 100644
--- a/mindspore/ccsrc/runtime/device/kernel_runtime.h
+++ b/mindspore/ccsrc/runtime/device/kernel_runtime.h
@@ -96,7 +96,7 @@ class KernelRuntime {
   void AssignCommunicationNodeMem(MemType type, const AnfNodePtr &node);
 
  private:
-  void AssignStaticMemoryOutput(session::KernelGraph *graph);
+  void AssignStaticMemoryOutput(const session::KernelGraph *graph);
   void GenLaunchArgs(const mindspore::kernel::KernelMod &kernel_mod, const AnfNodePtr &kernel,
                      AddressPtrList *kernel_inputs, AddressPtrList *kernel_workspaces, AddressPtrList *kernel_outputs);
   bool LaunchKernelMod(const session::KernelGraph &graph);