Fix to set _routine_accelerators

asi1024 · asi1024 · commit a4ab70ebbe6c · 2020-07-07T11:12:42.000Z
diff --git a/tests/cupy_tests/core_tests/test_ndarray_reduction.py b/tests/cupy_tests/core_tests/test_ndarray_reduction.py
@@ -223,11 +223,11 @@ def test_ptp_nan_imag(self, xp, dtype):
 class TestCubReduction(unittest.TestCase):
 
     def setUp(self):
-        self.old_accelerators = _accelerator.get_reduction_accelerators()
-        _accelerator.set_reduction_accelerators(['cub'])
+        self.old_accelerators = _accelerator.get_routine_accelerators()
+        _accelerator.set_routine_accelerators(['cub'])
 
     def tearDown(self):
-        _accelerator.set_reduction_accelerators(self.old_accelerators)
+        _accelerator.set_routine_accelerators(self.old_accelerators)
 
     @testing.for_contiguous_axes()
     @testing.for_all_dtypes(no_bool=True, no_float16=True)
