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