Cheetah - SKA - PSS - Prototype Time Domain Search Pipeline
cheetah
cuda_utils
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
Generated on Mon May 30 2022 15:36:01 for Cheetah - SKA - PSS - Prototype Time Domain Search Pipeline by
1.8.13