Searched defs:N_aligned (Results 1 – 2 of 2) sorted by relevance
141 size_t N_aligned, in oneShotAllReduceKernel()207 size_t N_aligned, in twoShotAllReduceKernel()363 size_t N_aligned, in hybridCubeMeshAllReduceKernel()462 size_t N_aligned, in getLaunchConfig()522 const size_t N_aligned = alignUp(input.numel(), numelPerWarp); in oneShotAllReduce() local585 size_t N_aligned = alignUp(input.numel(), worldSize_ * numelPerWarp); in twoShotAllReduce() local640 size_t N_aligned = alignUp(input.numel(), numelPerWarp); in hybridCubeMeshAllReduce() local
113 uint32_t N_aligned = api::utils::align_up(N, 4u); in rearrange_weights_dw() local197 uint32_t N_aligned = api::utils::align_up(N, 4u); in rearrange_weights_2d() local