llama.cpp/ggml/src/ggml-opencl/kernels/fill.cl


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

#pragma OPENCL EXTENSION cl_khr_fp16 : enable

//------------------------------------------------------------------------------
// fill
//------------------------------------------------------------------------------
__kernel void kernel_fill_f32(
        __global float *dst,
        ulong offsetd,
        float v,
        int n

) {
    dst = (global float*)((global char*)dst + offsetd);
    if(get_global_id(0) < n){
        dst[get_global_id(0)] = v;
    }
}