Function: get_step_timestamp

Applicability

Product	Supported (√/x)
Atlas A3 training products/Atlas A3 inference products	√
Atlas A2 training products/Atlas A2 inference products	√
Atlas training products	√
Atlas inference products	√
Atlas 200I/500 A2 inference products	√

Function Usage

Sets the start timestamp and end timestamp of an iteration (or step), to display the profiling analysis result by step after training is implemented by using single-operator loading and execution APIs.

Prototype

C Prototype

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

Python Function

ret =acl.prof.get_step_timestamp(stepinfo, tag, stream)

Parameter Description

Parameter	Description
stepinfo	Int, step information. Call acl.prof.create_step_info to create data of the aclprofStepInfo type in advance.
tag	Int, step start timestamp and end timestamp. Pass the value ACL_STEP_START at the step start and the value ACL_STEP_END at the step end.
stream	Int, stream.

Return Value Description

Return Value	Description
ret	Int, error code. 0 indicates success. Other values indicate failure.

Parent topic: Setting Iteration Time in PyTorch Scenarios