torch-mlir/python/npcomp/compiler/utils/logging.py

29 lines
634 B
Python
Raw Normal View History

# Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
# See https://llvm.org/LICENSE.txt for license information.
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
2020-06-08 06:15:19 +08:00
import os
import string
import sys
__all__ = ["debug", "debug_enabled", "enable"]
2020-06-08 06:15:19 +08:00
_ENABLED = "NPCOMP_DEBUG" in os.environ
_formatter = string.Formatter()
def enable():
global _ENABLED
_ENABLED = True
def debug_enabled():
return _ENABLED
def debug(format_string, *args, **kwargs):
if not _ENABLED:
return
formatted = _formatter.vformat(format_string, args, kwargs)
print("DEBUG:", formatted, file=sys.stderr)