Create a debug flag (#1546)

Create a `DEBUG` flag for onnxscript. Users can enable it via `onnxscript.DEBUG=True`. We will use this flag to enable additional invariance checking in the IR. 1/2 of #1545
microsoft · May 15, 2024 · b77f393 · b77f393
1 parent 2b6dc27
commit b77f393
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/onnxscript/__init__.py b/onnxscript/__init__.py
@@ -58,6 +58,7 @@
     "opset_ai_onnx_ml2",
     "opset_ai_onnx_ml3",
     "opset_ai_onnx_ml4",
+    "DEBUG",
 ]
 
 import importlib.metadata
@@ -122,6 +123,9 @@
 from ._internal.utils import external_tensor
 from .values import OnnxFunction, TracedOnnxFunction
 
+# Set DEBUG to True to enable additional debug checks
+DEBUG = False
+
 try:  # noqa: SIM105
     __version__ = importlib.metadata.version("onnxscript")
 except importlib.metadata.PackageNotFoundError: