fix: fix bug that python api doesn't pass truncate_long_and_double va…

…lue to internal.partition_info Signed-off-by: inocsin <[email protected]>
pytorch · Feb 14, 2022 · 828336d · 828336d
1 parent 4a516ed
commit 828336d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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;