expose logtostderr flag

Author: prigoyal

tensor_comprehensions/pybinds/pybind_engine.cc

@@ -39,6 +39,9 @@ namespace py = pybind11;
 using ATenCudaCompilationUnit = tc::ATenCompilationUnit<tc::CudaTcExecutor>;
 
 PYBIND11_MODULE(tc, m) {
+  m.def("set_logtostderr", [](bool logtostderr) {
+    FLAGS_logtostderr = logtostderr;
+  });
   m.def("set_debug_lang", [](bool debug_lang) {
     tc::FLAGS_debug_lang = debug_lang;
   });

tensor_comprehensions/tc_unit.py

@@ -19,7 +19,7 @@
 from torch.autograd import Variable
 
 from tensor_comprehensions.tc import ATenCompilationUnit
-from tensor_comprehensions.tc import set_debug_lang, set_debug_halide, set_debug_tc_mapper, set_debug_cuda, set_debug_tuner, set_dump_cuda
+from tensor_comprehensions.tc import set_logtostderr, set_debug_lang, set_debug_halide, set_debug_tc_mapper, set_debug_cuda, set_debug_tuner, set_dump_cuda
 from tensor_comprehensions.torch_tc.tc_function import TCFunction, unpack_variables, get_tensors, make_contiguous
 from tensor_comprehensions.autotuner import ATenAutotuner
 from tensor_comprehensions.mapping_options import Options
@@ -49,6 +49,7 @@ def set_gflags(
         self, debug_lang=False, debug_halide=False, debug_tc_mapper=False,
         debug_cuda=False, debug_tuner=False, dump_cuda=False, **kwargs
     ):
+        set_logtostderr(True)
         set_debug_lang(debug_lang)
         set_debug_halide(debug_halide)
         set_debug_tc_mapper(debug_tc_mapper)

test_python/layers/test_dump_cuda.py

@@ -16,7 +16,6 @@
 import tensor_comprehensions as tc
 
 import torch
-import torch.cuda
 import unittest
 
 # enable this to dump cuda code generated whenever tc layer runs: simple run or