remove verbosity levels ERROR and WARNING, consistent prefix for all prints, use warnings for warning messages

Author: nvssatoor

nsight/extraction.py

@@ -144,8 +144,6 @@ def extract_df_from_report(
     }
 
     # Extract all profiling data
-    if verbosity >= VerbosityLevel.INFO:
-        print(f"Extracting profiling data")
     profiling_data: dict[str, list[utils.NCUActionData]] = {}
     for range_idx in range(report.num_ranges()):
         current_range: ncu_report.IRange = report.range_by_idx(range_idx)
@@ -172,7 +170,7 @@ def extract_df_from_report(
 
     for annotation, annotation_data in profiling_data.items():
         if verbosity >= VerbosityLevel.INFO:
-            print(f"Extracting {annotation} profiling data")
+            print(f"[NSIGHT-PYTHON] Extracting {annotation} profiling data")
 
         configs_repeated = [
             (config,) if is_scalar(config) else config

nsight/thermovision.py

@@ -5,6 +5,7 @@
 from typing import Any, Literal
 
 from nsight.exceptions import CoolingTimeoutError
+import warnings
 
 """
 This module provides GPU thermal monitoring and throttling prevention using NVIDIA's NVML library
@@ -22,8 +23,8 @@
     CUDA_CORE_AVAILABLE = True
 except ImportError:
     CUDA_CORE_AVAILABLE = False
-    print(
-        "Warning: Cannot import cuda.core. Ensure nsight-python was installed properly with all dependencies."
+    warnings.warn(
+        "Cannot import cuda.core. Ensure nsight-python was installed properly with all dependencies."
     )
 
 # Default thermal threshold constants
@@ -270,7 +271,7 @@ def _is_temp_retrieval_supported(self) -> bool:
             self.device.temperature.margin
             return True
         except Exception:
-            print(
-                "Warning: Nsight Python Thermovision is not supported on this machine"
+            warnings.warn(
+                "Nsight Python Thermovision is not supported on this machine"
             )
             return False

nsight/utils.py

@@ -43,12 +43,10 @@ class VerbosityLevel(enum.IntEnum):
     """Verbosity level for ``@nsight.analyze.kernel``.
 
     Selecting level N enables output at all levels with value <= N.
-    For example, ``INFO`` enables ERROR, WARNING, and INFO but not DEBUG.
+    For example, ``INFO`` enables INFO but not DEBUG.
     """
 
     SILENT = 0
-    ERROR = 1
-    WARNING = 2
     INFO = 3
     DEBUG = 4
 
@@ -134,10 +132,10 @@ def format_time(seconds: float) -> str:
 # Sincerely stolen (and adjusted) from attention-gym
 def print_header(*lines: str) -> None:
     width = max(80, max(len(line) for line in lines) + 4)
-    print(purple("╔" + "═" * (width - 2) + "╗"))
+    print(purple("[NSIGHT-PYTHON] ╔" + "═" * (width - 2) + "╗"))
     for line in lines:
-        print(purple(f"║ {line.center(width - 4)} ║"))
-    print(purple("╚" + "═" * (width - 2) + "╝"))
+        print(purple(f"[NSIGHT-PYTHON] ║ {line.center(width - 4)} ║"))
+    print(purple("[NSIGHT-PYTHON] ╚" + "═" * (width - 2) + "╝"))
 
 
 @dataclass
@@ -203,13 +201,13 @@ def print_progress_bar(
         sys.stdout.write("\033[1A")  # Move cursor up 1 line
         sys.stdout.write("\033[2K\r")  # Clear line
         sys.stdout.write(
-            f"Progress: [{bar}] {progress * 100:.2f}% | Estimated time remaining: {formatted_time}\n"
+            f"[NSIGHT-PYTHON] Progress: [{bar}] {progress * 100:.2f}% | Estimated time remaining: {formatted_time}\n"
         )
         sys.stdout.flush()
 
     else:
         print(
-            f"Progress: [{bar}] {progress * 100:.2f}% | Estimated time remaining: {formatted_time}"
+            f"[NSIGHT-PYTHON] Progress: [{bar}] {progress * 100:.2f}% | Estimated time remaining: {formatted_time}"
         )