Browse Source

!11907 Fix the bug of gpu step_trace cannot recognize correct fp/bp node in train_eval mode

From: @gzhcv
Reviewed-by: @yelihua,@ouwenchang
Signed-off-by: @ouwenchang
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 4 years ago
parent
commit
4d2e0ea91f
1 changed files with 0 additions and 4 deletions
  1. +0
    -4
      mindspore/ccsrc/profiler/device/gpu/gpu_profiling_utils.cc

+ 0
- 4
mindspore/ccsrc/profiler/device/gpu/gpu_profiling_utils.cc View File

@@ -74,10 +74,6 @@ void ProfilingUtils::GetTraceHccl(const std::vector<CNodePtr> &cnode_exec_order)
}

void ProfilingUtils::SetTraceFpStart(const std::vector<CNodePtr> &cnode_exec_order) {
if (!profiling_trace.trace_fp_start.empty()) {
return;
}

const char *trace_fp_start = std::getenv(kFpStartNode);
if (trace_fp_start != nullptr) {
profiling_trace.trace_fp_start = std::string(trace_fp_start);


Loading…
Cancel
Save