mirror of https://github.com/llvm/torch-mlir
28 lines
805 B
C
28 lines
805 B
C
|
//===- debug.h --------------------------------------------------*- C++ -*-===//
|
||
|
|
//
|
||
|
|
// This file is licensed under a pytorch-style license
|
||
|
|
// See frontends/pytorch/LICENSE for license information.
|
||
|
|
//
|
||
|
|
//===----------------------------------------------------------------------===//
|
||
|
|
|
||
|
|
#ifndef NPCOMP_FRONTENDS_PYTORCH_CSRC_BUILDER_DEBUG_H
|
||
|
|
#define NPCOMP_FRONTENDS_PYTORCH_CSRC_BUILDER_DEBUG_H
|
||
|
|
|
||
|
|
#include <string>
|
||
|
|
|
||
|
|
namespace torch_mlir {
|
||
|
|
|
||
|
|
/// Whether debug tracing is enabled and calls to debugTrace() are more than
|
||
|
|
/// a no-op.
|
||
|
|
bool isDebugTraceEnabled();
|
||
|
|
|
||
|
|
/// Writes a message to the debug trace log.
|
||
|
|
void debugTrace(const std::string &message);
|
||
|
|
|
||
|
|
/// Enables writing debug traces to stderr.
|
||
|
|
void enableDebugTraceToStderr();
|
||
|
|
|
||
|
|
} // namespace torch_mlir
|
||
|
|
|
||
|
|
#endif // NPCOMP_FRONTENDS_PYTORCH_CSRC_BUILDER_DEBUG_H
|