Skip to content

Commit 8b7919f

Browse files
committed
fix: register the torch_fallback attribute in Python API
Signed-off-by: Bo Wang <[email protected]>
1 parent 100b090 commit 8b7919f

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

Diff for: py/trtorch/csrc/trtorch_py.cpp

+2-1
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,8 @@ PYBIND11_MODULE(_C, m) {
115115
.def_readwrite("num_min_timing_iters", &CompileSpec::num_min_timing_iters)
116116
.def_readwrite("num_avg_timing_iters", &CompileSpec::num_avg_timing_iters)
117117
.def_readwrite("workspace_size", &CompileSpec::workspace_size)
118-
.def_readwrite("max_batch_size", &CompileSpec::max_batch_size);
118+
.def_readwrite("max_batch_size", &CompileSpec::max_batch_size)
119+
.def_readwrite("torch_fallback", &CompileSpec::torch_fallback);
119120

120121
py::class_<Device>(m, "Device")
121122
.def(py::init<>())

0 commit comments

Comments
 (0)