merged weight (16480, 640) torch.bfloat16
whole step                         583.5 us
merged in-proj 2560->16480         371.8 us
out_proj 6144->2560                141.8 us