Thiago Crepaldi

@kit1980 could you import this PR? There is a check complaining about "Meta Internal-Only Changes Check — The Diff and Pull Request are not in sync!"

> Feel free to add me as a reviewer when you need the PR to be reviewed. Thank you. The current draft tries to add a variant to `aten::device`, but...

> To resolve the CLA issue, make sure to add the email address of the commit to your GitHub profile (to see the actual email address, add .patch at the...

> **pull / linux-focal-py3-clang7-mobile-build / build (pull_request) ** Failing after 8m Sure, I can add unit test when I find out what is going on with the `pull / linux-focal-py3-clang7-mobile-build...

@pytorchbot merge

@davidberard98 gentle ping

@wschin This is DORT, right?

Actually, it doesn't seem to be related to `ONNXRT_PROVIDER`, because the issue happens before it is actually consumed. Even with `ONNXRT_PROVIDER` unset, but both model and data on cpu, everything...

> @thiagocrepaldi I think there's an easy fix. Add an overload with index here > > https://github.com/pytorch/pytorch/blob/931a4913b1c74697711cb0a51bb227b89ba27f24/torch/csrc/jit/runtime/register_prim_ops.cpp#L2290 > > . > cc @davidberard98 @eellison I am giving it a shot...

@eellison @davidberard98 https://github.com/pytorch/pytorch/pull/97023 seems to be good to go