Mojo function
dual_gemv_kernel
dual_gemv_kernel[c_type: DType, c_shape: DimList[c_shape.values], a_type: DType, a_shape: DimList[a_shape.values], b_type: DType, b_shape: DimList[b_shape.values], simd_width: UInt, tile_m: UInt, tile_n: UInt, num_threads: UInt, binary_lambda_fn: binary_fn_type, elementwise_lambda_fn: Optional[elementwise_epilogue_type] = None, s_type: DType = get_accum_type[c_type]()](c: NDBuffer[c_type, MutAnyOrigin, c_shape, DimList.create_unknown[2]()], a: NDBuffer[a_type, MutAnyOrigin, a_shape, DimList.create_unknown[2]()], b0: NDBuffer[b_type, MutAnyOrigin, b_shape, DimList.create_unknown[2]()], b1: NDBuffer[b_type, MutAnyOrigin, b_shape, DimList.create_unknown[2]()])
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!