Add matrix_mul2 which uses shared memory