torch-mlir/frontends/pytorch/csrc/c10_dispatch/debug.cpp

30 lines
891 B
C++

//===- debug.cpp ------------------------------------------------*- C++ -*-===//
//
// This file is licensed under a pytorch-style license
// See frontends/pytorch/LICENSE for license information.
//
//===----------------------------------------------------------------------===//
#include "debug.h"
#include <iostream>
namespace torch_mlir {
static bool debugTraceToStderrEnabled = false;
/// Whether debug tracing is enabled and calls to debugTrace() are more than
/// a no-op.
bool isDebugTraceEnabled() { return debugTraceToStderrEnabled; }
/// Writes a message to the debug trace log.
void debugTrace(const std::string &message) {
if (debugTraceToStderrEnabled)
std::cerr << "TORCH_MLIR TRACE: " << message << "\n" << std::flush;
}
/// Enables writing debug traces to stderr.
void enableDebugTraceToStderr() { debugTraceToStderrEnabled = true; }
} // namespace torch_mlir