Fix KQ mask padding for the Vulkan back-end (#2350)
This commit is contained in:
parent
64109a4d60
commit
c574620b12
|
|
@ -2483,7 +2483,11 @@ extern "C" {
|
|||
int nk,
|
||||
int topk_experts);
|
||||
|
||||
#if GGML_USE_VULKAN
|
||||
#define GGML_KQ_MASK_PAD 64
|
||||
#else
|
||||
#define GGML_KQ_MASK_PAD 16
|
||||
#endif
|
||||
|
||||
// q: [n_embd, n_batch, n_head, 1]
|
||||
// k: [n_embd, n_kv, n_head_kv, 1]
|
||||
|
|
|
|||
Loading…
Reference in New Issue