[hotfix] dropout test tolerance (#170)

facebookresearch · Jan 3, 2022 · 154b819 · 154b819
1 parent 9be9167
commit 154b819
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_triton_dropout.py b/tests/test_triton_dropout.py
@@ -28,7 +28,7 @@
 
 # Testing odd (non-power-of-two for instance) shapes on purpose
 SHAPES = [
-    (384, 128),
+    (384, 512),
     (8, 384, 128),
     (8, 784, 512),
     (4, 16, 384),