@@ -447,7 +447,9 @@ PYBIND11_MODULE(tclib, m) {
447
447
return res;
448
448
});
449
449
450
- py::class_<TcExecutor>(m, " TcExecutor" , py::module_local ())
450
+ // Low-level stateful API compile returns an executor on which run and
451
+ // unchecked_run can be called.
452
+ py::class_<TcExecutor>(m, " TcExecutor" )
451
453
.def (
452
454
" run" ,
453
455
&TcExecutor::run,
@@ -470,7 +472,7 @@ PYBIND11_MODULE(tclib, m) {
470
472
});
471
473
472
474
// A TunerConfig object can be passed to configure a tuning run
473
- py::class_<TunerConfig>(m, " TunerConfig" , py::module_local (), R"DOC(
475
+ py::class_<TunerConfig>(m, " TunerConfig" , R"DOC(
474
476
Helper class to manage the behavior of the autotuner
475
477
)DOC" )
476
478
.def (py::init<>())
@@ -535,7 +537,7 @@ PYBIND11_MODULE(tclib, m) {
535
537
gflags::GetCommandLineFlagInfoOrDie (" stderrthreshold" ))
536
538
.c_str ());
537
539
538
- py::class_<Tuner>(m, " Tuner" , py::module_local () )
540
+ py::class_<Tuner>(m, " Tuner" )
539
541
.def (py::init<std::string>())
540
542
.def (py::init<std::string, std::string>())
541
543
.def (
@@ -563,8 +565,7 @@ PYBIND11_MODULE(tclib, m) {
563
565
}
564
566
});
565
567
566
- py::class_<MappingOptionsCache>(
567
- m, " MappingOptionsCache" , py::module_local (), R"DOC(
568
+ py::class_<MappingOptionsCache>(m, " MappingOptionsCache" , R"DOC(
568
569
Helper class to manipulate cache files containing serialized :class:`MappingOptions <tensor_comprehensions.tclib.MappingOptions>`
569
570
)DOC" )
570
571
.def (py::init<std::string>())
@@ -590,7 +591,7 @@ PYBIND11_MODULE(tclib, m) {
590
591
A vector of :class:`MappingOptions <tensor_comprehensions.tclib.MappingOptions>`
591
592
)DOC" );
592
593
593
- py::class_<CompilationCache>(m, " CompilationCache" , py::module_local () )
594
+ py::class_<CompilationCache>(m, " CompilationCache" )
594
595
.def (py::init<std::string>())
595
596
.def (" is_compiled" , &CompilationCache::isCompiled)
596
597
.def (" alloc_outputs" , &CompilationCache::allocOutputs)
0 commit comments