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 a0f3897 commit 933414cCopy full SHA for 933414c
ggml/src/ggml-vulkan/vulkan-shaders/rms_norm.comp
@@ -131,8 +131,12 @@ void main() {
131
rms_norm(num_blocks);
132
} else if (num_blocks > 16) {
133
rms_norm(32);
134
- } else if (num_blocks > 8) {
+ } else if (num_blocks > 12) {
135
rms_norm(16);
136
+ } else if (num_blocks > 10) {
137
+ rms_norm(12);
138
+ } else if (num_blocks > 8) {
139
+ rms_norm(10);
140
} else if (num_blocks > 4) {
141
rms_norm(8);
142
} else if (num_blocks == 4) {
0 commit comments