Python require_device_memoryの例

コード例 #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))

コード例 #2

ファイル: test_cuda_memory.py プロジェクト: 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))

コード例 #3

ファイル: test_cuda_memory.py プロジェクト: 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))