Home
last modified time | relevance | path

Searched refs:free_mem_size (Results 1 – 13 of 13) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_memory_allocator.cc31 size_t free_size = CudaDriver::free_mem_size(); in Init()
81 auto free_size = free_mem_size(); in AllocDeviceMem()
100 size_t GPUMemoryAllocator::free_mem_size() { return std::min(CudaDriver::free_mem_size(), available… in free_mem_size() function in mindspore::device::gpu::GPUMemoryAllocator
Dgpu_memory_allocator.h37 size_t free_mem_size() override;
Dcuda_driver.h50 static size_t free_mem_size();
Dgpu_memory_manager.cc88 …size_t mem_size = FloatToSize(GPUMemoryAllocator::GetInstance().free_mem_size() * init_gpu_mem_rat… in MallocDeviceMemory()
Dgpu_kernel_runtime.cc428 size_t free_mem_size = GPUMemoryAllocator::GetInstance().free_mem_size(); in FetchMemUnitSize() local
430 free_mem_size = kValidMemoryRatio * free_mem_size; in FetchMemUnitSize()
431 unit_size = std::min(unit_size, free_mem_size); in FetchMemUnitSize()
Dcuda_driver.cc135 size_t CudaDriver::free_mem_size() { in free_mem_size() function in mindspore::device::gpu::CudaDriver
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_memory_pool.cc51 auto device_free_mem_size = free_mem_size(); in CalMemBlockAllocSize()
138 size_t AscendMemoryPool::free_mem_size() { in free_mem_size() function in mindspore::device::ascend::AscendMemoryPool
Dascend_memory_pool.h42 size_t free_mem_size() override;
Dascend_memory_manager.cc219 auto available_mem_size = AscendMemoryPool::GetInstance().free_mem_size() + in GetAvailableMemSize()
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/cpu/
Dcpu_memory_pool.h38 size_t free_mem_size() override;
Dcpu_memory_pool.cc92 size_t CPUMemoryPool::free_mem_size() { return GetSystemMemorySize("MemAvailable"); } in free_mem_size() function in mindspore::device::cpu::CPUMemoryPool
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/
Dmem_dynamic_allocator.h116 virtual size_t free_mem_size() = 0;
Dmem_dynamic_allocator.cc148 auto device_free_mem_size = free_mem_size(); in CalMemBlockAllocSize()