xla

fb373129 - [Pallas] Make gmm support bf16 (#7133)

[Pallas] Make gmm support bf16 (#7133) Summary: This pull request does: 1. make gmm support bf16, 2. don't visit_empty_groups for gmm, 3. make the reference gmm torchy. Test Plan: python test/test_gmm.py