[op-bench] fix jit mode (#45774)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/45774
Fix RuntimeError: No such operator operator_benchmark::_consume
Test Plan: waitforsandcastle
Reviewed By: ngimel
Differential Revision: D24064982
fbshipit-source-id: 13160b6d18569e659ca1ab0ca1d444ed9947260c