[TESTING] allclose fixup (#724)

This commit is contained in:
Philippe Tillet
2022-09-28 15:49:05 -07:00
committed by GitHub
parent 998fd5f9af
commit dad97528b2

View File

@@ -102,7 +102,6 @@ def allclose(x, y, tol=1e-2):
diff = abs(x - y) diff = abs(x - y)
x_max = torch.max(x) x_max = torch.max(x)
y_max = torch.max(y) y_max = torch.max(y)
tol = 1e-2
err = torch.max(diff) / torch.max(x_max, y_max) err = torch.max(diff) / torch.max(x_max, y_max)
return err <= tol return err <= tol