pytorch
cf461a8b - [PyTorch] [Model Tracer] Use c10::Synchronized<T> for kernel dtype tracer (#74105)

Commit
3 years ago
[PyTorch] [Model Tracer] Use c10::Synchronized<T> for kernel dtype tracer (#74105) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/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 (cherry picked from commit 4315f6fcfa4ed60b45731e908f7c424b4c5bba4c)
Author
Committer
Parents
Loading