use torch.compile ca API for benchmarks (#149647)
Summary:
X-link: https://github.com/pytorch/pytorch/pull/149647
Approved by: https://github.com/jansel
Reviewed By: yangw-dev
Differential Revision: D71769080
fbshipit-source-id: c76c34cde5040dd653c3ad61a1a03ed8b67aa4aa
