Searched refs:AlignMemorySize (Results 1 – 8 of 8) sorted by relevance
29 size_t AlignMemorySize(size_t size) { in AlignMemorySize() function40 size_t GPULaunchkernel::AlignSizeForLaunchKernel(size_t size) { return AlignMemorySize(size); } in AlignSizeForLaunchKernel()
35 size_t AlignMemorySize(size_t size) { in AlignMemorySize() function62 auto align_size = AlignMemorySize(origin_size); in AllocateAllReduceAddr()
102 size_t aligned_size = (nccl_kernel_type_ != NCCL_ALL_REDUCE) ? size : AlignMemorySize(size); in Init()118 size_t aligned_size = (nccl_kernel_type_ != NCCL_ALL_REDUCE) ? size : AlignMemorySize(size); in Init()247 size_t AlignMemorySize(size_t size) const { in AlignMemorySize() function
25 static size_t AlignMemorySize(size_t size) { in AlignMemorySize() function61 size_t size = AlignMemorySize(output->Size()); in AssignOutputs()
52 size_t AlignMemorySize(size_t size) const override;
114 size_t AscendMemoryPool::AlignMemorySize(size_t size) const { in AlignMemorySize() function in mindspore::device::ascend::AscendMemoryPool
30 size_t align_size = AlignMemorySize(size); in AllocTensorMem()77 size_t DynamicMemPoolBestFit::AlignMemorySize(size_t size) const { in AlignMemorySize() function in mindspore::device::DynamicMemPoolBestFit
121 virtual size_t AlignMemorySize(size_t size) const;