@@ -360,26 +360,21 @@ class MappingOptionsCache {
360
360
PYBIND11_MODULE (tclib, m) {
361
361
m.doc () = " Python bindings for Tensor Comprehensions" ;
362
362
363
- m.def (" set_logtostderr" , [](bool logtostderr) {
364
- FLAGS_logtostderr = logtostderr;
365
- });
366
- m.def (" set_debug_lang" , [](bool debug_lang) {
367
- tc::FLAGS_debug_lang = debug_lang;
368
- });
369
- m.def (" set_debug_halide" , [](bool debug_halide) {
363
+ // Simple functions to set up debugging
364
+ m.def (
365
+ " logtostderr" , [](bool logtostderr) { FLAGS_logtostderr = logtostderr; });
366
+ m.def (
367
+ " debug_lang" , [](bool debug_lang) { tc::FLAGS_debug_lang = debug_lang; });
368
+ m.def (" debug_halide" , [](bool debug_halide) {
370
369
tc::FLAGS_debug_halide = debug_halide;
371
370
});
372
- m.def (" set_debug_tc_mapper " , [](bool debug_tc_mapper) {
371
+ m.def (" debug_tc_mapper " , [](bool debug_tc_mapper) {
373
372
tc::FLAGS_debug_tc_mapper = debug_tc_mapper;
374
373
});
375
- m.def (" set_debug_cuda" , [](bool debug_cuda) {
376
- tc::FLAGS_debug_cuda = debug_cuda;
377
- });
378
- m.def (" set_debug_tuner" , [](bool debug_tuner) {
374
+ m.def (" debug_tuner" , [](bool debug_tuner) {
379
375
tc::FLAGS_debug_tuner = debug_tuner;
380
376
});
381
- m.def (
382
- " set_dump_cuda" , [](bool dump_cuda) { tc::FLAGS_dump_cuda = dump_cuda; });
377
+ m.def (" dump_cuda" , [](bool dump_cuda) { tc::FLAGS_dump_cuda = dump_cuda; });
383
378
384
379
py::class_<TcExecutor>(m, " TcExecutor" , py::module_local ())
385
380
.def (" run" , &TcExecutor::run)
0 commit comments