Function flashSupported

flashSupported(q: MlxArray): boolean
Can the flash kernel handle this q? (head_dim supported, seq≥32, fp16/bf16.)
Parameters
- q: MlxArray
Returns boolean
- Defined in packages/inference/src/kernels/attention/flash.ts:645