xla
xla copied to clipboard
Lower `randperm`
Lower randperm.generator_out
This op falls back to aten
in PyTorch/XLA. This op is required for running Mask RCNN efficiently.
@thisisalbertliang can you take a pass at this issue please?
The PR for lowering aten::randperm.generator_out
has been created.
Re-assigning the issue to @vanbasten23 .
I created a PR but there is lingering issue: CircleCI tests fails (in particular, test_index_copy_xla_* and test_put_xla_*).
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.