From 3b910caa73aab4e92941e34a26711211f686f643 Mon Sep 17 00:00:00 2001 From: Anurag Dixit Date: Thu, 18 Aug 2022 19:02:21 -0700 Subject: [PATCH] chore: Truncate long and double for ptq CPP path Signed-off-by: Anurag Dixit --- examples/int8/ptq/main.cpp | 1 + 1 file changed, 1 insertion(+) diff --git a/examples/int8/ptq/main.cpp b/examples/int8/ptq/main.cpp index 4cc38148b6..e69227fb5a 100644 --- a/examples/int8/ptq/main.cpp +++ b/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;