Mojo function
naive_epilogue_kernel
naive_epilogue_kernel[c_type: DType, c_shape: DimList, *, elementwise_lambda_fn: fn[dtype: DType, width: Int, *, alignment: Int = 1](IndexList[2], SIMD[dtype, width]) capturing -> None](c: NDBuffer[c_type, 2, MutableAnyOrigin, c_shape])
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!