- fix jit::matmul bug 

input x, shape(m, k), weight, shape(k, n)