Fix docker tag in push (#2362)
Summary:
We need to set docker tag in the push step.
Test workflow: https://github.com/pytorch/benchmark/actions/runs/9793666962/job/27042062455
Pull Request resolved: https://github.com/pytorch/benchmark/pull/2362
Reviewed By: atalman
Differential Revision: D59374131
Pulled By: xuzhao9
fbshipit-source-id: 048c278bbce0d48fa7d74688c4fa35092f3e3e82
