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