1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29
|
//------------------------------------------------------------------------------
// GB_eslice: equal partition of e items to each task
//------------------------------------------------------------------------------
// SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2022, All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0
//------------------------------------------------------------------------------
// There are e items to split between ntasks. Task tid will own items
// Slice [tid] to Slice [tid+1]-1.
#include "GB.h"
void GB_eslice
(
int64_t *Slice, // array of size ntasks+1
int64_t e, // number items to partition amongst the tasks
const int ntasks // # of tasks
)
{
Slice [0] = 0 ;
for (int tid = 1 ; tid < ntasks ; tid++)
{
Slice [tid] = (int64_t) GB_PART (tid, e, ntasks) ;
}
Slice [ntasks] = e ;
}
|