Added test

Emilio Castillo · Emilio Castillo · commit eb88eabd91d5 · 2020-04-06T09:09:08.000Z
diff --git a/tests/cupy_tests/core_tests/test_reduction.py b/tests/cupy_tests/core_tests/test_reduction.py
@@ -139,3 +139,15 @@ def test_invalid_kernel_name(self):
         with self.assertRaisesRegex(ValueError, 'Invalid kernel name'):
             cupy.ReductionKernel(
                 'T x', 'T y', 'x', 'a + b', 'y = a', '0', name='1')
+
+
+@testing.gpu
+class TestLargeMultiDimReduction(
+        ReductionKernelTestBase, unittest.TestCase):
+
+    def test_large_dims_keep_kernels(self):
+        # This test creates a CUDA_ERROR_LAUNCH_OUT_OF_RESOURCES
+        # if the output array dims are not reduced
+        shape = (4, 3, 2, 4, 3, 2, 2)
+        axis = (1, 4, 3, 6)
+        self.check_int8_sum(shape, axis=axis, keepdims=True)
