提交 c9bef85d 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!4398 fix ssim precision error in PyNative mode

Merge pull request !4398 from chujinjin/fix_ssim_precision_error_in_pynative_mode
......@@ -329,6 +329,12 @@ std::string GetSingleOpGraphInfo(const OpExecInfoPtr &op_exec_info,
(void)std::for_each(tensor_shape.begin(), tensor_shape.end(),
[&](const auto &dim) { (void)graph_info.append(std::to_string(dim) + "_"); });
(void)graph_info.append(std::to_string(tensor->data_type()) + "_");
if (tensor->device_address() != nullptr) {
(void)graph_info.append(
std::to_string(std::dynamic_pointer_cast<device::DeviceAddress>(tensor->device_address())->type_id()) + "_");
(void)graph_info.append(std::dynamic_pointer_cast<device::DeviceAddress>(tensor->device_address())->format() +
"_");
}
}
// get prim and abstract info
(void)graph_info.append(op_exec_info->prim_id + "_");
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册