Skip to content

Commit 9fabfb0

Browse files
Relax test tolerance in float32
1 parent 6c580f3 commit 9fabfb0

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/tensor/signal/test_conv.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -176,7 +176,8 @@ def test_convolve2d(kernel_shape, data_shape, mode, boundary, boundary_kwargs):
176176
scipy_convolve2d(
177177
data_val, kernel_val, mode=mode, boundary=boundary, **boundary_kwargs
178178
),
179-
atol=1e-6 if config.floatX == "float32" else 1e-8,
179+
atol=1e-5 if config.floatX == "float32" else 1e-13,
180+
rtol=1e-5 if config.floatX == "float32" else 1e-13,
180181
)
181182

182183
utt.verify_grad(lambda k: op(data_val, k).sum(), [kernel_val])

0 commit comments

Comments
 (0)