Add function call tracing argument. Implement HPU Graph class DeviceR…

…unner. Add code to setup model to leverage HPU Graphs when HPU present.
CloudBrigade · Nov 23, 2023 · 914333d · 914333d
1 parent c76aa1f
commit 914333d
Show file tree

Hide file tree

Showing 6 changed files with 35,408 additions and 1 deletion.
diff --git a/launch.py b/launch.py
@@ -1,4 +1,5 @@
 from modules import launch_utils
+import sys
 
 args = launch_utils.args
 python = launch_utils.python
@@ -25,6 +26,9 @@
 
 
 def main():
+    if args.enable_trace:
+        sys.setprofile(tracefunc)
+
     if args.dump_sysinfo:
         filename = launch_utils.dump_sysinfo()