Some bugs with matrix multiplication kernel