[Pytorch benchmark] Add BMM benchmark (#59595)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/59595
ghstack-source-id: 130946743
Test Plan: bmm_test
Reviewed By: mingzhe09088
Differential Revision: D28873228
fbshipit-source-id: 6e4cb04bb6c63f5f68d8f23c13738e2d58ab499c
