Browse Source

modify graph info to fix bug of sync stream error

tags/v1.2.0-rc1
zuochuanyong 5 years ago
parent
commit
73cdfa6086
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      mindspore/ccsrc/backend/session/session_basic.cc

+ 3
- 0
mindspore/ccsrc/backend/session/session_basic.cc View File

@@ -1288,6 +1288,9 @@ GraphInfo SessionBasic::GetSingleOpGraphInfo(const CNodePtr &kernel,
const auto format = std::dynamic_pointer_cast<device::DeviceAddress>(tensor->device_address())->format();
(void)graph_info.append(format + "_");
}
for (const auto &padding_type : tensor->padding_type()) {
(void)graph_info.append(std::to_string(padding_type) + "_");
}
}
// get attr info
const auto &attr_map = prim->attrs();


Loading…
Cancel
Save