std :: ml::gpu_mps
Function std::mps_gated_gelu_bf16_chained
int mps_gated_gelu_bf16_chained(
struct MpsBuf bufRaw,
&unsigned short out,
unsigned long rows,
unsigned long intermediateSize,
)bf16 counterpart of mps_gated_gelu_f32_chained above -- 'bufRaw'/'out' are raw bf16 bits, same fused GeGLU step. Lets a whole bf16 matmul -> gated-GELU -> matmul FFN run as one GPU round trip.