We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 100b090 commit 8b7919fCopy full SHA for 8b7919f
py/trtorch/csrc/trtorch_py.cpp
@@ -115,7 +115,8 @@ PYBIND11_MODULE(_C, m) {
115
.def_readwrite("num_min_timing_iters", &CompileSpec::num_min_timing_iters)
116
.def_readwrite("num_avg_timing_iters", &CompileSpec::num_avg_timing_iters)
117
.def_readwrite("workspace_size", &CompileSpec::workspace_size)
118
- .def_readwrite("max_batch_size", &CompileSpec::max_batch_size);
+ .def_readwrite("max_batch_size", &CompileSpec::max_batch_size)
119
+ .def_readwrite("torch_fallback", &CompileSpec::torch_fallback);
120
121
py::class_<Device>(m, "Device")
122
.def(py::init<>())
0 commit comments