16 Byte Alignment for FP8 GEMMs in XLA
CC @reedwm.
Pads the operands of FP8 GEMMs so that the sizes of non-batch dimensions are multiples of 16 and slices the results to the expected shape before padding.