Refactor linting and formatting tools in Makefile and scripts

Author: FindHao

tritonparse/common.py

@@ -173,9 +173,9 @@ def print_parsed_files_summary(parsed_log_dir: str) -> None:
                 if size_bytes < 1024:
                     file_size = f"{size_bytes}B"
                 elif size_bytes < 1024 * 1024:
-                    file_size = f"{size_bytes/1024:.1f}KB"
+                    file_size = f"{size_bytes / 1024:.1f}KB"
                 else:
-                    file_size = f"{size_bytes/(1024*1024):.1f}MB"
+                    file_size = f"{size_bytes / (1024 * 1024):.1f}MB"
             except OSError:
                 pass
 

tritonparse/structured_logging.py

@@ -142,8 +142,6 @@ def convert(obj):
     Returns:
         A serializable version of the input object where dataclasses are converted to dictionaries
     """
-    from triton.language.core import dtype
-
     # 1. primitives that JSON already supports  -------------------------------
     if obj is None or isinstance(obj, (bool, int, str)):
         return obj
@@ -178,11 +176,6 @@ def convert(obj):
         return convert(
             asdict(obj)
         )  # Convert dataclass to dict and then process that dict
-
-    # 4. Common Triton constexpr objects
-    if isinstance(obj, dtype):
-        return f"triton.language.core.dtype('{str(obj)}')"
-
     log.warning(f"Unknown type: {type(obj)}")
     return str(obj)  # Return primitive types as-is
 
@@ -378,18 +371,18 @@ def extract_file_content(trace_data: Dict[str, Any], metadata_group: Dict[str, s
                 # Check file size before reading to avoid memory issues
                 file_size = os.path.getsize(file_path)
                 if file_size > MAX_FILE_SIZE:
-                    trace_data["file_content"][ir_filename] = (
-                        f"<file too large: {file_size} bytes>"
-                    )
+                    trace_data["file_content"][
+                        ir_filename
+                    ] = f"<file too large: {file_size} bytes>"
                     continue
 
                 with open(file_path, "r") as f:
                     trace_data["file_content"][ir_filename] = f.read()
             except (UnicodeDecodeError, OSError) as e:
                 # add more specific error type
-                trace_data["file_content"][ir_filename] = (
-                    f"<error reading file: {str(e)}>"
-                )
+                trace_data["file_content"][
+                    ir_filename
+                ] = f"<error reading file: {str(e)}>"
                 log.debug(f"Error reading file {file_path}: {e}")
 
 

tritonparse/utils.py

@@ -6,9 +6,6 @@
 from pathlib import Path
 from typing import Optional
 
-# argument parser for OSS
-parser = None
-
 from .common import (
     copy_local_to_tmpdir,
     is_fbcode,
@@ -19,6 +16,9 @@
 )
 from .source_type import Source, SourceType
 
+# argument parser for OSS
+parser = None
+
 
 def init_parser():
     global parser