[Pallas] xsa TPU sweep: use bfloat16 (was float16, hit Mosaic vector-type bug)#3204

norx1991 merged 1 commit into

mainpytorch/helion:mainfrom

yifeixu/xsa-bench-bf16pytorch/helion:yifeixu/xsa-bench-bf16Copy head branch name to clipboard