torch_npu.npu.iteration_end

API接口

torch_npu.npu.iteration_end()

功能描述

为了从prof图上区分step信息，每个step结束时打上结束标记。