Skip to content

Commit 828336d

Browse files
committed
fix: fix bug that python api doesn't pass truncate_long_and_double value to internal.partition_info
Signed-off-by: inocsin <[email protected]>
1 parent 4a516ed commit 828336d

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

Diff for: py/torch_tensorrt/csrc/tensorrt_classes.cpp

+1
Original file line numberDiff line numberDiff line change
@@ -216,6 +216,7 @@ core::CompileSpec CompileSpec::toInternalCompileSpec() {
216216
info.partition_info.enabled = torch_fallback.enabled;
217217
info.partition_info.min_block_size = torch_fallback.min_block_size;
218218
info.partition_info.forced_fallback_operators = torch_fallback.forced_fallback_operators;
219+
info.partition_info.truncate_long_and_double = truncate_long_and_double;
219220
info.lower_info.forced_fallback_modules = torch_fallback.forced_fallback_modules;
220221
info.convert_info.engine_settings.truncate_long_and_double = truncate_long_and_double;
221222

0 commit comments

Comments
 (0)