Add ncclDataType_t as type to ROCTX (#1512)

src/include/nvtx3/nvToolsExtPayload.h

@@ -254,6 +254,11 @@
 #define NVTX_PAYLOAD_ENTRY_TYPE_BF16       50
 #define NVTX_PAYLOAD_ENTRY_TYPE_TF32       52
 
+/**
+ * Generic data type
+ */
+#define NVTX_PAYLOAD_ENTRY_TYPE_DATATYPE   53
+
 /**
  * Data types are as defined by NVTXv3 core.
  */

src/include/roctx.h

@@ -32,6 +32,7 @@ enum roctxPayloadEntryType {
   ROCTX_PAYLOAD_ENTRY_TYPE_INT,
   ROCTX_PAYLOAD_ENTRY_TYPE_SIZE,
   ROCTX_PAYLOAD_ENTRY_TYPE_REDOP,
+  ROCTX_PAYLOAD_ENTRY_TYPE_DATATYPE,
   ROCTX_PAYLOAD_NUM_ENTRY_TYPES
 };
 
@@ -58,6 +59,7 @@ struct roctxPayloadSchemaEntryInfo {
     int typeInt;
     size_t typeSize;
     ncclDevRedOp_t typeRedOp;
+    ncclDataType_t typeDataType;
   } payload;
 };
 
@@ -88,6 +90,7 @@ typedef roctxPayloadInfo* roctxPayloadInfo_t;
 
 extern const char* roctxEntryTypeStr[ROCTX_PAYLOAD_NUM_ENTRY_TYPES];
 extern const char* ncclRedOpStr[ncclNumDevRedOps];
+extern const char* ncclDataTypeStr[ncclNumTypes];
 
 /**
  * \brief Maps nvtx types to roctx types.