Mojo function
apply_rope
apply_rope[dtype: DType, freq_dtype: DType, x_layout: Layout, rank: Int, width: Int, //, *, interleaved: Bool, alignment: Int, output_fn: fn[Int, Int](idx: IndexList[rank], val: SIMD[dtype, $0]) capturing -> None](x: LayoutTensor[dtype, x_layout, MutableAnyOrigin], idx: IndexList[rank], freq_val: SIMD[freq_dtype, width])
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!