TaskD Worker APIs
The user-side code calls this function to initialize TaskD Worker.
Table 1 Input parameters
| Parameter | Type | Description |
|---|---|---|
| rank_id | int | Global rank ID of the current training process. |
| upper_limit_of_disk_in_mb | int | Upper limit of storage space in the profiling folder that all training processes can use. The actual size fluctuates around this threshold, in MB. A non-negative value, defaulted to 5000. |
| framework | str | AI framework used by the job. |
| Return Value Type | Description |
|---|---|
| bool | Indicates whether the initialization is successful.
|
The user-side code calls this function to start TaskD Worker.
No input parameters.
| Parameter | Description |
|---|---|
| bool | Indicates whether the initialization is successful.
|
The user-side code calls this function to destroy the TaskD Worker communication resources. This function must be used after init_taskd_worker is called.
None
Table 1 Return value
| Return Value Type | Description |
|---|---|
| bool | Indicates whether the destruction was successful.
|