Fix typo

Author: bkryu

benchmarks/bench_logging_overhead.py

@@ -7,14 +7,14 @@
 
 Usage:
     # Set the logging level before running
-    export FLASHINFER_APILEVEL=3
+    export FLASHINFER_LOGLEVEL=3
     python bench_logging_overhead.py
 
     # Or run with different levels
-    FLASHINFER_APILEVEL=0 python bench_logging_overhead.py
-    FLASHINFER_APILEVEL=1 python bench_logging_overhead.py
-    FLASHINFER_APILEVEL=3 python bench_logging_overhead.py
-    FLASHINFER_APILEVEL=5 python bench_logging_overhead.py
+    FLASHINFER_LOGLEVEL=0 python bench_logging_overhead.py
+    FLASHINFER_LOGLEVEL=1 python bench_logging_overhead.py
+    FLASHINFER_LOGLEVEL=3 python bench_logging_overhead.py
+    FLASHINFER_LOGLEVEL=5 python bench_logging_overhead.py
 
     # Or use the helper script to run all levels
     bash benchmark_all_levels.sh
@@ -28,8 +28,8 @@
 from typing import List, Tuple
 
 # Get logging level BEFORE importing flashinfer
-LOGGING_LEVEL = int(os.environ.get("FLASHINFER_APILEVEL", "0"))
-LOG_DEST = os.environ.get("FLASHINFER_APIDEST", "/tmp/flashinfer_benchmark_log.txt")
+LOGGING_LEVEL = int(os.environ.get("FLASHINFER_LOGLEVEL", "0"))
+LOG_DEST = os.environ.get("FLASHINFER_LOGDEST", "/tmp/flashinfer_benchmark_log.txt")
 
 # Import the decorator
 from flashinfer.api_logging import flashinfer_api
@@ -209,8 +209,8 @@ def main():
 
     # Display logging configuration
     print("\nLogging Configuration:")
