fix: const not constexpr for SCALE
This commit is contained in:
@@ -53,7 +53,7 @@ void reference_attention(
|
||||
int main() {
|
||||
constexpr int HD = 64;
|
||||
constexpr int SK = 256; // 2 KV tiles
|
||||
constexpr float SCALE = 1.0f / sqrtf((float)HD);
|
||||
const float SCALE = 1.0f / sqrtf((float)HD);
|
||||
|
||||
// Allocate host data
|
||||
bf16_t h_q[HD], h_k[SK * HD], h_v[HD * SK];
|
||||
|
||||
Reference in New Issue
Block a user