diff --git a/py/trtorch/csrc/trtorch_py.cpp b/py/trtorch/csrc/trtorch_py.cpp index 747d9dcd73..6f6c06df9a 100644 --- a/py/trtorch/csrc/trtorch_py.cpp +++ b/py/trtorch/csrc/trtorch_py.cpp @@ -115,7 +115,8 @@ PYBIND11_MODULE(_C, m) { .def_readwrite("num_min_timing_iters", &CompileSpec::num_min_timing_iters) .def_readwrite("num_avg_timing_iters", &CompileSpec::num_avg_timing_iters) .def_readwrite("workspace_size", &CompileSpec::workspace_size) - .def_readwrite("max_batch_size", &CompileSpec::max_batch_size); + .def_readwrite("max_batch_size", &CompileSpec::max_batch_size) + .def_readwrite("torch_fallback", &CompileSpec::torch_fallback); py::class_(m, "Device") .def(py::init<>())