make a test pass by accepting a higher (rtol=5e-6) diff threshold between cpu and gpu. This is expected as the rounding and order are not the same on cpu and gpu.