diff --git a/common/kokkos-sampler/kp_sampler_skip.cpp b/common/kokkos-sampler/kp_sampler_skip.cpp index 9c53bcc5c..462318256 100644 --- a/common/kokkos-sampler/kp_sampler_skip.cpp +++ b/common/kokkos-sampler/kp_sampler_skip.cpp @@ -181,6 +181,7 @@ void kokkosp_end_parallel_for(const uint64_t kID) { (unsigned long long)(kID)); } (*endForCallee)(retrievedNestedkID); + infokIDSample.erase(kID); } } } @@ -212,6 +213,7 @@ void kokkosp_end_parallel_scan(const uint64_t kID) { (unsigned long long)(kID)); } (*endScanCallee)(retrievedNestedkID); + infokIDSample.erase(kID); } } } @@ -244,6 +246,7 @@ void kokkosp_end_parallel_reduce(const uint64_t kID) { (unsigned long long)(kID)); } (*endScanCallee)(retrievedNestedkID); + infokIDSample.erase(kID); } } }