cuBLAS中的logger配置

2.4.26. cublasLoggerConfigure()

cublasStatus_t cublasLoggerConfigure(
    int             logIsOn,
    int             logToStdOut,
    int             logToStdErr,
    const char*     logFileName)
1
2
3
4
5

此函数在运行时配置日志记录。 除了这种类型的配置之外，还可以使用特殊的环境变量来配置日志记录，这些环境变量将由 libcublas 进行检查：

• CUBLAS_LOGINFO_DBG- 设置环境。 变量为“1”表示打开日志记录（默认情况下日志记录是关闭的）。
• CUBLAS_LOGDEST_DBG - 设置环境。 变量编码如何记录。 “stdout”、“stderr”分别表示将日志消息输出到stdout或stderr。 在另一种情况下，它指定文件的“文件名”。

参数

logIsOn

• 输入。 完全打开/关闭日志记录。 默认情况下是关闭的，但通过调用 cublasSetLoggerCallback 到用户定义的回调函数来打开。

logToStdOut

• 输入。 打开/关闭标准错误 I/O 流的日志记录。 默认情况下是关闭的。

logToStdErr

• 输入。 打开/关闭标准错误 I/O 流的日志记录。 默认情况下是关闭的。

日志文件名

• 输入。 打开/关闭日志记录到由其名称指定的文件系统中的文件。 cublasLogger 配置 logFileName 的拷贝内容。 如果您对这种类型的日志记录不感兴趣，您应该提供空指针。

返回

CUBLAS_STATUS_SUCCESS

• 成功。

2.4.27. cublasGetLoggerCallback()

cublasStatus_t cublasGetLoggerCallback(
    cublasLogCallback* userCallback)
1
2

此函数通过 cublasSetLoggerCallback 检索指向先前安装的自定义用户定义回调函数的函数指针，否则为零。

参数

用户回调

• 输出。 指向用户定义的回调函数的指针。

返回

CUBLAS_STATUS_SUCCESS

• 成功。

2.4.28. cublasSetLoggerCallback()

cublasStatus_t cublasSetLoggerCallback(
    cublasLogCallback   userCallback)
1
2

此函数通过 cublas C 公共 API 安装自定义用户定义的回调函数。

参数

userCallback

• 输入。 指向用户定义的回调函数的指针。

返回值

CUBLAS_STATUS_SUCCESS

• 成功。