mynvtx.h

/*
 * mynvtx.h
 *
 *  Created on: 11 déc. 2024
 *      Author: rb263871
 */

#ifndef MYNVTX_H_
#define MYNVTX_H_

#ifdef __CUDACC__

#include <nvtx3/nvToolsExt.h>
#include <string.h>

/*
 * Function to convert a color name (e.g. "red") to a 32-bit ARGB color.
 * Adjust or add more colors if needed.
 */
static inline uint32_t colorFromName(const char *colorName)
{
    uint32_t color = 0xFFFFFFFF; // Default white: ARGB = FF FF FF FF
    if (strcasecmp(colorName, "red") == 0)
    {
        color = 0xFFFF0000; // Red
    }
    else if (strcasecmp(colorName, "green") == 0)
    {
        color = 0xFF00FF00; // Green
    }
    else if (strcasecmp(colorName, "blue") == 0)
    {
        color = 0xFF0000FF; // Blue
    }
    else if (strcasecmp(colorName, "white") == 0)
    {
        color = 0xFFFFFFFF; // White
    }
    else if (strcasecmp(colorName, "black") == 0)
    {
        color = 0xFF000000; // Black
    }
    else if (strcasecmp(colorName, "yellow") == 0)
    {
        color = 0xFFFFFF00; // Yellow
    }
    else if (strcasecmp(colorName, "cyan") == 0)
    {
        color = 0xFF00FFFF; // Cyan
    }
    else if (strcasecmp(colorName, "magenta") == 0)
    {
        color = 0xFFFF00FF; // Magenta
    }
    else if (strcasecmp(colorName, "gray") == 0)
    {
        color = 0xFF808080; // Gray
    }
    else if (strcasecmp(colorName, "orange") == 0)
    {
        color = 0xFFFFA500; // Orange
    }
    else if (strcasecmp(colorName, "purple") == 0)
    {
        color = 0xFF800080; // Purple
    }
    else if (strcasecmp(colorName, "pink") == 0)
    {
        color = 0xFFFFC0CB; // Pink
    }
    else
    {
        std::cout<<"mynvtx ERROR: unsupported color:"<<colorName<<std::endl;
    }

    return color;
}

/*
 * Helper function to create NVTX event attributes given a message and color.
 */
static inline nvtxEventAttributes_t createEventAttributes(const char *msg, const char *colorName)
{
    nvtxEventAttributes_t eventAttrib = {0};
    eventAttrib.version = NVTX_VERSION;
    eventAttrib.size = NVTX_EVENT_ATTRIB_STRUCT_SIZE;

    eventAttrib.messageType = NVTX_MESSAGE_TYPE_ASCII;
    eventAttrib.message.ascii = msg;

    eventAttrib.colorType = NVTX_COLOR_ARGB;
    eventAttrib.color = colorFromName(colorName);

    return eventAttrib;
}

/*
 * This is the wrapper function that you’ll call.
 * Example usage: mynvtxRangePush("myMessage", red);
 */
static inline void mynvtxRangePush(const char *msg, const char *colorName="black")
{
    nvtxEventAttributes_t attrib = createEventAttributes(msg, colorName);
    nvtxRangePushEx(&attrib);
}

//Wrapper if inputs are std::string
static inline void mynvtxRangePush(const std::string msg_string, const std::string color_string="black")
{
  mynvtxRangePush(msg_string.c_str(), color_string.c_str());
}

/*
 * Corresponding pop for the NVTX range.
 * You must call nvtxRangePop() when you finish.
 */
static inline void mynvtxRangePop()
{
    nvtxRangePop();
}

static inline void mynvtxRangePop(const char * msg)
{
    nvtxRangePop();
}

static inline void mynvtxRangePop(const std::string msg_string)
{
    nvtxRangePop();
}

#else // __CUDACC__

//If no cuda compiler, do nothing
static inline void mynvtxRangePush(const char *msg, const char *colorName="black"){}
static inline void mynvtxRangePush(const std::string msg_string, const std::string color_string){}
static inline void mynvtxRangePop(){}
static inline void mynvtxRangePop(const char * msg){}
static inline void mynvtxRangePop(const std::string msg_string){}



#endif //__CUDACC__
#endif /* MYNVTX_H_ */