Skip to content

[PyTorch][JIT] Don't refcount Type singletons #69579

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

Closed
wants to merge 13 commits into from

Conversation

swolchok
Copy link
Contributor

@swolchok swolchok commented Dec 8, 2021

Stack from ghstack:

This should help us avoid reference counting overhead on singleton Type subclasses without a major rewrite of the Type subsystem.

Differential Revision: D32923880

cc @ezyang @gchanan

This should help us avoid reference counting overhead on singleton Type subclasses without a major rewrite of the Type subsystem.

Differential Revision: [D32923880](https://our.internmc.facebook.com/intern/diff/D32923880/)

[ghstack-poisoned]
@facebook-github-bot
Copy link
Contributor

facebook-github-bot commented Dec 8, 2021

🔗 Helpful links

💊 CI failures summary and remediations

As of commit 5eb65da (more details on the Dr. CI page):


💚 💚 Looks good so far! There are no failures yet. 💚 💚


This comment was automatically generated by Dr. CI (expand for details).

Please report bugs/suggestions to the (internal) Dr. CI Users group.

Click here to manually regenerate this comment.

@facebook-github-bot facebook-github-bot added the oncall: jit Add this issue/PR to JIT oncall triage queue label Dec 8, 2021
@pytorch-probot
Copy link

pytorch-probot bot commented Dec 8, 2021

CI Flow Status

⚛️ CI Flow

Ruleset - Version: v1
Ruleset - File: https://github.com/pytorch/pytorch/blob/5eb65daf7ad54edbfe0e73d3675e2f00d405e287/.github/generated-ciflow-ruleset.json
PR ciflow labels: ciflow/default

Workflows Labels (bold enabled) Status
Triggered Workflows
linux-bionic-py3.7-clang9 ciflow/all, ciflow/cpu, ciflow/default, ciflow/linux, ciflow/noarch, ciflow/trunk ✅ triggered
linux-docs ciflow/all, ciflow/cpu, ciflow/default, ciflow/docs, ciflow/linux, ciflow/trunk ✅ triggered
linux-vulkan-bionic-py3.7-clang9 ciflow/all, ciflow/cpu, ciflow/default, ciflow/linux, ciflow/trunk, ciflow/vulkan ✅ triggered
linux-xenial-cuda11.3-py3.7-gcc7 ciflow/all, ciflow/cuda, ciflow/default, ciflow/linux, ciflow/trunk ✅ triggered
linux-xenial-cuda11.3-py3.7-gcc7-bazel-test ciflow/all, ciflow/bazel, ciflow/cpu, ciflow/default, ciflow/linux, ciflow/trunk ✅ triggered
linux-xenial-py3-clang5-mobile-build ciflow/all, ciflow/default, ciflow/linux, ciflow/mobile, ciflow/trunk ✅ triggered
linux-xenial-py3-clang5-mobile-custom-build-static ciflow/all, ciflow/default, ciflow/linux, ciflow/mobile, ciflow/trunk ✅ triggered
linux-xenial-py3.7-clang7-asan ciflow/all, ciflow/cpu, ciflow/default, ciflow/linux, ciflow/sanitizers, ciflow/trunk ✅ triggered
linux-xenial-py3.7-clang7-onnx ciflow/all, ciflow/cpu, ciflow/default, ciflow/linux, ciflow/onnx, ciflow/trunk ✅ triggered
linux-xenial-py3.7-gcc5.4 ciflow/all, ciflow/cpu, ciflow/default, ciflow/linux, ciflow/trunk ✅ triggered
linux-xenial-py3.7-gcc7 ciflow/all, ciflow/cpu, ciflow/default, ciflow/linux, ciflow/trunk ✅ triggered
linux-xenial-py3.7-gcc7-no-ops ciflow/all, ciflow/cpu, ciflow/default, ciflow/linux, ciflow/trunk ✅ triggered
pytorch-linux-xenial-py3-clang5-android-ndk-r19c-gradle-custom-build-single ciflow/all, ciflow/android, ciflow/cpu, ciflow/default, ciflow/linux, ciflow/trunk ✅ triggered
pytorch-linux-xenial-py3-clang5-android-ndk-r19c-gradle-custom-build-single-full-jit ciflow/all, ciflow/android, ciflow/cpu, ciflow/default, ciflow/linux, ciflow/trunk ✅ triggered
win-vs2019-cpu-py3 ciflow/all, ciflow/cpu, ciflow/default, ciflow/trunk, ciflow/win ✅ triggered
win-vs2019-cuda11.3-py3 ciflow/all, ciflow/cuda, ciflow/default, ciflow/trunk, ciflow/win ✅ triggered
Skipped Workflows
caffe2-linux-xenial-py3.7-gcc5.4 ciflow/all, ciflow/cpu, ciflow/linux, ciflow/trunk 🚫 skipped
docker-builds ciflow/all, ciflow/trunk 🚫 skipped
ios-12-5-1-arm64 ciflow/all, ciflow/ios, ciflow/macos, ciflow/trunk 🚫 skipped
ios-12-5-1-arm64-coreml ciflow/all, ciflow/ios, ciflow/macos, ciflow/trunk 🚫 skipped
ios-12-5-1-arm64-custom-ops ciflow/all, ciflow/ios, ciflow/macos, ciflow/trunk 🚫 skipped
ios-12-5-1-arm64-full-jit ciflow/all, ciflow/ios, ciflow/macos, ciflow/trunk 🚫 skipped
ios-12-5-1-arm64-metal ciflow/all, ciflow/ios, ciflow/macos, ciflow/trunk 🚫 skipped
ios-12-5-1-x86-64 ciflow/all, ciflow/ios, ciflow/macos, ciflow/trunk 🚫 skipped
ios-12-5-1-x86-64-coreml ciflow/all, ciflow/ios, ciflow/macos, ciflow/trunk 🚫 skipped
ios-12-5-1-x86-64-full-jit ciflow/all, ciflow/ios, ciflow/macos, ciflow/trunk 🚫 skipped
libtorch-linux-xenial-cuda10.2-py3.7-gcc7 ciflow/all, ciflow/cuda, ciflow/libtorch, ciflow/linux, ciflow/trunk 🚫 skipped
libtorch-linux-xenial-cuda11.3-py3.7-gcc7 ciflow/all, ciflow/cuda, ciflow/libtorch, ciflow/linux, ciflow/trunk 🚫 skipped
linux-bionic-cuda10.2-py3.9-gcc7 ciflow/all, ciflow/cuda, ciflow/linux, ciflow/slow, ciflow/trunk 🚫 skipped
linux-docs-push ciflow/all, ciflow/cpu, ciflow/linux, ciflow/scheduled 🚫 skipped
linux-xenial-cuda11.3-py3.7-gcc7-no-ops ciflow/all, ciflow/cuda, ciflow/linux, ciflow/trunk 🚫 skipped
macos-10-15-py3-arm64 ciflow/all, ciflow/macos, ciflow/trunk 🚫 skipped
macos-10-15-py3-lite-interpreter-x86-64 ciflow/all, ciflow/macos, ciflow/trunk 🚫 skipped
macos-11-py3-x86-64 ciflow/all, ciflow/macos, ciflow/trunk 🚫 skipped
parallelnative-linux-xenial-py3.7-gcc5.4 ciflow/all, ciflow/cpu, ciflow/linux, ciflow/trunk 🚫 skipped
periodic-libtorch-linux-bionic-cuda11.5-py3.7-gcc7 ciflow/all, ciflow/cuda, ciflow/libtorch, ciflow/linux, ciflow/scheduled 🚫 skipped
periodic-libtorch-linux-xenial-cuda11.1-py3.7-gcc7 ciflow/all, ciflow/cuda, ciflow/libtorch, ciflow/linux, ciflow/scheduled 🚫 skipped
periodic-linux-bionic-cuda11.5-py3.7-gcc7 ciflow/all, ciflow/cuda, ciflow/linux, ciflow/scheduled 🚫 skipped
periodic-linux-xenial-cuda10.2-py3-gcc7-slow-gradcheck ciflow/all, ciflow/cuda, ciflow/linux, ciflow/scheduled, ciflow/slow, ciflow/slow-gradcheck 🚫 skipped
periodic-linux-xenial-cuda11.1-py3.7-gcc7-debug ciflow/all, ciflow/cuda, ciflow/linux, ciflow/scheduled 🚫 skipped
periodic-win-vs2019-cuda11.1-py3 ciflow/all, ciflow/cuda, ciflow/scheduled, ciflow/win 🚫 skipped
periodic-win-vs2019-cuda11.5-py3 ciflow/all, ciflow/cuda, ciflow/scheduled, ciflow/win 🚫 skipped
pytorch-linux-xenial-py3-clang5-android-ndk-r19c-build ciflow/all, ciflow/android, ciflow/cpu, ciflow/linux, ciflow/trunk 🚫 skipped

You can add a comment to the PR and tag @pytorchbot with the following commands:
# ciflow rerun, "ciflow/default" will always be added automatically
@pytorchbot ciflow rerun

# ciflow rerun with additional labels "-l <ciflow/label_name>", which is equivalent to adding these labels manually and trigger the rerun
@pytorchbot ciflow rerun -l ciflow/scheduled -l ciflow/slow

For more information, please take a look at the CI Flow Wiki.

This should help us avoid reference counting overhead on singleton Type subclasses without a major rewrite of the Type subsystem.

Differential Revision: [D32923880](https://our.internmc.facebook.com/intern/diff/D32923880/)

[ghstack-poisoned]
swolchok added a commit that referenced this pull request Dec 8, 2021
Pull Request resolved: #69579

This should help us avoid reference counting overhead on singleton Type subclasses without a major rewrite of the Type subsystem.
ghstack-source-id: 145056617

Differential Revision: [D32923880](https://our.internmc.facebook.com/intern/diff/D32923880/)
This should help us avoid reference counting overhead on singleton Type subclasses without a major rewrite of the Type subsystem.

Differential Revision: [D32923880](https://our.internmc.facebook.com/intern/diff/D32923880/)

[ghstack-poisoned]
swolchok added a commit that referenced this pull request Dec 8, 2021
Pull Request resolved: #69579

This should help us avoid reference counting overhead on singleton Type subclasses without a major rewrite of the Type subsystem.
ghstack-source-id: 145145357

Differential Revision: [D32923880](https://our.internmc.facebook.com/intern/diff/D32923880/)
@suo
Copy link
Member

suo commented Dec 14, 2021

Should I review? Tests look sadge

@swolchok
Copy link
Contributor Author

Should I review? Tests look sadge

you can still review for broad direction. blocking on tests slows down the process.

@swolchok
Copy link
Contributor Author

Looks like there's some kind of CUDA build failure I don't understand (can we not do explicit destructor calls in CUDA? why is this code getting built for CUDA anyway?), and I have to patch the Python bridging stuff for the new classes.

This should help us avoid reference counting overhead on singleton Type subclasses without a major rewrite of the Type subsystem.

Differential Revision: [D32923880](https://our.internmc.facebook.com/intern/diff/D32923880/)

[ghstack-poisoned]
swolchok added a commit that referenced this pull request Dec 14, 2021
Pull Request resolved: #69579

This should help us avoid reference counting overhead on singleton Type subclasses without a major rewrite of the Type subsystem.
ghstack-source-id: 145671620

Differential Revision: [D32923880](https://our.internmc.facebook.com/intern/diff/D32923880/)
This should help us avoid reference counting overhead on singleton Type subclasses without a major rewrite of the Type subsystem.

Differential Revision: [D32923880](https://our.internmc.facebook.com/intern/diff/D32923880/)

[ghstack-poisoned]
swolchok added a commit that referenced this pull request Dec 15, 2021
Pull Request resolved: #69579

This should help us avoid reference counting overhead on singleton Type subclasses without a major rewrite of the Type subsystem.
ghstack-source-id: 145771095

Differential Revision: [D32923880](https://our.internmc.facebook.com/intern/diff/D32923880/)
This should help us avoid reference counting overhead on singleton Type subclasses without a major rewrite of the Type subsystem.

Differential Revision: [D32923880](https://our.internmc.facebook.com/intern/diff/D32923880/)

[ghstack-poisoned]
This should help us avoid reference counting overhead on singleton Type subclasses without a major rewrite of the Type subsystem.

Differential Revision: [D32923880](https://our.internmc.facebook.com/intern/diff/D32923880/)

[ghstack-poisoned]
swolchok added a commit that referenced this pull request Dec 16, 2021
Pull Request resolved: #69579

This should help us avoid reference counting overhead on singleton Type subclasses without a major rewrite of the Type subsystem.
ghstack-source-id: 145793691

Differential Revision: [D32923880](https://our.internmc.facebook.com/intern/diff/D32923880/)
This should help us avoid reference counting overhead on singleton Type subclasses without a major rewrite of the Type subsystem.

Differential Revision: [D32923880](https://our.internmc.facebook.com/intern/diff/D32923880/)

[ghstack-poisoned]
swolchok added a commit that referenced this pull request Dec 16, 2021
Pull Request resolved: #69579

This should help us avoid reference counting overhead on singleton Type subclasses without a major rewrite of the Type subsystem.
ghstack-source-id: 145813052

Differential Revision: [D32923880](https://our.internmc.facebook.com/intern/diff/D32923880/)
Comment on lines 236 to 241
// Without SharedPtrWrapper, this line would read
// `shared_.~shared_ptr()` and nvcc would complain with
// "error: expected primary-expression before '>' token"
// referring to the "t" in "shared_ptr". SharedPtrWrapper
// exists to work around this compiler bug.
shared_.~SharedPtrWrapper();
Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This should help us avoid reference counting overhead on singleton Type subclasses without a major rewrite of the Type subsystem.

Differential Revision: [D32923880](https://our.internmc.facebook.com/intern/diff/D32923880/)

[ghstack-poisoned]
This should help us avoid reference counting overhead on singleton Type subclasses without a major rewrite of the Type subsystem.

Differential Revision: [D32923880](https://our.internmc.facebook.com/intern/diff/D32923880/)

[ghstack-poisoned]
swolchok added a commit that referenced this pull request Dec 16, 2021
Pull Request resolved: #69579

This should help us avoid reference counting overhead on singleton Type subclasses without a major rewrite of the Type subsystem.
ghstack-source-id: 145875239

Differential Revision: [D32923880](https://our.internmc.facebook.com/intern/diff/D32923880/)

**NOTE FOR REVIEWERS**: This PR has internal Facebook specific changes or comments, please review them on [Phabricator](https://our.internmc.facebook.com/intern/diff/D32923880/)!
This should help us avoid reference counting overhead on singleton Type subclasses without a major rewrite of the Type subsystem.

Differential Revision: [D32923880](https://our.internmc.facebook.com/intern/diff/D32923880/)

[ghstack-poisoned]
swolchok added a commit that referenced this pull request Jan 4, 2022
Pull Request resolved: #69579

This should help us avoid reference counting overhead on singleton Type subclasses without a major rewrite of the Type subsystem.
ghstack-source-id: 146468426

Differential Revision: [D32923880](https://our.internmc.facebook.com/intern/diff/D32923880/)

**NOTE FOR REVIEWERS**: This PR has internal Facebook specific changes or comments, please review them on [Phabricator](https://our.internmc.facebook.com/intern/diff/D32923880/)!
This should help us avoid reference counting overhead on singleton Type subclasses without a major rewrite of the Type subsystem.

Differential Revision: [D32923880](https://our.internmc.facebook.com/intern/diff/D32923880/)

[ghstack-poisoned]
swolchok added a commit that referenced this pull request Jan 5, 2022
Pull Request resolved: #69579

This should help us avoid reference counting overhead on singleton Type subclasses without a major rewrite of the Type subsystem.
ghstack-source-id: 146558363

Differential Revision: [D32923880](https://our.internmc.facebook.com/intern/diff/D32923880/)

**NOTE FOR REVIEWERS**: This PR has internal Facebook specific changes or comments, please review them on [Phabricator](https://our.internmc.facebook.com/intern/diff/D32923880/)!
This should help us avoid reference counting overhead on singleton Type subclasses without a major rewrite of the Type subsystem.

Differential Revision: [D32923880](https://our.internmc.facebook.com/intern/diff/D32923880/)

[ghstack-poisoned]
swolchok added a commit that referenced this pull request Jan 6, 2022
Pull Request resolved: #69579

This should help us avoid reference counting overhead on singleton Type subclasses without a major rewrite of the Type subsystem.
ghstack-source-id: 146643993

Differential Revision: [D32923880](https://our.internmc.facebook.com/intern/diff/D32923880/)

**NOTE FOR REVIEWERS**: This PR has internal Facebook specific changes or comments, please review them on [Phabricator](https://our.internmc.facebook.com/intern/diff/D32923880/)!
@swolchok swolchok added the module: bc-breaking Related to a BC-breaking change label Jan 7, 2022
@swolchok
Copy link
Contributor Author

swolchok commented Jan 7, 2022

I've conservatively labeled this bc-breaking because it is possible for it to cause compile errors in downstream code that tried to use std::dynamic_pointer_cast on TypePtrs instead of going through the Type::cast API. I ran into one example of this internally and the fix was to just use Type::cast.

@facebook-github-bot facebook-github-bot deleted the gh/swolchok/419/head branch January 10, 2022 15:17
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
cla signed module: bc-breaking Related to a BC-breaking change oncall: jit Add this issue/PR to JIT oncall triage queue
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants