vrr/koboldcpp
2
0
Fork 0
mirror of https://github.com/LostRuins/koboldcpp.git synced 2026-10-03 19:45:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

vulkan: work around NV bug with argsort_large.comp (#28975)

Browse source
This commit is contained in:
Jeff Bolz 2026-09-16 18:40:07 -05:00 • committed by GitHub
parent fb27a525d2
commit 4bc272fd72
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 8 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

8
ggml/src/ggml-vulkan/vulkan-shaders/argsort_large.comp
View file

@ -27,6 +27,8 @@ layout (push_constant) uniform parameter {
uint inner_end;
} p;
shared int s;
void argsort(bool needs_bounds_check, const uint row) {
// bitonic sort
int col = int(gl_GlobalInvocationID.x);
@ -35,6 +37,12 @@ void argsort(bool needs_bounds_check, const uint row) {
const uint row_offset = row * p.ncols;
uint idx_offset = row * p.ncols_padded;
// workaround for NV driver/compiler bug - dummy use of shared memory
if (gl_LocalInvocationIndex == 0) {
s = 0;
col += s;
}
bool need_barrier = false;
// initialize indices