1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
|
//@HEADER
// ************************************************************************
//
// Kokkos v. 4.0
// Copyright (2022) National Technology & Engineering
// Solutions of Sandia, LLC (NTESS).
//
// Under the terms of Contract DE-NA0003525 with NTESS,
// the U.S. Government retains certain rights in this software.
//
// Part of Kokkos, under the Apache License v2.0 with LLVM Exceptions.
// See https://kokkos.org/LICENSE for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//@HEADER
#include <inttypes.h>
#include <iostream>
struct Kokkos_Profiling_KokkosPDeviceInfo;
// just get the basename for print_help/parse_args
std::string get_basename(char* cmd, int idx = 0) {
if (idx > 0) return cmd;
std::string _cmd = cmd;
auto _pos = _cmd.find_last_of('/');
if (_pos != std::string::npos) return _cmd.substr(_pos + 1);
return _cmd;
}
struct SpaceHandle {
char name[64];
};
const int parallel_for_id = 0;
const int parallel_reduce_id = 1;
const int parallel_scan_id = 2;
extern "C" void kokkosp_init_library(
const int /*loadSeq*/, const uint64_t /*interfaceVer*/,
const uint32_t /*devInfoCount*/,
Kokkos_Profiling_KokkosPDeviceInfo* /* deviceInfo */) {
std::cout << "kokkosp_init_library::";
}
extern "C" void kokkosp_finalize_library() {
std::cout << "kokkosp_finalize_library::";
}
extern "C" void kokkosp_print_help(char* exe) {
std::cout << "kokkosp_print_help:" << get_basename(exe) << "::";
}
extern "C" void kokkosp_parse_args(int argc, char** argv) {
std::cout << "kokkosp_parse_args:" << argc;
for (int i = 0; i < argc; ++i) std::cout << ":" << get_basename(argv[i], i);
std::cout << "::";
}
extern "C" void kokkosp_begin_parallel_for(const char* name,
const uint32_t devID,
uint64_t* kID) {
*kID = parallel_for_id;
std::cout << "kokkosp_begin_parallel_for:" << name << ":" << devID << ":"
<< *kID << "::";
}
extern "C" void kokkosp_end_parallel_for(const uint64_t kID) {
std::cout << "kokkosp_end_parallel_for:" << kID << "::";
}
extern "C" void kokkosp_begin_parallel_scan(const char* name,
const uint32_t devID,
uint64_t* kID) {
*kID = parallel_scan_id;
std::cout << "kokkosp_begin_parallel_scan:" << name << ":" << devID << ":"
<< *kID << "::";
}
extern "C" void kokkosp_end_parallel_scan(const uint64_t kID) {
std::cout << "kokkosp_end_parallel_scan:" << kID << "::";
}
extern "C" void kokkosp_begin_parallel_reduce(const char* name,
const uint32_t devID,
uint64_t* kID) {
*kID = parallel_reduce_id;
std::cout << "kokkosp_begin_parallel_reduce:" << name << ":" << devID << ":"
<< *kID << "::";
}
extern "C" void kokkosp_end_parallel_reduce(const uint64_t kID) {
std::cout << "kokkosp_end_parallel_reduce:" << kID << "::";
}
extern "C" void kokkosp_push_profile_region(const char* regionName) {
std::cout << "kokkosp_push_profile_region:" << regionName << "::";
}
extern "C" void kokkosp_pop_profile_region() {
std::cout << "kokkosp_pop_profile_region::";
}
extern "C" void kokkosp_allocate_data(SpaceHandle handle, const char* name,
const void* ptr, uint64_t size) {
std::cout << "kokkosp_allocate_data:" << handle.name << ":" << name << ":"
<< ptr << ":" << size << "::";
}
extern "C" void kokkosp_deallocate_data(SpaceHandle handle, const char* name,
const void* ptr, uint64_t size) {
std::cout << "kokkosp_deallocate_data:" << handle.name << ":" << name << ":"
<< ptr << ":" << size << "::";
}
extern "C" void kokkosp_begin_deep_copy(SpaceHandle dst_handle,
const char* dst_name,
const void* dst_ptr,
SpaceHandle src_handle,
const char* src_name,
const void* src_ptr, uint64_t size) {
std::cout << "kokkosp_begin_deep_copy:" << dst_handle.name << ":" << dst_name
<< ":" << dst_ptr << ":" << src_handle.name << ":" << src_name
<< ":" << src_ptr << ":" << size << "::";
}
extern "C" void kokkosp_end_deep_copy() {
std::cout << "kokkosp_end_deep_copy::";
}
uint32_t section_id = 3;
extern "C" void kokkosp_create_profile_section(const char* name,
uint32_t* sec_id) {
*sec_id = section_id;
std::cout << "kokkosp_create_profile_section:" << name << ":" << *sec_id
<< "::";
}
extern "C" void kokkosp_start_profile_section(uint32_t sec_id) {
std::cout << "kokkosp_start_profile_section:" << sec_id << "::";
}
extern "C" void kokkosp_stop_profile_section(uint32_t sec_id) {
std::cout << "kokkosp_stop_profile_section:" << sec_id << "::";
}
extern "C" void kokkosp_destroy_profile_section(uint32_t sec_id) {
std::cout << "kokkosp_destroy_profile_section:" << sec_id << "::";
}
extern "C" void kokkosp_profile_event(const char* name) {
std::cout << "kokkosp_profile_event:" << name << "::";
}
extern "C" void kokkosp_declare_metadata(const char* key, const char* value) {
std::cout << "kokkosp_declare_metadata:" << key << ":" << value << "::";
}
|
