Add Makefile and format fix script for tritonparse project

Summary:
- Introduced a Makefile to streamline project tasks including formatting, linting, testing, and cleaning.
- Added a format fix script to automate code formatting and linting using usort, ruff, and black.
- Updated pyproject.toml to include configuration for usort and ruff.

This enhancement improves development workflow by providing easy-to-use commands for maintaining code quality.

Author: FindHao

Makefile

@@ -0,0 +1,59 @@
+# Makefile for tritonparse project
+
+.PHONY: help format format-check lint lint-check test test-cuda clean install-dev
+
+# Default target
+help:
+	@echo "Available targets:"
+	@echo "  format        - Format all Python files"
+	@echo "  format-check  - Check formatting without making changes"
+	@echo "  lint          - Run all linters"
+	@echo "  lint-check    - Check linting without making changes"
+	@echo "  test          - Run tests (CPU only)"
+	@echo "  test-cuda     - Run tests (including CUDA tests)"
+	@echo "  clean         - Clean up cache files"
+	@echo "  install-dev   - Install development dependencies"
+
+# Formatting targets
+format:
+	@echo "Running format fix script..."
+	python -m tritonparse.tools.format_fix --verbose
+
+format-check:
+	@echo "Checking formatting..."
+	python -m tritonparse.tools.format_fix --check-only --verbose
+
+# Linting targets
+lint:
+	@echo "Running linters..."
+	ruff check .
+	ruff format --check .
+
+lint-check:
+	@echo "Checking linting..."
+	ruff check --diff .
+	ruff format --diff .
+
+# Testing targets
+test:
+	@echo "Running tests (CPU only)..."
+	pytest tests/ -v -m "not cuda"
+
+test-cuda:
+	@echo "Running all tests (including CUDA)..."
+	pytest tests/ -v
+
+# Utility targets
+clean:
+	@echo "Cleaning up cache files..."
+	find . -type d -name "__pycache__" -exec rm -rf {} + 2>/dev/null || true
+	find . -type f -name "*.pyc" -delete 2>/dev/null || true
+	find . -type f -name "*.pyo" -delete 2>/dev/null || true
+	find . -type f -name ".coverage" -delete 2>/dev/null || true
+	find . -type d -name "*.egg-info" -exec rm -rf {} + 2>/dev/null || true
+	find . -type d -name ".pytest_cache" -exec rm -rf {} + 2>/dev/null || true
+
+install-dev:
+	@echo "Installing development dependencies..."
+	pip install -e ".[test]"
+	pip install ruff black usort 

pyproject.toml

@@ -24,5 +24,12 @@ line-length = 88
 line-length = 88
 target-version = ["py310"]
 
+[tool.ufmt]
+formatter = "ruff-api"
+sorter = "usort"
+
+[tool.usort]
+first_party_detection = false
+
 [project.urls]
 "Homepage" = "https://github.com/pytorch-labs/tritonparse"

tritonparse/tools/format_fix.py

