ttnn-visualizer 0.63.1__py3-none-any.whl → 0.65.0__py3-none-any.whl

ttnn_visualizer/app.py

@@ -25,7 +25,7 @@ from ttnn_visualizer.exceptions import (
 )
 from ttnn_visualizer.instances import create_instance_from_local_paths
 from ttnn_visualizer.settings import Config, DefaultConfig
-from ttnn_visualizer.utils import find_gunicorn_path
+from ttnn_visualizer.utils import find_gunicorn_path, get_app_data_directory
 from werkzeug.debug import DebuggedApplication
 from werkzeug.middleware.proxy_fix import ProxyFix
 
@@ -55,8 +55,9 @@ def create_app(settings_override=None):
         static_folder=config.STATIC_ASSETS_DIR,
         static_url_path=f"{config.BASE_PATH}static",
     )
-    logging.basicConfig(level=app.config.get("LOG_LEVEL", "INFO"))
 
+    # logging.basicConfig(level=app.config.get("LOG_LEVEL", "DEBUG"))
+    logging.basicConfig(level=logging.DEBUG)
     app.config.from_object(config)
 
     if settings_override:
@@ -118,6 +119,8 @@ def extensions(app: flask.Flask):
     flask_static_digest.init_app(app)
     if app.config["USE_WEBSOCKETS"]:
         socketio.init_app(app)
+
+    Path(app.config["APP_DATA_DIRECTORY"]).mkdir(parents=True, exist_ok=True)
     db.init_app(app)
 
     if app.config["USE_WEBSOCKETS"]:
@@ -192,6 +195,23 @@ def parse_args():
     parser.add_argument(
         "--tt-metal-home", help="Specify a TT-Metal home path", default=None
     )
+    parser.add_argument(
+        "--host",
+        type=str,
+        help="Host to bind to (default: auto-detected based on environment)",
+        default=None,
+    )
+    parser.add_argument(
+        "--port",
+        type=str,
+        help="Port to bind to (default: 8000)",
+        default=None,
+    )
+    parser.add_argument(
+        "--server",
+        action="store_true",
+        help="Bind to all network interfaces (0.0.0.0) and enable server mode. Useful for servers and VMs",
+    )
     parser.add_argument(
         "-d",
         "--daemon",
@@ -250,13 +270,56 @@ def main():
 
     args = parse_args()
 
+    # Handle host/port CLI overrides
+    # Priority: CLI args > env vars > auto-detection (in settings.py)
+    # Note: We need to set env vars before creating Config, but also
+    # manually update the config object in case it was already instantiated
+    if args.host:
+        os.environ["HOST"] = args.host
+        print(f"🌐 Binding to host: {args.host} (from --host flag)")
+    elif args.server:
+        os.environ["HOST"] = "0.0.0.0"
+        os.environ["SERVER_MODE"] = "true"
+        print("🌐 Binding to all interfaces (0.0.0.0) via --server flag")
+        print("🖥️  Server mode enabled")
+
+    if args.port:
+        os.environ["PORT"] = args.port
+        print(f"🔌 Binding to port: {args.port}")
+
     config = cast(DefaultConfig, Config())
+
+    # Apply CLI overrides directly to config object
+    # (Config is a singleton that may have been created before we set env vars)
+    if args.host:
+        config.HOST = args.host
+    elif args.server:
+        config.HOST = "0.0.0.0"
+        config.SERVER_MODE = True
+
+    if args.port:
+        config.PORT = args.port
+
+    # Recalculate GUNICORN_BIND with the updated values
+    config.GUNICORN_BIND = f"{config.HOST}:{config.PORT}"
+
     instance_id = None
 
     # Display mode information first (using config only, no DB needed)
     if args.tt_metal_home:
+        os.environ["TT_METAL_HOME"] = args.tt_metal_home
         config.TT_METAL_HOME = args.tt_metal_home
 
+        if not os.getenv("APP_DATA_DIRECTORY"):
+            config.APP_DATA_DIRECTORY = get_app_data_directory(
+                args.tt_metal_home, config.APPLICATION_DIR
+            )
+            # Recalculate database path with new APP_DATA_DIRECTORY
+            _db_file_path = str(
+                Path(config.APP_DATA_DIRECTORY) / f"ttnn_{config.DB_VERSION}.db"
+            )
+            config.SQLALCHEMY_DATABASE_URI = f"sqlite:///{_db_file_path}"
+
     display_mode_info_without_db(config)
 
     # If profiler/performance paths are provided, create an instance

ttnn_visualizer/csv_queries.py

@@ -398,6 +398,7 @@ class OpsPerformanceReportQueries:
         "total_percent",
         "bound",
         "op_code",
+        "device",
         "device_time",
         "op_to_op_gap",
         "cores",
@@ -430,12 +431,26 @@ class OpsPerformanceReportQueries:
         "flops_std",
     ]
 
