From dead0bbf800b940d60c21ee6d0a9fe8d60accadd Mon Sep 17 00:00:00 2001
From: gzhcv <guozhaohong@huawei.com>
Date: Thu, 26 Nov 2020 17:33:45 +0800
Subject: [PATCH] fix the bug of index out of range in step trace feature

---
 mindspore/ccsrc/runtime/device/gpu/gpu_kernel_runtime.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mindspore/ccsrc/runtime/device/gpu/gpu_kernel_runtime.cc b/mindspore/ccsrc/runtime/device/gpu/gpu_kernel_runtime.cc
index 51ffb02ea2..864d9f82b7 100644
--- a/mindspore/ccsrc/runtime/device/gpu/gpu_kernel_runtime.cc
+++ b/mindspore/ccsrc/runtime/device/gpu/gpu_kernel_runtime.cc
@@ -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);