Matrix Vector Multiplication

Perform multiplication of a matrix and a vector:

$$C[i] = \sum_{k=0}^{K-1} A[i][k] \cdot B[k]$$

Input:

• Matrix $A$ of size $M \times K$
• Vector $B$ of size $K \times 1$

Output:

• Vector $C = AB$ of size $M \times 1$

Notes:

• Matrix $\text{A}$ is stored in row-major order
• This problem is adapted from KernelBench