mirror of https://github.com/llvm/torch-mlir
30 lines
891 B
C++
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
|