函数:get_step_timestamp
产品支持情况
产品 |
是否支持 |
---|---|
√ |
|
√ |
|
√ |
|
√ |
|
√ |
|
√ |
功能说明
利用单算子加载与执行接口实现训练的场景下,使用本接口用于标记迭代开始与结束时间,为后续Profiling解析提供迭代标识,以便以迭代为粒度展示性能数据。
函数原型
- C函数原型
1
aclError aclprofGetStepTimestamp(aclprofStepInfo* stepInfo, aclprofStepTag tag, aclrtStream stream)
- python函数
1
ret =acl.prof.get_step_timestamp(stepinfo, tag, stream)
参数说明
参数名 |
说明 |
---|---|
stepinfo |
int,指定迭代信息。需提前调用acl.prof.create_step_info接口创建aclprofStepInfo类型的数据。 |
tag |
int,用于标记迭代开始或结束。在迭代开始时传入ACL_STEP_START,迭代结束时需传入ACL_STEP_END。 |
stream |
int,指定Stream。 |
返回值说明
返回值 |
说明 |
---|---|
ret |
int,错误码。
|
父主题: PyTorch场景标记迭代时间