mirror of
https://github.com/deepseek-ai/FlashMLA
synced 2025-06-26 18:15:54 +00:00
14 lines
252 B
C++
14 lines
252 B
C++
#pragma once
|
|
|
|
namespace Config {
|
|
|
|
static constexpr int BLOCK_SIZE_M = 64;
|
|
static constexpr int PAGE_BLOCK_SIZE = 64;
|
|
|
|
static constexpr int HEAD_DIM_K = 576;
|
|
static constexpr int HEAD_DIM_V = 512;
|
|
|
|
static constexpr int FIXED_OVERHEAD_NUM_BLOCKS = 5;
|
|
|
|
}
|