commit 5030d76acfcdd48492e988e3fc1aa19bebe9366a
author Adam Paszke <adam.paszke@gmail.com> Fri Sep 23 21:10:28 2016 -0700
committer Adam Paszke <adam.paszke@gmail.com> Fri Sep 23 21:10:28 2016 -0700
tree 801afb6e2594ba601fde1f1a7b344e91fbd429c9
parent c51e2c8b8cd8c216fca2fd3744185132e10c5f69

Reduce precision of CUDA blas tests

test/test_cuda.py

diff --git a/test/test_cuda.py b/test/test_cuda.py
index e144b22..e523c56 100644
--- a/test/test_cuda.py
+++ b/test/test_cuda.py
@@ -200,6 +200,9 @@
 custom_precision = {
     'addbmm': 1e-4,
     'addmm': 1e-4,
+    'addmv': 1e-4,
+    'addr': 1e-4,
+    'baddbmm': 1e-4,
     'rsqrt': 1e-4,
     'cumprod': 1e-4,
 }