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