kp_sampler_skip.cpp: erase in end_parallel_reduce

for infokIDSample efficiency in space
kokkos · Sep 14, 2023 · 0a8db28 · 0a8db28
1 parent 200a1c0
commit 0a8db28
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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);
     }
   }
 }