Align run.py profile_one_step and run_one_step (#1296)
Summary:
Pull Request resolved: https://github.com/pytorch/benchmark/pull/1296
Align the print summary for profile_one_step and run_one_step, as well as fix the repeat arg in profile. repeat=1 will avoid repeat saves of profiling traces. This will also fix the table generated by prof.key_averages.
Test Plan: CI tests
Reviewed By: xuzhao9, FindHao
Differential Revision: D41170045
Pulled By: aaronenyeshi
fbshipit-source-id: 3f115670b38ccf27e9619096bcc022c32582d26b
