Added more to jit_csr_matvec