fix: constexpr HD for template param

This commit is contained in:
2026-05-28 08:01:18 +00:00
parent 7436315309
commit 3549a2388b

View File

@@ -29,7 +29,7 @@ float cosine_sim(const float* a, const float* b, int n) {
int main() {
printf("=== tcgen05.mma QK GEMM Verification ===\n\n");
int HD = 64;
constexpr int HD = 64;
int s_k = 128;
int B = 1, H = 1;
float scale = 1.0f / sqrtf((float)HD);