pytorch/test/cpp
Ilia Cherniavskii b50825e011 Make RecordFunction more robust for async use cases (#34122)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/34122

Earlier work added support for async rpc cases when RecordFunction's
end callbacks might be called in a different thread; in addition some
extra care was needed to handle pointer to parent function;

This PR makes RecordFunction aware of potentially multiple threads in
use, as well as removes unused parent() call and restricts current()
RecordFunction to scope-based record functions (RECORD_FUNCTION macro)

Test Plan: unit tests

Differential Revision: D20297709

Pulled By: ilia-cher

fbshipit-source-id: 46a59e1b2eea0bbd8a59630385e193b38d30f9d1
2020-03-05 22:28:53 -08:00
..
api [C++ API Parity] Adam: updated step and class design (#33730) 2020-03-05 19:15:24 -08:00
common
dist_autograd [Dist Autograd] Functional API for Dist Autograd and Dist Optimizer (#33711) 2020-02-26 19:08:28 -08:00
jit Make RecordFunction more robust for async use cases (#34122) 2020-03-05 22:28:53 -08:00
rpc Remove dead includes in caffe2/test 2020-01-21 11:30:34 -08:00
tensorexpr [TensorExpr] Add IR Printer. (#33220) 2020-02-21 13:10:26 -08:00
__init__.py