1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
|
//------------------------------------------------------------------------------
// GB_jit_kernel_iso_expand.c: expand an iso scalar into an array
//------------------------------------------------------------------------------
// SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2025, All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0
//------------------------------------------------------------------------------
GB_JIT_GLOBAL GB_JIT_KERNEL_ISO_EXPAND_PROTO (GB_jit_kernel) ;
GB_JIT_GLOBAL GB_JIT_KERNEL_ISO_EXPAND_PROTO (GB_jit_kernel)
{
GB_GET_CALLBACKS ;
GB_A_TYPE *restrict Z = (GB_A_TYPE *) X ;
GB_A_TYPE x = (* ((GB_A_TYPE *) scalar)) ;
int64_t p ;
#pragma omp parallel for num_threads(nthreads) schedule(static)
for (p = 0 ; p < n ; p++)
{
Z [p] = x ;
}
return (GrB_SUCCESS) ;
}
|