+    STACKED_REPORT_COLUMNS_WITH_DEVICE = [
+        "percent",
+        "op_code",
+        "device",
+        "device_time_sum_us",
+        "ops_count",
+        "flops_min",
+        "flops_max",
+        "flops_mean",
+        "flops_std",
+    ]
+
     PASSTHROUGH_COLUMNS = {
         "pm_ideal_ns": "PM IDEAL [ns]",
     }
 
-    DEFAULT_SIGNPOST = None
+    DEFAULT_START_SIGNPOST = None
+    DEFAULT_END_SIGNPOST = None
     DEFAULT_IGNORE_SIGNPOSTS = True
+    DEFAULT_PRINT_SIGNPOSTS = True
     DEFAULT_MIN_PERCENTAGE = 0.5
     DEFAULT_ID_RANGE = None
     DEFAULT_NO_ADVICE = False
@@ -444,6 +459,7 @@ class OpsPerformanceReportQueries:
     DEFAULT_NO_HOST_OPS = False
     DEFAULT_NO_STACKED_REPORT = False
     DEFAULT_NO_STACK_BY_IN0 = True
+    DEFAULT_MERGE_DEVICES = True
 
     @classmethod
     def generate_report(cls, instance, **kwargs):
@@ -451,31 +467,39 @@ class OpsPerformanceReportQueries:
         csv_file = StringIO(raw_csv)
         csv_output_file = tempfile.mktemp(suffix=".csv")
         csv_stacked_output_file = tempfile.mktemp(suffix=".csv")
-        signpost = kwargs.get("signpost", cls.DEFAULT_SIGNPOST)
+        start_signpost = kwargs.get("start_signpost", cls.DEFAULT_START_SIGNPOST)
+        end_signpost = kwargs.get("end_signpost", cls.DEFAULT_END_SIGNPOST)
         ignore_signposts = cls.DEFAULT_IGNORE_SIGNPOSTS
+        print_signposts = kwargs.get("print_signposts", cls.DEFAULT_PRINT_SIGNPOSTS)
         stack_by_in0 = kwargs.get("stack_by_in0", cls.DEFAULT_NO_STACK_BY_IN0)
+        no_host_ops = kwargs.get("hide_host_ops", cls.DEFAULT_NO_HOST_OPS)
+        merge_devices = kwargs.get("merge_devices", cls.DEFAULT_MERGE_DEVICES)
 
-        if signpost:
+        if start_signpost or end_signpost:
             ignore_signposts = False
 
         # perf_report currently generates a PNG alongside the CSV using the same temp name - we'll just delete it afterwards
-        stacked_png_file = os.path.splitext(csv_output_file)[0] + ".png"
+        stacked_png_file = csv_stacked_output_file + ".png"
+        stacked_csv_file = csv_stacked_output_file + ".csv"
 
         try:
             perf_report.generate_perf_report(
                 [csv_file],
-                signpost,
+                start_signpost,
+                end_signpost,
                 ignore_signposts,
+                print_signposts,
                 cls.DEFAULT_MIN_PERCENTAGE,
                 cls.DEFAULT_ID_RANGE,
                 csv_output_file,
                 cls.DEFAULT_NO_ADVICE,
                 cls.DEFAULT_TRACING_MODE,
                 cls.DEFAULT_RAW_OP_CODES,
-                cls.DEFAULT_NO_HOST_OPS,
+                no_host_ops,
                 cls.DEFAULT_NO_STACKED_REPORT,
                 stack_by_in0,
                 csv_stacked_output_file,
+                not merge_devices,
             )
         except Exception as e:
             logger.error(f"Error generating performance report: {e}")
@@ -552,16 +576,23 @@ class OpsPerformanceReportQueries:
 
         stacked_report = []
 
-        if os.path.exists(csv_stacked_output_file):
+        if os.path.exists(stacked_csv_file):
             try:
-                with open(csv_stacked_output_file, newline="") as csvfile:
+                with open(stacked_csv_file, newline="") as csvfile:
                     reader = csv.reader(csvfile, delimiter=",")
                     next(reader, None)
 
