def setUp(self):
self.pool = memory.SingleDeviceMemoryPool(allocator=mock_alloc)
self.unit = memory._allocation_unit_size
self.stream = stream_module.Stream()
self.stream_ptr = self.stream.ptr
def setUp(self):
self.pool = memory.SingleDeviceMemoryPool()
def setUp(self):
self.pool = memory.SingleDeviceMemoryPool(allocator=mock_alloc)
def setUp(self):
self.pool = memory.SingleDeviceMemoryPool(allocator=mock_alloc)
self.unit = self.pool._allocation_unit_size
