Mojo function
padded_copy_kernel
padded_copy_kernel[InputLayoutType: TensorLayout, input_origin: ImmutOrigin, OutputLayoutType: TensorLayout, output_origin: MutOrigin, dtype: DType, simd_width: Int](input_tensor: TileTensor[dtype, InputLayoutType, input_origin], output_tensor: TileTensor[dtype, OutputLayoutType, output_origin], rows_per_sm: Int, total_rows: Int, row_length: Int, scaled_row_length: Int)
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!