torch_npu.npu.mstx.range_start
API接口
torch_npu.npu.mstx.range_start(msg: str, stream = None) -> int
功能描述
标识打点开始。
参数说明
参数 |
输入/输出 |
说明 |
|---|---|---|
msg |
输入 |
打点携带信息字符串指针。 |
stream |
输入 |
用于执行打点任务的Stream。
|
返回值
range_id：用于标识该range；如果接口执行失败，返回0。
支持的型号
- Atlas 训练系列产品
- Atlas A2 训练系列产品
父主题： profiler