We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 130d0c9 commit 8dd19a4Copy full SHA for 8dd19a4
ggml/src/ggml-vulkan/vulkan-shaders/soft_max.comp
@@ -32,7 +32,7 @@ shared FLOAT_TYPE vals[BLOCK_SIZE];
32
void soft_max(uint num_iters) {
33
const uint tid = gl_LocalInvocationID.x;
34
const uint rowx = gl_WorkGroupID.z * 262144 + gl_WorkGroupID.y * 512 + gl_WorkGroupID.x;
35
- const uint rowy = rowx % p.KY;
+ const uint rowy = (p.KY > 0) ? (rowx % p.KY) : 0;
36
37
if (rowx >= p.nrows_x) {
38
return;
0 commit comments