Home
last modified time | relevance | path

Searched refs:kMemAlignSize (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dmemory_manager.cc31 return (input_size + kMemAlignSize + kAlignBytes - 1) / kMemAlignSize * kMemAlignSize; in GetCommonAlignSize()
35 return (input_size + kMemAlignSize - 1) / kMemAlignSize * kMemAlignSize + kTwiceMemAlignSize; in GetCommunicationAlignSize()
93 address->communication_ptr_ = ptr - kMemAlignSize; in MallocOutputMem()
Dmemory_manager.h31 constexpr uint64_t kMemAlignSize = 512; variable
32 constexpr uint64_t kTwiceMemAlignSize = kMemAlignSize << 1;
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dkernel_build_ascend.cc142 …uto clean_item = (output_men_size.at(index) + kMemAlignSize + kAlignBytes) / kMemAlignSize * kMemA… in CalCleanZerosSize()
151 …clean_item = (workspace_men_sizes.at(index) + kMemAlignSize + kAlignBytes) / kMemAlignSize * kMemA… in CalCleanZerosSize()
Dascend_memory_manager.cc162 return alloc_address + kMemAlignSize; in MallocStaticMem()
194 return device_mem_base_ + offset + kMemAlignSize; in MallocDynamicMem()
215 return base_ptr + kMemAlignSize; in MallocCommunicationMemFromMemPool()
Dascend_bucket.cc79 uint8_t *origin_dev_addr = ar_input_addr_ - kMemAlignSize; in FreeAllDeviceMem()
84 uint8_t *origin_dev_addr = ar_output_addr_ - kMemAlignSize; in FreeAllDeviceMem()
/third_party/mindspore/mindspore/ccsrc/utils/
Dutils.h500 const size_t kMemAlignSize = 512; variable