We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 4a516ed commit 828336dCopy full SHA for 828336d
py/torch_tensorrt/csrc/tensorrt_classes.cpp
@@ -216,6 +216,7 @@ core::CompileSpec CompileSpec::toInternalCompileSpec() {
216
info.partition_info.enabled = torch_fallback.enabled;
217
info.partition_info.min_block_size = torch_fallback.min_block_size;
218
info.partition_info.forced_fallback_operators = torch_fallback.forced_fallback_operators;
219
+ info.partition_info.truncate_long_and_double = truncate_long_and_double;
220
info.lower_info.forced_fallback_modules = torch_fallback.forced_fallback_modules;
221
info.convert_info.engine_settings.truncate_long_and_double = truncate_long_and_double;
222
0 commit comments