Implement FloatMatrixMul kernel