Skip to content
This repository was archived by the owner on Apr 28, 2023. It is now read-only.

Commit fb0d365

Browse files
Rename functions to enable debugging
1 parent 3a36e60 commit fb0d365

File tree

1 file changed

+9
-14
lines changed

1 file changed

+9
-14
lines changed

tensor_comprehensions/pybinds/tclib.cc

Lines changed: 9 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -360,26 +360,21 @@ class MappingOptionsCache {
360360
PYBIND11_MODULE(tclib, m) {
361361
m.doc() = "Python bindings for Tensor Comprehensions";
362362

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) {
370369
tc::FLAGS_debug_halide = debug_halide;
371370
});
372-
m.def("set_debug_tc_mapper", [](bool debug_tc_mapper) {
371+
m.def("debug_tc_mapper", [](bool debug_tc_mapper) {
373372
tc::FLAGS_debug_tc_mapper = debug_tc_mapper;
374373
});
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) {
379375
tc::FLAGS_debug_tuner = debug_tuner;
380376
});
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; });
383378

384379
py::class_<TcExecutor>(m, "TcExecutor", py::module_local())
385380
.def("run", &TcExecutor::run)

0 commit comments

Comments
 (0)