Exemplos de host_memory_extents em Python

Exemplo n.º 1

 def test_single_element_array(self):
     ary = np.asarray(np.uint32(1234))
     arysz = ary.dtype.itemsize
     s, e = driver.host_memory_extents(ary)
     self.assertTrue(ary.ctypes.data == s)
     self.assertTrue(arysz == driver.host_memory_size(ary))

Exemplo n.º 2

 def test_f_contiguous_array(self):
     ary = np.asfortranarray(np.arange(100).reshape(2, 50))
     arysz = ary.dtype.itemsize * np.prod(ary.shape)
     s, e = driver.host_memory_extents(ary)
     self.assertTrue(ary.ctypes.data == s)
     self.assertTrue(arysz == driver.host_memory_size(ary))

Exemplo n.º 3

 def test_c_contiguous_array(self):
     ary = np.arange(100)
     arysz = ary.dtype.itemsize * ary.size
     s, e = driver.host_memory_extents(ary)
     self.assertTrue(ary.ctypes.data == s)
     self.assertTrue(arysz == driver.host_memory_size(ary))

Exemplo n.º 4

Arquivo: test_cuda_memory.py Projeto: cpcloud/numba

 def test_single_element_array(self):
     ary = np.asarray(np.uint32(1234))
     arysz = ary.dtype.itemsize
     s, e = driver.host_memory_extents(ary)
     self.assertTrue(ary.ctypes.data == s)
     self.assertTrue(arysz == driver.host_memory_size(ary))

Exemplo n.º 5

Arquivo: test_cuda_memory.py Projeto: cpcloud/numba

 def test_f_contiguous_array(self):
     ary = np.asfortranarray(np.arange(100).reshape(2, 50))
     arysz = ary.dtype.itemsize * np.prod(ary.shape)
     s, e = driver.host_memory_extents(ary)
     self.assertTrue(ary.ctypes.data == s)
     self.assertTrue(arysz == driver.host_memory_size(ary))

Exemplo n.º 6

Arquivo: test_cuda_memory.py Projeto: cpcloud/numba

 def test_c_contiguous_array(self):
     ary = np.arange(100)
     arysz = ary.dtype.itemsize * ary.size
     s, e = driver.host_memory_extents(ary)
     self.assertTrue(ary.ctypes.data == s)
     self.assertTrue(arysz == driver.host_memory_size(ary))