From 828336db22cb65c12a39cecf9b51b0388e6d7e22 Mon Sep 17 00:00:00 2001 From: inocsin Date: Mon, 14 Feb 2022 15:24:43 +0800 Subject: [PATCH] fix: fix bug that python api doesn't pass truncate_long_and_double value to internal.partition_info Signed-off-by: inocsin --- py/torch_tensorrt/csrc/tensorrt_classes.cpp | 1 + 1 file changed, 1 insertion(+) diff --git a/py/torch_tensorrt/csrc/tensorrt_classes.cpp b/py/torch_tensorrt/csrc/tensorrt_classes.cpp index 2361abd457..a89fe692bd 100644 --- a/py/torch_tensorrt/csrc/tensorrt_classes.cpp +++ b/py/torch_tensorrt/csrc/tensorrt_classes.cpp @@ -216,6 +216,7 @@ core::CompileSpec CompileSpec::toInternalCompileSpec() { info.partition_info.enabled = torch_fallback.enabled; info.partition_info.min_block_size = torch_fallback.min_block_size; info.partition_info.forced_fallback_operators = torch_fallback.forced_fallback_operators; + info.partition_info.truncate_long_and_double = truncate_long_and_double; info.lower_info.forced_fallback_modules = torch_fallback.forced_fallback_modules; info.convert_info.engine_settings.truncate_long_and_double = truncate_long_and_double;