p12346870
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

debugger won't read unused slots for BatchNorm

This commit is contained in:
John Tzanakakis 2020-11-13 10:51:02 -05:00
parent 90aaf8a58c
commit d45ef4cacb
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/ccsrc/runtime/device/gpu/gpu_kernel_runtime.cc
View File

@ -89,8 +89,8 @@ std::vector<int> CheckRealOutput(const std::string &node_name, const size_t &out
std::vector<int> real_outputs;
// P.FusedBatchNorm is used for training; P.BatchNorm is used for inference
// can add the filter list for more operators here....
if (node_name == "FusedBatchNorm") {
MS_LOG(INFO) << "loading node named FusedBatchNorm.";
if (node_name == "FusedBatchNorm" || node_name == "BatchNorm") {
MS_LOG(INFO) << "loading node named " << node_name;
real_outputs.insert(real_outputs.end(), {0, 3, 4});
} else {
// by default, TensorLoader will load all outputs