Revert D34230284: [pytorch][PR] [WIP][Kineto] Manual Submodule Update

Test Plan: revert-hammer Differential Revision: D34230284 (pytorch@c088c8f) Original commit changeset: cbecaf79e51a Original Phabricator Diff: D34230284 (pytorch@c088c8f) fbshipit-source-id: 6976e979c3d43ccd518248ab62082538504654c3 (cherry picked from commit 9a0ad66)
zmmwl · Mar 6, 2022 · e7730ea · e7730ea
1 parent 486bd9f
commit e7730ea
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/third_party/kineto b/third_party/kineto
diff --git a/torch/csrc/autograd/profiler_kineto.cpp b/torch/csrc/autograd/profiler_kineto.cpp
@@ -549,7 +549,12 @@ struct KinetoThreadLocalState : public ProfilerThreadLocalStateBase {
       auto iter = tidSeq2activity.find(key);
       if (iter != tidSeq2activity.end()) {
         libkineto::GenericTraceActivity* fwd = iter->second;
+#ifdef USE_KINETO_UPDATED
         fwd->flow.start = true;
+#else
+        activity.flow.linkedActivity = fwd; // Only destination side set this,
+                                            // to distinguish with start side.
+#endif
         activity.flow.id = fwd->flow.id = fwd_bwd_link_id;
         activity.flow.type = fwd->flow.type = libkineto::kLinkFwdBwd;
         ++fwd_bwd_link_id;