Searched refs:variable_memory_max_size (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/tests/ut/python/pynative_mode/ |
| D | test_context.py | 88 context.set_context(variable_memory_max_size=True)
90 context.set_context(variable_memory_max_size=1)
91 context.set_context(variable_memory_max_size="1G")
92 context.set_context.__wrapped__(variable_memory_max_size="3GB")
|
| /third_party/mindspore/mindspore/ |
| D | context.py | 232 def set_variable_memory_max_size(self, variable_memory_max_size): argument
234 if not Validator.check_str_by_regular(variable_memory_max_size, _re_pattern):
236 if int(variable_memory_max_size[:-2]) > _DEVICE_APP_MEMORY_SIZE:
238 variable_memory_max_size_ = variable_memory_max_size[:-2] + " * 1024 * 1024 * 1024"
239 graph_memory_max_size = _DEVICE_APP_MEMORY_SIZE - int(variable_memory_max_size[:-2])
241 self.set_param(ms_ctx_param.variable_memory_max_size, variable_memory_max_size_)
513 @args_unreset_check(device_id=int, variable_memory_max_size=str, max_device_memory=str)
516 save_dump_path=str, enable_reduce_precision=bool, variable_memory_max_size=str,
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_memory_manager.cc | 85 auto variable_memory_max_size = context->get_param<std::string>(MS_CTX_VARIABLE_MEMORY_MAX_SIZE); in GetDeviceMemSizeFromContext() local
86 if (variable_memory_max_size == "0") { in GetDeviceMemSizeFromContext()
89 MS_LOG(INFO) << "context variable_memory_max_size:" << variable_memory_max_size; in GetDeviceMemSizeFromContext()
90 auto pos = variable_memory_max_size.find('*'); in GetDeviceMemSizeFromContext()
94 auto gb_str = variable_memory_max_size.substr(0, pos); in GetDeviceMemSizeFromContext()
|
| /third_party/mindspore/tests/st/ops/ascend/test_tbe_ops/ |
| D | test_relu.py | 23 context.set_context(device_target="Ascend", mode=context.GRAPH_MODE, variable_memory_max_size="31GB…
|