Skip to content

Commit 32afcf6

Browse files
cleaning(simple-kernel-timer): promote find_index to kp_shared.hpp
1 parent 91d9df2 commit 32afcf6

File tree

3 files changed

+12
-26
lines changed

3 files changed

+12
-26
lines changed

profiling/simple-kernel-timer/kp_json_writer.cpp

-13
Original file line numberDiff line numberDiff line change
@@ -54,19 +54,6 @@ inline void write_json(std::ostream& os, KernelPerformanceInfo const& kp,
5454
os << indent << '}';
5555
}
5656

57-
int find_index(std::vector<KernelPerformanceInfo*>& kernels,
58-
const char* kernelName) {
59-
for (unsigned int i = 0; i < kernels.size(); i++) {
60-
KernelPerformanceInfo* nextKernel = kernels[i];
61-
62-
if (strcmp(nextKernel->getName(), kernelName) == 0) {
63-
return i;
64-
}
65-
}
66-
67-
return -1;
68-
}
69-
7057
int main(int argc, char* argv[]) {
7158
if (argc == 1) {
7259
fprintf(stderr, "Did you specify any data files on the command line!\n");

profiling/simple-kernel-timer/kp_reader.cpp

-13
Original file line numberDiff line numberDiff line change
@@ -25,19 +25,6 @@
2525

2626
using namespace KokkosTools::KernelTimer;
2727

28-
int find_index(std::vector<KernelPerformanceInfo*>& kernels,
29-
const char* kernelName) {
30-
for (unsigned int i = 0; i < kernels.size(); i++) {
31-
KernelPerformanceInfo* nextKernel = kernels[i];
32-
33-
if (strcmp(nextKernel->getName(), kernelName) == 0) {
34-
return i;
35-
}
36-
}
37-
38-
return -1;
39-
}
40-
4128
int main(int argc, char* argv[]) {
4229
if (argc == 1) {
4330
fprintf(stderr, "Did you specify any data files on the command line!\n");

profiling/simple-kernel-timer/kp_shared.h

+12
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,11 @@
1717
#ifndef _H_KOKKOSP_KERNEL_SHARED
1818
#define _H_KOKKOSP_KERNEL_SHARED
1919

20+
#include <algorithm>
2021
#include <map>
2122
#include <memory>
23+
#include <vector>
24+
2225
#include "kp_kernel_info.h"
2326

2427
namespace KokkosTools::KernelTimer {
@@ -39,6 +42,15 @@ inline bool compareKernelPerformanceInfo(KernelPerformanceInfo* left,
3942
return left->getTime() > right->getTime();
4043
};
4144

45+
inline int find_index(const std::vector<KernelPerformanceInfo*>& kernels,
46+
const char* kernelName) {
47+
const auto it =
48+
std::find_if(kernels.begin(), kernels.end(), [&](const auto& kernel) {
49+
return strcmp(kernel->getName(), kernelName) == 0;
50+
});
51+
return it == kernels.cend() ? -1 : std::distance(kernels.begin(), it);
52+
}
53+
4254
} // namespace KokkosTools::KernelTimer
4355

4456
#endif // _H_KOKKOSP_KERNEL_SHARED

0 commit comments

Comments
 (0)