-    print(f"  FLASHINFER_APILEVEL = {LOGGING_LEVEL}")
-    print(f"  FLASHINFER_APIDEST = {LOG_DEST}")
+    print(f"  FLASHINFER_LOGLEVEL = {LOGGING_LEVEL}")
+    print(f"  FLASHINFER_LOGDEST = {LOG_DEST}")
 
     # Get level name
     level_names = {
@@ -314,7 +314,7 @@ def main():
     print("\nTo benchmark other levels, run:")
     for level in [0, 1, 3, 5]:
         if level != LOGGING_LEVEL:
-            print(f"  FLASHINFER_APILEVEL={level} python {sys.argv[0]}")
+            print(f"  FLASHINFER_LOGLEVEL={level} python {sys.argv[0]}")
 
     print("\n" + "=" * 80)
     print("Benchmark complete!")

flashinfer/api_logging.py

@@ -39,8 +39,8 @@ def _substitute_process_id(path: str) -> str:
 
 
 # Read environment variables once at module load time
-_API_LOG_LEVEL = int(os.environ.get("FLASHINFER_APILEVEL", "0"))
-_API_LOG_DEST = _substitute_process_id(os.environ.get("FLASHINFER_APIDEST", "stdout"))
+_API_LOG_LEVEL = int(os.environ.get("FLASHINFER_LOGLEVEL", "0"))
+_API_LOG_DEST = _substitute_process_id(os.environ.get("FLASHINFER_LOGDEST", "stdout"))
 
 # Create logger using Python's logging library
 _logger = logging.getLogger("flashinfer.api")
@@ -54,7 +54,7 @@ def _setup_logger():
         _logger.setLevel(logging.CRITICAL + 1)  # Higher than any level
         return
 
-    # All enabled levels use loggging.DEBUG; verbosity is controlled by FLASHINFER_APILEVEL instead
+    # All enabled levels use loggging.DEBUG; verbosity is controlled by FLASHINFER_LOGLEVEL instead
     _logger.setLevel(logging.DEBUG)
 
     # Remove any existing handlers
@@ -466,17 +466,17 @@ def flashinfer_api(func: Callable = None) -> Callable:
     Decorator to log FlashInfer API calls using Python's logging library.
 
     This decorator integrates with Python's standard logging infrastructure while
-    maintaining zero overhead when disabled (FLASHINFER_APILEVEL=0).
+    maintaining zero overhead when disabled (FLASHINFER_LOGLEVEL=0).
 
     Environment Variables
     ---------------------
-    FLASHINFER_APILEVEL : int (default: 0)
+    FLASHINFER_LOGLEVEL : int (default: 0)
         - 0: No logging (zero overhead - decorator returns original function)
         - 1: Log function name only (logged BEFORE execution - crash-safe)
         - 3: Log function name + inputs/outputs with metadata (inputs logged BEFORE execution - crash-safe)
         - 5: Log function name + inputs/outputs with metadata + tensor statistics (inputs logged BEFORE execution - crash-safe)
 
-    FLASHINFER_APIDEST : str (default: "stdout")
+    FLASHINFER_LOGDEST : str (default: "stdout")
         - "stdout": Log to standard output
         - "stderr": Log to standard error
         - <path>: Log to specified file path
@@ -494,7 +494,7 @@ def flashinfer_api(func: Callable = None) -> Callable:
     -----
     - Key header lines include a timestamp in the format: [YYYY-MM-DD HH:MM:SS]
       (e.g., "FlashInfer API Call: function_name", "FlashInfer API Logging - System Information")
-    - When FLASHINFER_APILEVEL=0, the decorator has truly zero overhead
+    - When FLASHINFER_LOGLEVEL=0, the decorator has truly zero overhead
       as it returns the original function unchanged.
     - Function names and inputs are logged BEFORE execution:
       - Level 1: Function name only

tests/utils/test_logging.py

@@ -47,30 +47,30 @@ class TestAPILogging:
     def setup_and_teardown(self):
         """Reset environment and reimport logging module for each test."""
         # Store original environment
-        original_level = os.environ.get("FLASHINFER_APILEVEL")
-        original_dest = os.environ.get("FLASHINFER_APIDEST")
+        original_level = os.environ.get("FLASHINFER_LOGLEVEL")
+        original_dest = os.environ.get("FLASHINFER_LOGDEST")
 
         yield
 
         # Restore original environment
         if original_level is not None:
-            os.environ["FLASHINFER_APILEVEL"] = original_level
-        elif "FLASHINFER_APILEVEL" in os.environ:
-            del os.environ["FLASHINFER_APILEVEL"]
+            os.environ["FLASHINFER_LOGLEVEL"] = original_level
+        elif "FLASHINFER_LOGLEVEL" in os.environ:
+            del os.environ["FLASHINFER_LOGLEVEL"]
 
         if original_dest is not None:
-            os.environ["FLASHINFER_APIDEST"] = original_dest
-        elif "FLASHINFER_APIDEST" in os.environ:
-            del os.environ["FLASHINFER_APIDEST"]
+            os.environ["FLASHINFER_LOGDEST"] = original_dest
+        elif "FLASHINFER_LOGDEST" in os.environ:
+            del os.environ["FLASHINFER_LOGDEST"]
 
         # Force reimport to pick up new environment variables
         if "flashinfer.api_logging" in sys.modules:
             del sys.modules["flashinfer.api_logging"]
 
     def setup_logging(self, level: int, dest: str = "stdout"):
         """Helper to set up logging environment and reimport."""
-        os.environ["FLASHINFER_APILEVEL"] = str(level)
-        os.environ["FLASHINFER_APIDEST"] = dest
+        os.environ["FLASHINFER_LOGLEVEL"] = str(level)
+        os.environ["FLASHINFER_LOGDEST"] = dest
 
         # Force reimport
         if "flashinfer.api_logging" in sys.modules: