xla

157e06e4 - Reapply lower randperm (#6482)

Reapply lower randperm (#6482) Also updates test infrastructure to copy upstream pytorch test_input_copy and test_put to call XLA required `mark_step` during tests.