Fix torchao benchmarking script (#126736)
Summary:
As the title says.
X-link: https://github.com/pytorch/pytorch/pull/126736
Approved by: https://github.com/jerryzh168, https://github.com/huydhn
Reviewed By: DanilBaibak
Differential Revision: D57670005
Pulled By: xuzhao9
fbshipit-source-id: 954af8b13d17b014fa863906f2c6b138f81790c3
