Skip to main content

Module split_qkv

Module split_qkv 

Source
Expand description

split_qkv op-diff harness — see crate::op_diff.

Splits a fused [tokens, q_dim + 2*kv_dim] projection into separate q/k/v buffers. Pure slicing (exact); compared output is [q, k, v] concatenated.

Structs§

SplitQkvOp