fix: corrected number of heads in multi-head attention shader.

b4rtaz · b4rtaz · commit a586ec7526b9 · 2025-08-08T00:34:55.000+02:00
diff --git a/src/nn/nn-vulkan.cpp b/src/nn/nn-vulkan.cpp
@@ -518,7 +518,7 @@ static void resolveShaderGroups(const NnOpConfig *opConfig, const NnUint batchSi
         }
     }
     else if (opConfig->code == OP_MULTIHEAD_ATT)
-        groupCount[2] = ((NnMultiHeadAttOpConfig *)opConfig->config)->nHeads;
+        groupCount[2] = ((NnMultiHeadAttOpConfig *)opConfig->config)->nHeads0;
 }
 
 static std::vector<uint32_t> readShader(const char *fileName) {

Original file line number	Diff line number	Diff line change
`@@ -518,7 +518,7 @@ static void resolveShaderGroups(const NnOpConfig *opConfig, const NnUint batchSi`
`518`	`518`	`}`
`519`	`519`	`}`
`520`	`520`	`else if (opConfig->code == OP_MULTIHEAD_ATT)`
`521`		`- groupCount[2] = ((NnMultiHeadAttOpConfig *)opConfig->config)->nHeads;`
	`521`	`+ groupCount[2] = ((NnMultiHeadAttOpConfig *)opConfig->config)->nHeads0;`
`522`	`522`	`}`
`523`	`523`
`524`	`524`	`static std::vector<uint32_t> readShader(const char *fileName) {`