Skip to content

Commit

Permalink
[DI] create a new key for threadLocalDebugInfo (pytorch#48762)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: pytorch#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
  • Loading branch information
Ruichao Xiao authored and facebook-github-bot committed Dec 4, 2020
1 parent 0f9823d commit ca3ae7d
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions c10/util/ThreadLocalDebugInfo.h
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit ca3ae7d

Please sign in to comment.