benchmark

9806a7ec - replace torchbench impelmentation with torchrec that shards the module (#1401)

replace torchbench impelmentation with torchrec that shards the module (#1401) Summary: Pull Request resolved: https://github.com/pytorch/benchmark/pull/1401 as title, actually shard the module using composable API Reviewed By: xuzhao9 Differential Revision: D43105741 fbshipit-source-id: c74d93aad9997b66f4a11a6c7e068c840d1a0a13