From a2366d41755f3c41f682e5d124a09836289a42fa Mon Sep 17 00:00:00 2001 From: wangyu0516 Date: Tue, 15 Feb 2022 10:23:47 +0800 Subject: [PATCH] debug hccl --- ge/graph/load/model_manager/task_info/hccl_task_info.cc | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/ge/graph/load/model_manager/task_info/hccl_task_info.cc b/ge/graph/load/model_manager/task_info/hccl_task_info.cc index a3cef836..f0a5e3fd 100644 --- a/ge/graph/load/model_manager/task_info/hccl_task_info.cc +++ b/ge/graph/load/model_manager/task_info/hccl_task_info.cc @@ -22,7 +22,7 @@ #include "framework/common/debug/ge_log.h" #include "graph/load/model_manager/davinci_model.h" #include "graph/load/model_manager/model_utils.h" - +#include namespace ge { std::mutex HcclTaskInfo::hccl_follow_stream_mutex_; @@ -299,6 +299,14 @@ Status HcclTaskInfo::SetAddrs(const std::shared_ptr &op_desc, input_data_addr = input_data_addrs.empty() ? nullptr : input_data_addrs[i]; output_data_addr = output_data_addrs.empty() ? nullptr : output_data_addrs[i]; } + std::cout << "hccl task info" << std::endl; + if (input_data_addr != nullptr) { + std::cout << "input " << input_data_addr << std::endl; + } + if (output_data_addr != nullptr) { + std::cout << "output " << output_data_addr << std::endl; + } + kernel_hccl_infos[i].inputDataAddr = input_data_addr; if (hccl_type == HCOMALLGATHER || hccl_type == HCOMRECEIVE || hccl_type == HVDCALLBACKALLGATHER) { kernel_hccl_infos[i].outputDataAddr = output_data_addr; @@ -309,6 +317,10 @@ Status HcclTaskInfo::SetAddrs(const std::shared_ptr &op_desc, kernel_hccl_infos[i].outputDataAddr = output_data_addr; kernel_hccl_infos[i].opType = op_type; } + auto env = std::getenv("SKIP_OUTPUT"); + if (env != nullptr) { + davinci_model_->DisableZeroCopy(output_data_addr); + } davinci_model_->DisableZeroCopy(input_data_addr); } return SUCCESS;