Merge pull request #208 from vlkale/efficientUnorderedMap

kp_sampler_skip.cpp: erase for efficiency of unordered map
kokkos · Oct 5, 2023 · c3e85a6 · c3e85a6
2 parents 1542afe + d1b89e2
commit c3e85a6
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
@@ -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);
     }
   }
 }