xla

3b0097d2 - Fix op test for `nn.functional.scaled_dot_product_attention"nn.functi… (#8025)