Searched refs:device_mem_size (Results 1 – 2 of 2) sorted by relevance
31 void AscendMemoryPool::Init(uint8_t *device_mem_base, uint64_t device_mem_size, uint64_t dynamic_me… in Init() argument40 if (dynamic_mem_offset > device_mem_size) { in Init()42 << " exceed the device memory size: " << device_mem_size; in Init()44 set_device_mem_size(device_mem_size); in Init()45 set_device_mem_pool_offset(device_mem_size); in Init()126 …id AscendMemoryPool::set_device_mem_size(uint64_t device_mem_size) { device_mem_size_ = device_mem… in set_device_mem_size() argument
32 void Init(uint8_t *device_mem_base, uint64_t device_mem_size, uint64_t dynamic_mem_offset);36 void set_device_mem_size(uint64_t device_mem_size);