Browse Source

!6180 clean codex warning in memreuse

Merge pull request !6180 from laiyongqiang/codex
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
c543db0585
3 changed files with 3 additions and 4 deletions
  1. +1
    -2
      mindspore/ccsrc/backend/optimizer/mem_reuse/mem_reuse_allocator.cc
  2. +1
    -1
      mindspore/ccsrc/runtime/device/kernel_runtime.cc
  3. +1
    -1
      mindspore/ccsrc/runtime/device/kernel_runtime.h

+ 1
- 2
mindspore/ccsrc/backend/optimizer/mem_reuse/mem_reuse_allocator.cc View File

@@ -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;
}



+ 1
- 1
mindspore/ccsrc/runtime/device/kernel_runtime.cc View File

@@ -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;


+ 1
- 1
mindspore/ccsrc/runtime/device/kernel_runtime.h View File

@@ -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);


Loading…
Cancel
Save