Home
last modified time | relevance | path

Searched refs:MS_CTX_ENABLE_DYNAMIC_MEM_POOL (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_memory_manager.cc77 if (context_ptr->get_param<bool>(MS_CTX_ENABLE_DYNAMIC_MEM_POOL)) { in MallocDeviceMemory()
108 if (context_ptr->get_param<bool>(MS_CTX_ENABLE_DYNAMIC_MEM_POOL)) { in MallocStaticMem()
Dgpu_kernel_runtime.cc443 bool is_enable_dynamic_mem = context_ptr->get_param<bool>(MS_CTX_ENABLE_DYNAMIC_MEM_POOL); in AssignMemory()
462 bool is_enable_dynamic_mem = context_ptr->get_param<bool>(MS_CTX_ENABLE_DYNAMIC_MEM_POOL); in Run()
/third_party/mindspore/mindspore/core/utils/
Dms_context.cc79 set_param<bool>(MS_CTX_ENABLE_DYNAMIC_MEM_POOL, true); in MsContext()
Dms_context.h70 MS_CTX_ENABLE_DYNAMIC_MEM_POOL, enumerator