-
Notifications
You must be signed in to change notification settings - Fork 26.3k
[PyTorch] [Model Tracer] Use c10::Synchronized<T> for kernel dtype tracer #74105
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
Conversation
…acer Currently, the kernel dtype tracer relies on the user to for advisory locking of the mutex when accessing shared data structures. This changes it to be mandatory locking using the `c10::Synchronized<T>` abstraction. Differential Revision: [D34822912](https://our.internmc.facebook.com/intern/diff/D34822912/) [ghstack-poisoned]
CI Flow Status⚛️ CI FlowRuleset - Version:
|
🔗 Helpful links
💊 CI failures summary and remediationsAs of commit 1853ab1 (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. |
…el dtype tracer" Currently, the kernel dtype tracer relies on the user to for advisory locking of the mutex when accessing shared data structures. This changes it to be mandatory locking using the `c10::Synchronized<T>` abstraction. Differential Revision: [D34822912](https://our.internmc.facebook.com/intern/diff/D34822912/) [ghstack-poisoned]
…acer (#74105) Summary: Pull Request resolved: #74105 Currently, the kernel dtype tracer relies on the user to for advisory locking of the mutex when accessing shared data structures. This changes it to be mandatory locking using the `c10::Synchronized<T>` abstraction. ghstack-source-id: 151471744 Test Plan: Built model tracer successfully using: ``` buck build -c pt.disable_per_op_profiling=0 -c pt.enable_record_kernel_dtype=1 --show-output xplat/caffe2/fb/model_tracer:model_tracer ``` Reviewed By: malfet Differential Revision: D34822912 fbshipit-source-id: 628529928ba768f04f4a1be898cb2d73a9e444a4
Stack from ghstack (oldest at bottom):
Currently, the kernel dtype tracer relies on the user to for advisory locking of the mutex when accessing shared data structures. This changes it to be mandatory locking using the
c10::Synchronized<T>abstraction.Differential Revision: D34822912