xla

e51af253 - [Kernel] add heuristic gmm block sizes choosing logic (#9289)