mirror of
https://github.com/ggerganov/whisper.cpp.git
synced 2025-06-23 11:11:26 +02:00
metal : fix typo in FA kernel comments (llama/13651)
This commit is contained in:
parent
ee3f177cba
commit
e04e8f1c79
@ -3255,7 +3255,7 @@ template<
|
|||||||
typename kd4x4_t, // key type in device memory
|
typename kd4x4_t, // key type in device memory
|
||||||
short nl_k,
|
short nl_k,
|
||||||
void (*deq_k)(device const kd4x4_t *, short, thread k4x4_t &),
|
void (*deq_k)(device const kd4x4_t *, short, thread k4x4_t &),
|
||||||
typename vd4x4_t, // key type in device memory
|
typename vd4x4_t, // value type in device memory
|
||||||
short nl_v,
|
short nl_v,
|
||||||
void (*deq_v)(device const vd4x4_t *, short, thread v4x4_t &),
|
void (*deq_v)(device const vd4x4_t *, short, thread v4x4_t &),
|
||||||
short DK, // K head size
|
short DK, // K head size
|
||||||
@ -3776,7 +3776,7 @@ template<
|
|||||||
typename kd4_t, // key type in device memory
|
typename kd4_t, // key type in device memory
|
||||||
short nl_k,
|
short nl_k,
|
||||||
void (*deq_k_t4)(device const kd4_t *, short, thread k4_t &),
|
void (*deq_k_t4)(device const kd4_t *, short, thread k4_t &),
|
||||||
typename vd4_t, // key type in device memory
|
typename vd4_t, // value type in device memory
|
||||||
short nl_v,
|
short nl_v,
|
||||||
void (*deq_v_t4)(device const vd4_t *, short, thread v4_t &),
|
void (*deq_v_t4)(device const vd4_t *, short, thread v4_t &),
|
||||||
short DK, // K head size
|
short DK, // K head size
|
||||||
|
Loading…
x
Reference in New Issue
Block a user