Skip to content

Commit

Permalink
fix typo
Browse files Browse the repository at this point in the history
  • Loading branch information
zylo117 committed Dec 23, 2024
1 parent 9371cff commit 1d9fa32
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions axengine/axcl_session.py
Original file line number Diff line number Diff line change
Expand Up @@ -384,8 +384,8 @@ def run(self, output_names, input_feed, run_options=None):
for i, output_tensor in enumerate(self.mgroup_output_tensors[grp_id])
if self.get_outputs()[i].name in output_names]

print(f'[INFO] cost time in host to device: {cost_host_to_device * 1009:.3f}ms, '
f'inference: {cost_inference * 1009:.3f}ms, '
f'device to host: {cost_device_to_host * 1009:.3f}ms')
print(f'[INFO] cost time in host to device: {cost_host_to_device * 1000:.3f}ms, '
f'inference: {cost_inference * 1000:.3f}ms, '
f'device to host: {cost_device_to_host * 1000:.3f}ms')

return outputs

0 comments on commit 1d9fa32

Please sign in to comment.