@@ -0,0 +1,196 @@
+#!/usr/bin/env python3
+"""
+Format fix script for tritonparse project.
+
+This script runs all linter tools to format and fix code issues:
+- usort: Import sorting
+- ruff: Code formatting and linting
+- black: Code formatting (fallback)
+
+Usage:
+    python -m tritonparse.tools.format_fix [options]
+
+Options:
+    --check-only    Only check for issues, don't fix them
+    --verbose       Verbose output
+    --help          Show this help message
+"""
+
+import argparse
+import subprocess
+import sys
+from pathlib import Path
+from typing import List
+
+
+def run_command(cmd: List[str], verbose: bool = False) -> bool:
+    """Run a command and return success status."""
+    if verbose:
+        print(f"Running: {' '.join(cmd)}")
+    
+    try:
+        result = subprocess.run(
+            cmd, 
+            capture_output=True, 
+            text=True, 
+            check=False
+        )
+        
+        if result.returncode != 0:
+            if verbose:
+                print(f"Command failed with return code {result.returncode}")
+                if result.stdout:
+                    print("STDOUT:", result.stdout)
+                if result.stderr:
+                    print("STDERR:", result.stderr)
+            return False
+        
+        if verbose and result.stdout:
+            print(result.stdout)
+        
+        return True
+    except Exception as e:
+        if verbose:
+            print(f"Error running command: {e}")
+        return False
+
+
+def get_python_files() -> List[str]:
+    """Get all Python files to format."""
+    project_root = Path(__file__).parent.parent.parent
+    python_files = []
+    
+    # Add main package files
+    for pattern in ["*.py", "tests/*.py"]:
+        python_files.extend(str(f) for f in project_root.glob(pattern))
+    
+    # Filter out __pycache__ and other unwanted directories
+    python_files = [
+        f for f in python_files 
+        if "__pycache__" not in f and ".git" not in f
+    ]
+    
+    return python_files
+
+
+def run_usort(files: List[str], check_only: bool = False, 
+               verbose: bool = False) -> bool:
+    """Run usort for import sorting."""
+    cmd = ["usort"]
+    
+    if check_only:
+        cmd.append("check")
+    else:
+        cmd.append("format")
+    
+    cmd.extend(files)
+    
+    return run_command(cmd, verbose)
+
+
+def run_ruff(files: List[str], check_only: bool = False, 
+              verbose: bool = False) -> bool:
+    """Run ruff for code formatting and linting."""
+    cmd = ["ruff"]
+    
+    if check_only:
+        cmd.extend(["check", "--diff"])
+    else:
+        cmd.extend(["format"])
+    
+    cmd.extend(files)
+    
+    return run_command(cmd, verbose)
+
+
+def run_black(files: List[str], check_only: bool = False, 
+               verbose: bool = False) -> bool:
+    """Run black for code formatting (fallback)."""
+    cmd = ["black"]
+    
+    if check_only:
+        cmd.append("--check")
+    else:
+        cmd.append("--quiet")
+    
+    cmd.extend(files)
+    
+    return run_command(cmd, verbose)
+
+
+def main():
+    """Main function."""
+    parser = argparse.ArgumentParser(
+        description="Format fix script for tritonparse project",
+        epilog="""
+Examples:
+    # Fix all formatting issues
+    python -m tritonparse.tools.format_fix
+    
+    # Check for issues without fixing
+    python -m tritonparse.tools.format_fix --check-only
+    
+    # Verbose output
+    python -m tritonparse.tools.format_fix --verbose
+        """
+    )
+    
+    parser.add_argument(
+        "--check-only",
+        action="store_true",
+        help="Only check for issues, don't fix them"
+    )
+    parser.add_argument(
+        "--verbose",
+        action="store_true",
+        help="Verbose output"
+    )
+    
+    args = parser.parse_args()
+    
+    # Get Python files to format
+    files = get_python_files()
+    
+    if args.verbose:
+        print(f"Found {len(files)} Python files to process:")
+        for f in files:
+            print(f"  {f}")
+        print()
+    
+    # Run formatters
+    success = True
+    
+    # 1. Run usort for import sorting
+    print("Running usort for import sorting...")
+    if not run_usort(files, args.check_only, args.verbose):
+        print("❌ usort failed")
+        success = False
+    else:
+        print("✅ usort completed")
+    
+    # 2. Run ruff for code formatting and linting
+    print("Running ruff for code formatting and linting...")
+    if not run_ruff(files, args.check_only, args.verbose):
+        print("❌ ruff failed")
+        success = False
+    else:
+        print("✅ ruff completed")
+    
+    # 3. Run black as fallback
+    print("Running black for additional formatting...")
+    if not run_black(files, args.check_only, args.verbose):
+        print("❌ black failed")
+        success = False
+    else:
+        print("✅ black completed")
+    
+    if success:
+        print("\n🎉 All formatting tools completed successfully!")
+        return 0
+    else:
+        print("\n❌ Some formatting tools failed. Please check the output above.")
+        return 1
+
+
+if __name__ == "__main__":
+    sys.exit(main())