Browse Source

add ps cache sparse mode check

tags/v1.2.0-rc1
lizhenyu 4 years ago
parent
commit
c89e00e13b
1 changed files with 5 additions and 0 deletions
  1. +5
    -0
      mindspore/ccsrc/runtime/device/kernel_runtime.cc

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

@@ -1151,6 +1151,11 @@ void KernelRuntime::CheckIfSupportPSEmbeddingCache(const session::KernelGraph *g
<< input_index.first->fullname_with_scope();
MS_LOG(EXCEPTION) << "The embeddingLookup whose input index isn't from dataset doesn't support cache in "
"parameter server training mode.";
} else if (input_index.first->isa<CNode>() && (AnfAlgo::GetCNodeName(input_index.first) == kGetNextOpName)) {
MS_LOG(ERROR) << "The EmbeddingLookup kernel(" << kernel->fullname_with_scope() << ") doesn't enable cache.";
MS_LOG(EXCEPTION) << "All EmbeddingLookup kernels whose input indices are from dataset must enable cache at "
"the same time and parameter 'sparse' must be equal to the value of 'enable_sparse' in "
"context setting in parameter server training mode.";
}
}
}


Loading…
Cancel
Save