示例#1
0
 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
示例#2
0
 def setUp(self):
     self.pool = memory.SingleDeviceMemoryPool()
示例#3
0
 def setUp(self):
     self.pool = memory.SingleDeviceMemoryPool(allocator=mock_alloc)
示例#4
0
 def setUp(self):
     self.pool = memory.SingleDeviceMemoryPool(allocator=mock_alloc)
     self.unit = self.pool._allocation_unit_size