From 0a8db2858600fb34eb4793fe6d239c25be78cc7f Mon Sep 17 00:00:00 2001
From: Vivek Kale <11766050+vlkale@users.noreply.github.com>
Date: Thu, 14 Sep 2023 10:10:19 -0700
Subject: [PATCH] kp_sampler_skip.cpp: erase in end_parallel_reduce

for infokIDSample efficiency in space
---
 common/kokkos-sampler/kp_sampler_skip.cpp | 1 +
 1 file changed, 1 insertion(+)

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);
     }
   }
 }