mirror of
https://github.com/microsoft/mscclpp.git
synced 2026-05-13 09:46:00 +00:00
76 lines
2.3 KiB
C++
76 lines
2.3 KiB
C++
/*************************************************************************
|
|
* Copyright (c) 2015-2022, NVIDIA CORPORATION. All rights reserved.
|
|
*
|
|
* See LICENSE.txt for license information
|
|
************************************************************************/
|
|
|
|
#ifndef MSCCLPP_DEBUG_H_
|
|
#define MSCCLPP_DEBUG_H_
|
|
|
|
#include "mscclpp.h"
|
|
#include <chrono>
|
|
#include <stdio.h>
|
|
#include <type_traits>
|
|
|
|
#include <limits.h>
|
|
#include <pthread.h>
|
|
#include <string.h>
|
|
|
|
// Conform to pthread and NVTX standard
|
|
#define MSCCLPP_THREAD_NAMELEN 16
|
|
|
|
typedef enum
|
|
{
|
|
MSCCLPP_LOG_NONE = 0,
|
|
MSCCLPP_LOG_VERSION = 1,
|
|
MSCCLPP_LOG_WARN = 2,
|
|
MSCCLPP_LOG_INFO = 3,
|
|
MSCCLPP_LOG_ABORT = 4,
|
|
MSCCLPP_LOG_TRACE = 5
|
|
} mscclppDebugLogLevel;
|
|
typedef enum
|
|
{
|
|
MSCCLPP_INIT = 1,
|
|
MSCCLPP_COLL = 2,
|
|
MSCCLPP_P2P = 4,
|
|
MSCCLPP_SHM = 8,
|
|
MSCCLPP_NET = 16,
|
|
MSCCLPP_GRAPH = 32,
|
|
MSCCLPP_TUNING = 64,
|
|
MSCCLPP_ENV = 128,
|
|
MSCCLPP_ALLOC = 256,
|
|
MSCCLPP_CALL = 512,
|
|
MSCCLPP_ALL = ~0
|
|
} mscclppDebugLogSubSys;
|
|
|
|
extern int mscclppDebugLevel;
|
|
extern uint64_t mscclppDebugMask;
|
|
extern pthread_mutex_t mscclppDebugLock;
|
|
extern FILE* mscclppDebugFile;
|
|
extern mscclppResult_t getHostName(char* hostname, int maxlen, const char delim);
|
|
|
|
void mscclppDebugDefaultLogHandler(const char* msg);
|
|
void mscclppDebugLog(mscclppDebugLogLevel level, unsigned long flags, const char* filefunc, int line, const char* fmt,
|
|
...) __attribute__((format(printf, 5, 6)));
|
|
mscclppResult_t mscclppDebugSetLogHandler(mscclppLogHandler_t handler);
|
|
|
|
// Let code temporarily downgrade WARN into INFO
|
|
extern thread_local int mscclppDebugNoWarn;
|
|
extern char mscclppLastError[];
|
|
|
|
#define WARN(...) mscclppDebugLog(MSCCLPP_LOG_WARN, MSCCLPP_ALL, __FILE__, __LINE__, __VA_ARGS__)
|
|
#define INFO(FLAGS, ...) mscclppDebugLog(MSCCLPP_LOG_INFO, (FLAGS), __func__, __LINE__, __VA_ARGS__)
|
|
#define TRACE_CALL(...) mscclppDebugLog(MSCCLPP_LOG_TRACE, MSCCLPP_CALL, __func__, __LINE__, __VA_ARGS__)
|
|
|
|
#ifdef ENABLE_TRACE
|
|
#define TRACE(FLAGS, ...) mscclppDebugLog(MSCCLPP_LOG_TRACE, (FLAGS), __func__, __LINE__, __VA_ARGS__)
|
|
extern std::chrono::steady_clock::time_point mscclppEpoch;
|
|
#else
|
|
#define TRACE(...)
|
|
#endif
|
|
|
|
void mscclppSetThreadName(pthread_t thread, const char* fmt, ...);
|
|
void mscclppSetThreadName(pthread_t thread, const char* fmt, ...);
|
|
|
|
#endif
|