Compute follows data legalization pass crash

[diptorupd]

The following reproducer crashes the CFD legalization pass:

@dpjit
def foo2(a,b):
    b[:,:,0] = a


a1 = dpnp.arange(64)
a1 = a1.reshape(4,16)

print(a1)

b1 = dpnp.empty((4,16,4))

foo2(a1, b1)

print(b1)

print("---------------------------------")

a2 = numpy.arange(64)
a2 = a2.reshape(4,16)
b2 = dpnp.empty((4,16,4))

foo2(a2, b2)

print(b2)