Exemplos de require_device_memory em Python

Exemplo n.º 1

 def _template(self, obj):
     self.assertTrue(driver.is_device_memory(obj))
     driver.require_device_memory(obj)
     if driver.USE_NV_BINDING:
         expected_class = driver.binding.CUdeviceptr
     else:
         expected_class = drvapi.cu_device_ptr
     self.assertTrue(isinstance(obj.device_ctypes_pointer, expected_class))

Exemplo n.º 2

Arquivo: test_cuda_memory.py Projeto: cpcloud/numba

 def _template(self, obj):
     self.assertTrue(driver.is_device_memory(obj))
     driver.require_device_memory(obj)
     self.assertTrue(isinstance(obj.device_ctypes_pointer,
                                drvapi.cu_device_ptr))

Exemplo n.º 3

Arquivo: test_cuda_memory.py Projeto: ChiLi90/LifetimeFit

 def _template(self, obj):
     self.assertTrue(driver.is_device_memory(obj))
     driver.require_device_memory(obj)
     self.assertTrue(isinstance(obj.device_ctypes_pointer,
                                drvapi.cu_device_ptr))