Merge pull request #792 from Kaggle/add-cupy-test

Add cupy tests

Author: rosbo

tests/test_cupy.py

@@ -0,0 +1,22 @@
+import unittest
+
+from common import gpu_test
+
+
+class TestCupy(unittest.TestCase):
+    @gpu_test
+    def test_kernel(self):
+        import cupy as cp
+        x = cp.arange(6, dtype='f').reshape(2, 3)
+        y = cp.arange(3, dtype='f')
+        kernel = cp.ElementwiseKernel(
+            'float32 x, float32 y', 'float32 z',
+            '''if (x - 2 > y) {
+                z = x * y;
+            } else {
+                z = x + y;
+            }''',
+            'my_kernel')
+        r = kernel(x, y)
+        
+        self.assertEqual((2, 3), r.shape)