Acly
e29acf74fe
* vulkan (DRAFT): split shader generation by GLSL source file, to improve incremental build times * support dep-files so shaders are recompiled if their included files change * rename shader files which are used as "headers" to use .glsl extension * move glslc extension detection shaders to separate folders * the above is to prevent them from getting glob'd with the actual compute shaders that need to be compiled * vulkan : only write embedded shader .hpp/.cpp when they change * avoid recompiling ggml-vulkan.cpp when editing shaders * pass single --source argument instead of --input-dir & --filter to shader gen * check for source file match earlier * fix hang in vulkan-shaders-gen when there are compilation errors * early out did not decrement compile_count * clean up * fix glslc integer dot product test * unconditionally write the embedded shader cpp output * replace output filepath in generated dep-files to match output in CMakeLists --------- Co-authored-by: Jeff Bolz <jbolz@nvidia.com>
71 lines
1.9 KiB
Plaintext
71 lines
1.9 KiB
Plaintext
#version 450
|
|
|
|
#include "types.glsl"
|
|
|
|
#extension GL_EXT_control_flow_attributes : enable
|
|
|
|
layout(local_size_x_id = 0, local_size_y = 1, local_size_z = 1) in;
|
|
|
|
layout (binding = 0) readonly buffer A {A_TYPE data_a[];};
|
|
layout (binding = 1) writeonly buffer D {D_TYPE data_d[];};
|
|
|
|
layout (constant_id = 0) const uint BLOCK_SIZE = 32;
|
|
|
|
layout (push_constant) uniform parameter
|
|
{
|
|
uint n_cols;
|
|
uint ne01, ne02;
|
|
uint nb01, nb02, nb03;
|
|
uint nb11, nb12, nb13;
|
|
float weight;
|
|
uint misalign_offsets;
|
|
uint ne0_12mp, ne0_12L;
|
|
uint ne0_1mp, ne0_1L;
|
|
} p;
|
|
|
|
uint get_aoffset() { return p.misalign_offsets >> 16; }
|
|
uint get_doffset() { return p.misalign_offsets & 0xFFFF; }
|
|
|
|
// see init_fastdiv_values in ggml-vulkan.cpp
|
|
uint fastdiv(uint n, uint mp, uint L) {
|
|
uint msbs, lsbs;
|
|
// msbs = mulhi(n, mp)
|
|
umulExtended(n, mp, msbs, lsbs);
|
|
return (msbs + n) >> L;
|
|
}
|
|
|
|
|
|
shared FLOAT_TYPE tmp[BLOCK_SIZE];
|
|
|
|
void main() {
|
|
const uint row = gl_WorkGroupID.z * 262144 + gl_WorkGroupID.y * 512 + gl_WorkGroupID.x;
|
|
const uint col = gl_LocalInvocationID.x;
|
|
const float weight = p.weight;
|
|
|
|
const uint i03 = fastdiv(row, p.ne0_12mp, p.ne0_12L);
|
|
const uint i03_offset = i03 * p.ne01*p.ne02;
|
|
const uint i02 = fastdiv(row - i03_offset, p.ne0_1mp, p.ne0_1L);
|
|
const uint i01 = row - i03_offset - i02*p.ne01;
|
|
|
|
const uint src_idx = get_aoffset() + i01 * p.nb01 + i02 * p.nb02 + i03 * p.nb03;
|
|
const uint dst_idx = get_doffset() + i01 * p.nb11 + i02 * p.nb12 + i03 * p.nb13;
|
|
|
|
tmp[col] = FLOAT_TYPE(0.0);
|
|
|
|
for (uint i = col; i < p.n_cols; i += BLOCK_SIZE) {
|
|
tmp[col] += FLOAT_TYPE(data_a[src_idx + i]);
|
|
}
|
|
|
|
barrier();
|
|
[[unroll]] for (int s = int(BLOCK_SIZE) / 2; s > 0; s >>= 1) {
|
|
if (col < s) {
|
|
tmp[col] += tmp[col + s];
|
|
}
|
|
barrier();
|
|
}
|
|
|
|
if (col == 0) {
|
|
data_d[dst_idx] = D_TYPE(tmp[0] * weight);
|
|
}
|
|
}
|