char-fork/llama.cpp

git clone https://git.t4t.associates/char-fork/llama.cpp

Giuseppe Scrivanovulkan: implement ADD1, ARANGE, FILL, SOFTPLUS, STEP, ROUND, CEIL, FLOOR, TRUNC (#17319)7d77f0732

char/aliceai-t5-moe
422 B20 linesraw
1#version 450
2
3#include "generic_head.glsl"
4#include "types.glsl"
5
6layout(local_size_x = 512, local_size_y = 1, local_size_z = 1) in;
7
8layout (binding = 0) writeonly buffer D {D_TYPE data_d[];};
9
10void main() {
11    const uint i = gl_GlobalInvocationID.x;
12
13    if (i >= p.KX) {
14        return;
15    }
16
17    // p.param1 = start, p.param2 = step
18    float value = p.param1 + p.param2 * float(i);
19    data_d[i] = D_TYPE(value);
20}