std :: ml::gpu_mps
Function std::mps_linear_attention_ex
int mps_linear_attention_ex(
struct MpsBuf q,
struct MpsBuf k,
struct MpsBuf v,
struct MpsBuf out,
struct MpsAttnShape s,
int dtype,
int causal,
)std :: ml::gpu_mps
int mps_linear_attention_ex(
struct MpsBuf q,
struct MpsBuf k,
struct MpsBuf v,
struct MpsBuf out,
struct MpsAttnShape s,
int dtype,
int causal,
)