updating trt workspace int64 (#1094)

Signed-off-by: Manix <manickavela1998@gmail.com>
2024-07-08 18:08:16 +05:30
parent 4fd0493037
commit 3e4307e2fb
3 changed files with 4 additions and 4 deletions
--- a/sherpa-onnx/python/csrc/tensorrt-config.cc
+++ b/sherpa-onnx/python/csrc/tensorrt-config.cc
@@ -14,7 +14,7 @@ void PybindTensorrtConfig(py::module *m) {
  using PyClass = TensorrtConfig;
  py::class_<PyClass>(*m, "TensorrtConfig")
        .def(py::init<>())
-        .def(py::init([](int32_t trt_max_workspace_size,
+        .def(py::init([](int64_t trt_max_workspace_size,
                      int32_t trt_max_partition_iterations,
                      int32_t trt_min_subgraph_size,
                      bool trt_fp16_enable,