def setUp(self):
self.pool = memory.MemoryAsyncPool()
self.unit = memory._allocation_unit_size
self.stream = stream_module.Stream()
self.stream_ident = self.stream.ptr
cupy.get_default_memory_pool().free_all_blocks()
cupy.cuda.Device().synchronize()
def setUp(self):
if cupy.cuda.runtime.deviceGetAttribute(
cupy.cuda.runtime.cudaDevAttrMemoryPoolsSupported, 0) == 0:
pytest.skip('malloc_async is not supported on device 0')
self.pool = memory.MemoryAsyncPool()
self.unit = memory._allocation_unit_size
self.stream = stream_module.Stream()
self.stream_ident = self.stream.ptr
cupy.get_default_memory_pool().free_all_blocks()
cupy.cuda.Device().synchronize()
def setUp(self):
self.pool = memory.MemoryAsyncPool()
cupy.get_default_memory_pool().free_all_blocks()
cupy.cuda.Device().synchronize()