Searched refs:kStaticMem (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | memory_manager.cc | 89 if (type == kStaticMem) { in MallocOutputMem()
105 if (type == kStaticMem) { in MallocOutputMem()
129 if (type == kStaticMem) { in MallocMem()
|
| D | kernel_runtime.cc | 623 …if (mem_manager_->MallocMem(kStaticMem, tensor_size, device_address, graph.graph_id()) == nullptr)… in AssignStaticMemoryInput()
624 …MS_LOG(EXCEPTION) << "Cannot alloc address when flag is: " << kStaticMem << ", tensor size is: " <… in AssignStaticMemoryInput()
644 AssignCommunicationNodeMem(kStaticMem, kernel_with_index.first); in AssignStaticMemoryOutput()
653 AssignNodeOutputMem(kStaticMem, item_with_index.first, SizeToInt(item_with_index.second)); in AssignStaticMemoryOutput()
918 if (type == kStaticMem) { in AssignNodeOutputMem()
942 …uint8_t *ptr = mem_manager_->MallocOutputMem(node, index, kStaticMem, tensor_address->size(), devi… in AssignExtraStaticMem()
987 if (mem_manager_->MallocMem(kStaticMem, node_size, address, graph_id) == nullptr) { in AssignValueNodeTensor()
988 …MS_LOG(EXCEPTION) << "Cannot alloc address when flag is: " << kStaticMem << ", tensor size is: " <… in AssignValueNodeTensor()
1025 …if (mem_manager_->MallocMem(kStaticMem, device_address->size_, device_address, graph.graph_id())) { in AssignStaticMemoryValueNode()
1049 … if (mem_manager_->MallocMem(kStaticMem, tensor_size, address, graph.graph_id()) == nullptr) { in AssignStaticMemoryValueNode()
[all …]
|
| D | memory_manager.h | 29 enum MemType { kStaticMem, kDynamicMem, kSomasReuseDynamicMem }; enumerator
|
| D | kernel_adjust.cc | 836 …if (runtime_instance->MallocMem(kStaticMem, LongToSize(tensor->data().nbytes()), device_address) =… in InitCtrlInputs()
837 MS_LOG(EXCEPTION) << "Cannot alloc address when flag is : " << kStaticMem in InitCtrlInputs()
|