diff --git a/src/vulkan/matmul-q40-q80.comp b/src/vulkan/matmul-q40-q80.comp index 6c7d51b..fd6a0ff 100644 --- a/src/vulkan/matmul-q40-q80.comp +++ b/src/vulkan/matmul-q40-q80.comp @@ -6,7 +6,7 @@ #define FLOAT_TYPE float #define QK_40_80 32 -#define N_THREADS 32 +#define N_THREADS 16 layout(local_size_x = N_THREADS, local_size_y = 1, local_size_z = 1) in;