clean codex warning

pull/6180/head
laiyongqiang 4 years ago
parent 0118930c6b
commit 4063a69846

@ -176,7 +176,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()) {
@ -203,7 +202,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;
}

@ -422,7 +422,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;

@ -107,7 +107,7 @@ class KernelRuntime {
#endif
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