↓ 8 callersFunctionropeRoPE: apply rotary position embeddings to Q and K in-place. Kernel: `rope_forward(Q, K, positions, seq_len, num_q_heads, num_kv_heads, head_dim, rota
crates/spark-model/src/layers/ops/embeddings.rs:24
↓ 7 callersMethodstep_tile(
&self,
ctx: &CudaCtx,
q: u64,
k_pool: u64,
v_pool: u64,
tile
crates/spark-storage/src/tiled_attention.rs:181