mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!23818 fix dump CtrlTensor issue when dump not enabled 

Merge pull request !23818 from john_tzanakakis/jt_bug_fixes
This commit is contained in:
i-robot 2021-09-22 01:50:13 +00:00 committed by Gitee
parent e5bdb5dd6d 7efb50ea8d
commit a17915b2c4
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/ccsrc/backend/session/ascend_session.cc
View File

@ -291,8 +291,9 @@ size_t LoadCtrlInputTensor(const std::shared_ptr<KernelGraph> &graph, std::vecto
void UpdateCtrlInputTensor(const std::shared_ptr<KernelGraph> &graph, std::vector<tensor::TensorPtr> *inputs,
size_t *input_ctrl_size) {
if (graph->input_ctrl_tensors()) {
auto &dump_json_parser = DumpJsonParser::GetInstance();
bool sink_mode = (ConfigManager::GetInstance().dataset_mode() == DS_SINK_MODE || graph->IsDatasetGraph());
if (sink_mode) {
if (sink_mode || !dump_json_parser.async_dump_enabled()) {
*input_ctrl_size = LoadCtrlInputTensor(graph, inputs);
} else {
LoadCtrlInputTensor(graph, nullptr);