More info in assert all close

[pommedeterresautee]

To check outputs are similar to reference implementation we use torch all close function.
To help in debugging we may want to add more info, like what is done in Triton project:

• max diff
• max relative diff
• nb mismatch elements

Example from Triton:

E       AssertionError: 
E       Arrays are not almost equal to 1 decimals
E       
E       Mismatched elements: 700185 / 786432 (89%)
E       Max absolute difference: 6.9219756
E       Max relative difference: 12265.675
E        x: array([[[[ 8.0e-02, -4.7e-01, -4.6e-02, ...,  8.7e-01,  7.6e-01,
E                 -2.5e-01],
E                [ 7.6e-02, -3.2e-01,  2.3e-01, ...,  2.1e-01,  2.3e-01,...
E        y: array([[[[ 8.0e-02, -4.7e-01, -4.6e-02, ...,  8.7e-01,  7.6e-01,
E                 -2.5e-01],
E                [ 7.3e-02, -2.3e-01,  4.0e-01, ..., -2.0e-01, -9.3e-02,...