migrate intraop benchmarks to the new interface (#21202)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/21202
Migrate Ilia's op benchmarks to the new interface
Reviewed By: hl475
Differential Revision: D15322577
fbshipit-source-id: 8e75d51e7ddacbd56896c55f2996a9358491d83e