torch_npu.npu.mstx.mstx_range
API接口
torch_npu.npu.mstx.mstx_range(msg: str, stream = None)
功能描述
range装饰器，用来采集被装饰函数的range执行耗时。
参数说明
参数 |
输入/输出 |
说明 |
|---|---|---|
msg |
输入 |
打点携带信息字符串指针。 |
stream |
输入 |
用于执行打点任务的Stream。
|
返回值
无。
支持的型号
- Atlas 训练系列产品
- Atlas A2 训练系列产品
父主题： profiler