xref: /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/onnx_log.cpp (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1 #include <torch/csrc/jit/passes/onnx/onnx_log.h>
2 #include <iostream>
3 
4 namespace torch::jit::onnx {
5 
6 namespace {
7 bool log_enabled = false;
8 std::shared_ptr<std::ostream> out;
9 } // namespace
10 
is_log_enabled()11 bool is_log_enabled() {
12   return log_enabled;
13 }
14 
set_log_enabled(bool enabled)15 void set_log_enabled(bool enabled) {
16   log_enabled = enabled;
17 }
18 
set_log_output_stream(std::shared_ptr<std::ostream> out_stream)19 void set_log_output_stream(std::shared_ptr<std::ostream> out_stream) {
20   out = std::move(out_stream);
21 }
22 
_get_log_output_stream()23 std::ostream& _get_log_output_stream() {
24   return out ? *out : std::cout;
25 }
26 
27 } // namespace torch::jit::onnx
28