+                    # Use the appropriate column list based on merge_devices flag
+                    stacked_columns = (
+                        cls.STACKED_REPORT_COLUMNS_WITH_DEVICE
+                        if not merge_devices
+                        else cls.STACKED_REPORT_COLUMNS
+                    )
+
                     for row in reader:
                         processed_row = {
                             column: row[index]
-                            for index, column in enumerate(cls.STACKED_REPORT_COLUMNS)
+                            for index, column in enumerate(stacked_columns)
                             if index < len(row)
                         }
 
@@ -577,12 +608,10 @@ class OpsPerformanceReportQueries:
             except csv.Error as e:
                 raise DataFormatError() from e
             finally:
-                os.unlink(csv_stacked_output_file)
+                os.unlink(stacked_csv_file)
                 if os.path.exists(stacked_png_file):
                     os.unlink(stacked_png_file)
 
-                    stacked_report.append(processed_row)
-
         return {
             "report": report,
             "stacked_report": stacked_report,

ttnn_visualizer/settings.py

@@ -7,7 +7,11 @@ from pathlib import Path
 
 from dotenv import load_dotenv
 from sqlalchemy.pool import NullPool
-from ttnn_visualizer.utils import str_to_bool
+from ttnn_visualizer.utils import (
+    get_app_data_directory,
+    is_running_in_container,
+    str_to_bool,
+)
 
 load_dotenv()
 
@@ -41,9 +45,13 @@ class DefaultConfig(object):
     PERFORMANCE_DIRECTORY_NAME = "performance-reports"
     NPE_DIRECTORY_NAME = "npe-reports"
     APPLICATION_DIR = os.path.abspath(os.path.join(__file__, "..", os.pardir))
-    APP_DATA_DIRECTORY = os.getenv("APP_DATA_DIRECTORY", APPLICATION_DIR)
-    STATIC_ASSETS_DIR = Path(APPLICATION_DIR).joinpath("ttnn_visualizer", "static")
     TT_METAL_HOME = os.getenv("TT_METAL_HOME", None)
+    APP_DATA_DIRECTORY = os.getenv(
+        "APP_DATA_DIRECTORY",
+        get_app_data_directory(TT_METAL_HOME, APPLICATION_DIR),
+    )
+
+    STATIC_ASSETS_DIR = Path(APPLICATION_DIR).joinpath("ttnn_visualizer", "static")
     SEND_FILE_MAX_AGE_DEFAULT = 0
 
     LAUNCH_BROWSER_ON_START = str_to_bool(os.getenv("LAUNCH_BROWSER_ON_START", "true"))
@@ -75,7 +83,7 @@ class DefaultConfig(object):
     GUNICORN_WORKERS = os.getenv("GUNICORN_WORKERS", "1")
     GUNICORN_TIMEOUT = os.getenv("GUNICORN_TIMEOUT", "60")
     PORT = os.getenv("PORT", "8000")
-    HOST = os.getenv("HOST", "localhost")
+    HOST = os.getenv("HOST", "0.0.0.0" if is_running_in_container() else "localhost")
     DEV_SERVER_PORT = "5173"
     DEV_SERVER_HOST = "localhost"
 

ttnn_visualizer/static/assets/{allPaths-dy9nsQF7.js → allPaths-B6-2k5TG.js}

@@ -1 +1 @@
-import{I as s}from"./index-voJy5fZe.js";import{I as r}from"./index-BZITDwoa.js";import{p as n,I as c}from"./index-B_nutJ_M.js";function p(t,a){const o=n(t);return a===c.STANDARD?s[o]:r[o]}export{s as IconSvgPaths16,r as IconSvgPaths20,p as getIconPaths};
+import{I as s}from"./index-voJy5fZe.js";import{I as r}from"./index-BZITDwoa.js";import{p as n,I as c}from"./index-DMzz7de9.js";function p(t,a){const o=n(t);return a===c.STANDARD?s[o]:r[o]}export{s as IconSvgPaths16,r as IconSvgPaths20,p as getIconPaths};

ttnn_visualizer/static/assets/allPathsLoader-Cylj0kPO.js

@@ -0,0 +1,2 @@
+const __vite__mapDeps=(i,m=__vite__mapDeps,d=(m.f||(m.f=["assets/allPaths-B6-2k5TG.js","assets/index-voJy5fZe.js","assets/index-BZITDwoa.js","assets/index-DMzz7de9.js","assets/index-C7m_PE7l.css"])))=>i.map(i=>d[i]);
+import{_ as e}from"./index-DMzz7de9.js";const s=async(t,a)=>{const{getIconPaths:o}=await e(async()=>{const{getIconPaths:r}=await import("./allPaths-B6-2k5TG.js");return{getIconPaths:r}},__vite__mapDeps([0,1,2,3,4]));return o(t,a)};export{s as allPathsLoader};