Skip to content

Commit c3e85a6

Browse files
authored
Merge pull request #208 from vlkale/efficientUnorderedMap
kp_sampler_skip.cpp: erase for efficiency of unordered map
2 parents 1542afe + d1b89e2 commit c3e85a6

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

common/kokkos-sampler/kp_sampler_skip.cpp

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -181,6 +181,7 @@ void kokkosp_end_parallel_for(const uint64_t kID) {
181181
(unsigned long long)(kID));
182182
}
183183
(*endForCallee)(retrievedNestedkID);
184+
infokIDSample.erase(kID);
184185
}
185186
}
186187
}
@@ -212,6 +213,7 @@ void kokkosp_end_parallel_scan(const uint64_t kID) {
212213
(unsigned long long)(kID));
213214
}
214215
(*endScanCallee)(retrievedNestedkID);
216+
infokIDSample.erase(kID);
215217
}
216218
}
217219
}
@@ -244,6 +246,7 @@ void kokkosp_end_parallel_reduce(const uint64_t kID) {
244246
(unsigned long long)(kID));
245247
}
246248
(*endScanCallee)(retrievedNestedkID);
249+
infokIDSample.erase(kID);
247250
}
248251
}
249252
}

0 commit comments

Comments
 (0)