bump tolerances for flex attn test (#194000)

~~Shouldn't be used by flex itself but prevents the references from being too far off
Not in the spirit of the test to exercise TF32~~

corrected by claude, just bump tolerances instead

authored manually

Pull Request resolved: https://github.com/pytorch/pytorch/pull/194000
Approved by: https://github.com/drisspg