Explicit safe MLA RoPE primitives and consumers (#5944)