frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[DI] create a new key for threadLocalDebugInfo (#48762)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/48762

In distributed inference, we want to use a new type info to pass some information to operators. add a new key to threadLocalDebugInfo to unblock the development.

Test Plan: Only add a new key. Should have not effect on current build.

Reviewed By: dzhulgakov

Differential Revision: D25291242

fbshipit-source-id: c71565ff7a38cc514d7cd65246c7d5f6b2ce3b8b
This commit is contained in:
Ruichao Xiao
2020-12-04 15:03:56 -08:00
committed by Facebook GitHub Bot
parent 0f9823d888
commit ca3ae7dc73
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
c10/util/ThreadLocalDebugInfo.h
View File

@ -13,6 +13,7 @@ enum class C10_API_ENUM DebugInfoKind : uint8_t {
PRODUCER_INFO = 0,
MOBILE_RUNTIME_INFO,
PROFILER_STATE,
INFERENCE_CONTEXT, // for inference usage
TEST_INFO, // used only in tests
TEST_INFO_2, // used only in tests