Browse Source

!9070 Fix the bug of index out of range in step trace feature

From: @gzhcv
Reviewed-by: @ouwenchang,@lilongfei15
Signed-off-by: @lilongfei15
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
10f795da29
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/runtime/device/gpu/gpu_kernel_runtime.cc

+ 1
- 1
mindspore/ccsrc/runtime/device/gpu/gpu_kernel_runtime.cc View File

@@ -379,9 +379,9 @@ bool GPUKernelRuntime::RunOneStep(const session::KernelGraph *graph) {
// Normally run graph
return LaunchKernelDynamic(graph);
}
is_first_step_map_[graph_id] = false;
// Mock run first step
bool ret = LaunchKernelDynamic(graph, true, false);
is_first_step_map_[graph_id] = false;
if (ret) {
// Normally run graph
return LaunchKernelDynamic(graph);


Loading…
Cancel
Save