char-fork/llama.cpp

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

Jeff Bolzvulkan: Programmatically add RoundingModeRTE to all shaders when the device supports it (#21572)1f30ac0ce

char/aliceai-t5-moe
587 B19 linesraw
1#include "types.glsl"
2
3#extension GL_EXT_shader_16bit_storage : require
4
5#include "rope_params.glsl"
6
7layout(local_size_x = 1, local_size_y = 256, local_size_z = 1) in;
8
9layout (binding = 0) readonly buffer X {A_TYPE rope_data_a[];};
10layout (binding = 1) readonly buffer Y {int rope_data_pos[];};
11layout (binding = 2) readonly buffer Z {float rope_data_ff[];};
12layout (binding = 3) writeonly buffer D {ROPE_D_TYPE rope_data_d[];};
13layout (binding = 4) readonly buffer I {uvec2 rope_data_i[];}; // indices for set_rows
14
15
16layout (push_constant) uniform parameter {
17    rope_params pc;
18};
19