mirror of https://github.com/llvm/torch-mlir
29 lines
634 B
Python
29 lines
634 B
Python
|
# 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
|
||
|
|
|
||
|
|
import os
|
||
|
|
import string
|
||
|
|
import sys
|
||
|
|
|
||
|
|
__all__ = ["debug", "debug_enabled", "enable"]
|
||
|
|
|
||
|
|
_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)
|