Update kp_sampler_skip.cpp: put erase back

Erase got deleted (there on the develop branch) Took it out due to copy-paste
kokkos · Oct 23, 2023 · 9df6748 · 9df6748
1 parent d0a24ba
commit 9df6748
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/common/kokkos-sampler/kp_sampler_skip.cpp b/common/kokkos-sampler/kp_sampler_skip.cpp
@@ -299,6 +299,7 @@ void kokkosp_end_parallel_for(const uint64_t kID) {
         invoke_ktools_fence(0);
       }
       (*endForCallee)(retrievedNestedkID);
+      infokIDSample.erase(kID);
     }
   }
 }
@@ -338,6 +339,7 @@ void kokkosp_end_parallel_scan(const uint64_t kID) {
         invoke_ktools_fence(0);
       }
       (*endScanCallee)(retrievedNestedkID);
+      infokIDSample.erase(kID);
     }
   }
 }
@@ -377,6 +379,7 @@ void kokkosp_end_parallel_reduce(const uint64_t kID) {
         invoke_ktools_fence(0);
       }
       (*endScanCallee)(retrievedNestedkID);
+      infokIDSample.erase(kID);
     }
   }
 }