Work around vdot not being implemented in pypy

inducer · Nov 20, 2015 · 425ca5e · 425ca5e
1 parent 5d54e91
commit 425ca5e
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/test/test_algorithm.py b/test/test_algorithm.py
@@ -346,7 +346,17 @@ def test_dot(ctx_factory):
 
             assert abs(dot_ab_gpu - dot_ab) / abs(dot_ab) < 1e-4
 
-            vdot_ab = np.vdot(a, b)
+            try:
+                vdot_ab = np.vdot(a, b)
+            except NotImplementedError:
+                import sys
+                is_pypy = '__pypy__' in sys.builtin_module_names
+                if is_pypy:
+                    print("PYPY: VDOT UNIMPLEMENTED")
+                    continue
+                else:
+                    raise
+
             vdot_ab_gpu = cl_array.vdot(a_gpu, b_gpu).get()
 
             assert abs(vdot_ab_gpu - vdot_ab) / abs(vdot_ab) < 1e-4