For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).
Mojo function
row_mean_of_squares_gpu_block
def row_mean_of_squares_gpu_block[in_dtype: DType, out_dtype: DType, //, simd_width: Int, max_warps_per_block: Int, input_fn: def[width: Int](row: Int, col: Int) capturing thin -> SIMD[in_dtype, width], output_fn: def(row: Int, val: Scalar[out_dtype]) capturing thin -> None](num_cols_dev: Int32)