torch_npu.npu.iteration_end

torch_npu.npu.iteration_end()
为了从prof图上区分step信息,每个step结束时打上结束标记