def test_cumsum_complex128(self): x = np.array([1, 4j, 3, 2j, 8], np.complex128) x_gpu = gpuarray.to_gpu(x) c_gpu = misc.cumsum(x_gpu) assert np.allclose(c_gpu.get(), np.cumsum(x))
def test_cumsum_float64(self): x = np.array([1, 4, 3, 2, 8], np.float64) x_gpu = gpuarray.to_gpu(x) c_gpu = misc.cumsum(x_gpu) assert np.allclose(c_gpu.get(), np.cumsum(x))