[3/N] Refactor FR script - Add a processor module (pytorch#133933)

Pull Request resolved: pytorch#133933 Approved by: https://github.com/c-p-i-o ghstack dependencies: pytorch#133927, pytorch#133929
mori360 · Aug 20, 2024 · e41b520 · e41b520
1 parent bce0cab
commit e41b520
Show file tree

Hide file tree

Showing 3 changed files with 901 additions and 892 deletions.
diff --git a/test/distributed/flight_recorder/test_fr_analysis.py b/test/distributed/flight_recorder/test_fr_analysis.py
@@ -7,7 +7,7 @@
 REPO_ROOT = pathlib.Path(__file__).resolve().parent.parent.parent.parent
 
 sys.path.insert(0, str(REPO_ROOT))
-from tools.flight_recorder.fr_trace import match_one_event, MatchState
+from tools.flight_recorder.components.processor import match_one_event, MatchState
 
 
 # Make sure to remove REPO_ROOT after import is done