Expand the test of torch.bmm on CUDA (#47124)
Summary:
basically https://github.com/pytorch/pytorch/pull/47070, enabled on all CI with `ci-all`
Pull Request resolved: https://github.com/pytorch/pytorch/pull/47124
Reviewed By: ejguan
Differential Revision: D24735130
Pulled By: ngimel
fbshipit-source-id: c2124562a9f9d1caf24686e5d8a1106c79366233
