def testToGPUExecution(self):
raw = np.random.rand(10, 10)
x = tensor(raw, chunk_size=3)
gx = to_gpu(x)
res = self.executor.execute_tensor(gx, concat=True)[0]
np.testing.assert_array_equal(res.get(), raw)
def test_to_gpu():
x = tensor(np.random.rand(10, 10), chunk_size=3)
gx = to_gpu(x)
assert gx.dtype == x.dtype
assert gx.order == x.order
assert gx.op.gpu is True
gx, x = tile(gx, x)
assert gx.chunks[0].dtype == x.chunks[0].dtype
assert gx.chunks[0].order == x.chunks[0].order
assert gx.chunks[0].op.gpu is True
def testToGPU(self):
x = tensor(np.random.rand(10, 10), chunk_size=3)
gx = to_gpu(x)
self.assertEqual(gx.dtype, x.dtype)
self.assertEqual(gx.order, x.order)
self.assertTrue(gx.op.gpu)
gx.tiles()
self.assertEqual(gx.chunks[0].dtype, x.chunks[0].dtype)
self.assertEqual(gx.chunks[0].order, x.chunks[0].order)
self.assertTrue(gx.chunks[0].op.gpu)
