def testToCPUExecution(self): raw = np.random.rand(10, 10) x = tensor(raw, chunk_size=3, gpu=True) cx = to_cpu(x) res = self.executor.execute_tensor(cx, concat=True)[0] np.testing.assert_array_equal(res, raw)
def test_to_cpu(): x = tensor(np.random.rand(10, 10), chunk_size=3, gpu=True) cx = to_cpu(x) assert cx.dtype == x.dtype assert cx.order == x.order assert cx.op.gpu is False cx, x = tile(cx, x) assert cx.chunks[0].dtype == x.chunks[0].dtype assert cx.chunks[0].order == x.chunks[0].order assert cx.chunks[0].op.gpu is False
def testToCPU(self): x = tensor(np.random.rand(10, 10), chunk_size=3, gpu=True) cx = to_cpu(x) self.assertEqual(cx.dtype, x.dtype) self.assertEqual(cx.order, x.order) self.assertFalse(cx.op.gpu) cx.tiles() self.assertEqual(cx.chunks[0].dtype, x.chunks[0].dtype) self.assertEqual(cx.chunks[0].order, x.chunks[0].order) self.assertFalse(cx.chunks[0].op.gpu)