Mojo function
padded_copy_kernel
padded_copy_kernel[dtype: DType, tensor_layout: Layout, simd_width: Int](input_tensor: LayoutTensor[dtype, tensor_layout, MutAnyOrigin], output_tensor: LayoutTensor[dtype, tensor_layout, MutAnyOrigin], 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!