CANN/runtime PyTorch场景标记迭代时间
2026/5/9 15:18:03 网站建设 项目流程

19-04 PyTorch场景标记迭代时间

【免费下载链接】runtime本项目提供CANN运行时组件和维测功能组件。项目地址: https://gitcode.com/cann/runtime

本章节描述 PyTorch 场景下标记迭代时间的接口。

  • aclError aclprofGetStepTimestamp(aclprofStepInfo* stepInfo, aclprofStepTag tag, aclrtStream stream):利用单算子模型执行接口实现训练的场景下,使用本接口用于标记迭代开始与结束时间,为后续Profiling解析提供迭代标识,以便以迭代为粒度展示性能数据。

aclprofGetStepTimestamp

aclError aclprofGetStepTimestamp(aclprofStepInfo* stepInfo, aclprofStepTag tag, aclrtStream stream)

产品支持情况

产品是否支持
Ascend 950PR/Ascend 950DT
Atlas A3 训练系列产品/Atlas A3 推理系列产品
Atlas A2 训练系列产品/Atlas A2 推理系列产品

功能说明

利用单算子模型执行接口实现训练的场景下,使用本接口用于标记迭代开始与结束时间,为后续Profiling解析提供迭代标识,以便以迭代为粒度展示性能数据。

参数说明

参数名输入/输出说明
stepinfo输入指定迭代信息。需提前调用aclprofCreateStepInfo接口创建aclprofStepInfo类型的数据。
类型定义请参见aclprofStepInfo、aclprofStepTag和aclrtStream。
tag输入用于标记迭代开始或结束。在迭代开始时传入枚举值ACL_STEP_START,迭代结束时需传入枚举值ACL_STEP_END。
取值详见aclprofStepTag。
stream输入指定Stream。
取值详见aclrtStream。

返回值说明

返回0表示成功,返回其他值表示失败,请参见aclError。

【免费下载链接】runtime本项目提供CANN运行时组件和维测功能组件。项目地址: https://gitcode.com/cann/runtime

创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

需要专业的网站建设服务?

联系我们获取免费的网站建设咨询和方案报价,让我们帮助您实现业务目标

立即咨询