func InitTaskMonitor(rank int, upperLimitOfDiskInMb int) C.int

功能说明

TaskD后端导出函数,由taskd前端自行调用,用户侧不调用。初始化TaskD monitor模块,包括初始化日志、加载功能so等。

输入参数说明

表1 输入参数说明

参数

类型

说明

rank

int

当前训练进程的global rank号。

upperLimitOfDiskInMb

int

所有训练进程能使用的profiling文件夹存储空间上限,实际大小在此阈值上下波动。

返回值说明

表2 返回值说明

返回值类型

说明

int

表明初始化是否成功。

0:初始化成功。

1:初始化失败。