Browse Source

fix the bug of index out of range in step trace feature

tags/v1.1.0
gzhcv 5 years ago
parent
commit
dead0bbf80
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