Cheetah - SKA - PSS - Prototype Time Domain Search Pipeline
nvtx.h
1 /*
2  * The MIT License (MIT)
3  *
4  * Copyright (c) 2016 The SKA organisation
5  *
6  * Permission is hereby granted, free of charge, to any person obtaining a copy
7  * of this software and associated documentation files (the "Software"), to deal
8  * in the Software without restriction, including without limitation the rights
9  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
10  * copies of the Software, and to permit persons to whom the Software is
11  * furnished to do so, subject to the following conditions:
12  *
13  * The above copyright notice and this permission notice shall be included in all
14  * copies or substantial portions of the Software.
15  *
16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22  * SOFTWARE.
23  */
24 #ifndef SKA_CHEETAH_CUDA_UTILS_NVTX_H
25 #define SKA_CHEETAH_CUDA_UTILS_NVTX_H
26 
27 /* NVTX wrapper
28  * http://http.developer.nvidia.com/ParallelNsight/1.51/UserGuide/HTML/How_To_Use_the_NVIDIA_Tools_Extension_Library_%28nvtx%29.html
29  */
30 
31 #ifdef USE_NVTX
32 #include "nvToolsExt.h"
33 
34 static const uint32_t colors[] = { 0x0000ff00, 0x000000ff, 0x00ffff00, 0x00ff00ff, 0x0000ffff, 0x00ff0000, 0x00ffffff };
35 static const int num_colors = sizeof(colors)/sizeof(uint32_t);
36 
37 #define PUSH_NVTX_RANGE(name,cid) { \
38  int color_id = cid; \
39  color_id = color_id%num_colors;\
40  nvtxEventAttributes_t eventAttrib = {0}; \
41  eventAttrib.version = NVTX_VERSION; \
42  eventAttrib.size = NVTX_EVENT_ATTRIB_STRUCT_SIZE; \
43  eventAttrib.colorType = NVTX_COLOR_ARGB; \
44  eventAttrib.color = colors[color_id]; \
45  eventAttrib.messageType = NVTX_MESSAGE_TYPE_ASCII; \
46  eventAttrib.message.ascii = name; \
47  nvtxRangePushEx(&eventAttrib); \
48 }
49 #define POP_NVTX_RANGE nvtxRangePop();
50 #else
51 #define PUSH_NVTX_RANGE(name,cid)
52 #define POP_NVTX_RANGE
53 
54 #endif // USE_NVTX
55 #endif // SKA_CHEETAH_CUDA_UTILS_NVTX_H