diff --git a/‎tests/test_cupy.py‎
Lines changed: 23 additions & 0 deletions b/‎tests/test_cupy.py‎
Lines changed: 23 additions & 0 deletions
@@ -0,0 +1,23 @@
+import unittest
+
+import cupy as cp
+
+from common import gpu_test
+
+
+class TestCupy(unittest.TestCase):
+    @gpu_test
+    def test_kernel(self):
+        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)