> torch.testing.assert_close(ref_out, output, atol=1e-3, rtol=1e-3)
E AssertionError: Tensor-likes are not close!
E
E Mismatched elements: 2069 / 524288 (0.4%)
E Greatest absolute difference: 54.87181854248047 at index (399, 467) (up to 0.001 allowed)
E Greatest relative difference: 40.02494812011719 at index (143, 159) (up to 0.001 allowed)