Merge pull request #1291 from pytorch/anuragd/float64_truncate_ptq

chore: Truncate long and double for ptq CPP path

Author: peri044

examples/int8/ptq/main.cpp

@@ -50,6 +50,7 @@ torch::jit::Module compile_int8_model(const std::string& data_dir, torch::jit::M
   compile_spec.enabled_precisions.insert(torch::kI8);
   /// Use the TensorRT Entropy Calibrator
   compile_spec.ptq_calibrator = calibrator;
+  compile_spec.truncate_long_and_double = true;
 
 #ifdef SAVE_ENGINE
   std::cout << "Compiling graph to save as TRT engine (/tmp/engine_converted_from_jit.trt)" << std::endl;