Merge pull request #868 from inocsin/fix_truncate_long_and_double

Fix truncate long and double

Author: narendasan

cpp/bin/torchtrtc/main.cpp

@@ -460,9 +460,6 @@ int main(int argc, char** argv) {
     compile_settings.debug = true;
   }
 
-  if (use_strict_types) {
-    compile_settings.strict_types = true;
-  }
 
   if (allow_gpu_fallback) {
     compile_settings.device.allow_gpu_fallback = true;
@@ -586,9 +583,6 @@ int main(int argc, char** argv) {
     compile_settings.workspace_size = args::get(workspace_size);
   }
 
-  if (max_batch_size) {
-    compile_settings.max_batch_size = args::get(max_batch_size);
-  }
 
   if (truncate_long_and_double) {
     compile_settings.truncate_long_and_double = true;

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;