xla

2675e689 - Add sm_scale in ragged attention kernel (#8771)