Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[AOTI][torchgen] Update NativeFunctionsGroup mapping #125962

Closed
wants to merge 2 commits into from

Conversation

desertfire
Copy link
Contributor

@desertfire desertfire commented May 10, 2024

Stack from ghstack (oldest at bottom):

Summary: When looking up for what backend call to use for a fallback op (see get_backend_index_for_aoti), sometimes we need to search for a NativeFunction's structured delegate. Previous str:NativeFunctionsGroup dict missed some cases, such as aten.index.Tensor, and that's why aten.index.Tensor was specified in the fallback_ops list but no C shim entry was generated for it. This PR uses a more robust OperatorName:NativeFunctionsGroup mapping.

Summary: When looking up for what backend call to use for a fallback op (see get_backend_index_for_aoti), sometimes we need to search for a NativeFunction's structured delegate. Previous str:NativeFunctionsGroup dict missed some cases, such as aten.index.Tensor, and that's why aten.index.Tensor was specified in the fallback_ops list but no C shim entry was generated for it. This PR uses a more robust OperatorName:NativeFunctionsGroup mapping.

[ghstack-poisoned]
Copy link

pytorch-bot bot commented May 10, 2024

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/125962

Note: Links to docs will display an error until the docs builds have been completed.

❌ 1 New Failure

As of commit b56175a with merge base 6ffc94f (image):

NEW FAILURE - The following job has failed:

This comment was automatically generated by Dr. CI and updates every 15 minutes.

desertfire added a commit that referenced this pull request May 10, 2024
Summary: When looking up for what backend call to use for a fallback op (see get_backend_index_for_aoti), sometimes we need to search for a NativeFunction's structured delegate. Previous str:NativeFunctionsGroup dict missed some cases, such as aten.index.Tensor, and that's why aten.index.Tensor was specified in the fallback_ops list but no C shim entry was generated for it. This PR uses a more robust OperatorName:NativeFunctionsGroup mapping.

ghstack-source-id: 929149a9582ef9dd50de90354694bb5a63efe571
Pull Request resolved: #125962
Summary: When looking up for what backend call to use for a fallback op (see get_backend_index_for_aoti), sometimes we need to search for a NativeFunction's structured delegate. Previous str:NativeFunctionsGroup dict missed some cases, such as aten.index.Tensor, and that's why aten.index.Tensor was specified in the fallback_ops list but no C shim entry was generated for it. This PR uses a more robust OperatorName:NativeFunctionsGroup mapping.

[ghstack-poisoned]
@desertfire desertfire added the topic: not user facing topic category label May 13, 2024
smalltalkman pushed a commit to smalltalkman/pytorch that referenced this pull request May 15, 2024
Summary: They appear in some unit tests.

Pull Request resolved: pytorch#126013
Approved by: https://github.com/chenyang78
ghstack dependencies: pytorch#125962
ZelboK pushed a commit to ZelboK/pytorch that referenced this pull request May 19, 2024
Summary: When looking up for what backend call to use for a fallback op (see get_backend_index_for_aoti), sometimes we need to search for a NativeFunction's structured delegate. Previous str:NativeFunctionsGroup dict missed some cases, such as aten.index.Tensor, and that's why aten.index.Tensor was specified in the fallback_ops list but no C shim entry was generated for it. This PR uses a more robust OperatorName:NativeFunctionsGroup mapping.
Pull Request resolved: pytorch#125962
Approved by: https://github.com/chenyang78
ZelboK pushed a commit to ZelboK/pytorch that referenced this pull request May 19, 2024
Summary: They appear in some unit tests.

Pull Request resolved: pytorch#126013
Approved by: https://github.com/chenyang78
ghstack dependencies: pytorch#125962
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

3 participants