Scheduling fixes on MPS (#10549)

Commit

1 year ago

Scheduling fixes on MPS (#10549) * use np.int32 in scheduling * test_add_noise_device * -np.int32, fixes