xla icon indicating copy to clipboard operation
xla copied to clipboard

Lower `randperm`

Open miladm opened this issue 3 years ago • 4 comments

Lower randperm.generator_out

This op falls back to aten in PyTorch/XLA. This op is required for running Mask RCNN efficiently.

miladm avatar Feb 11 '22 08:02 miladm

@thisisalbertliang can you take a pass at this issue please?

miladm avatar Mar 28 '22 17:03 miladm

The PR for lowering aten::randperm.generator_out has been created.

thisisalbertliang avatar Apr 15 '22 02:04 thisisalbertliang

Re-assigning the issue to @vanbasten23 .

yeounoh avatar Jul 06 '22 18:07 yeounoh

I created a PR but there is lingering issue: CircleCI tests fails (in particular, test_index_copy_xla_* and test_put_xla_*).

vanbasten23 avatar Aug 17 '22 20:08 vanbasten23

I'm getting the same thing.

pt-xla-profiler: Op(s) not lowered: aten::randperm.generator_out, Please open a GitHub issue with the above op lowering requests.

opooladz avatar Mar 31 '23 01:03 opooladz