feat(//cpp/api): Remove the extra includes in the API header

Signed-off-by: Naren Dasan <[email protected]>
Signed-off-by: Naren Dasan <[email protected]>

Author: narendasan

cpp/api/include/trtorch/ptq.h

@@ -11,6 +11,13 @@ class IInt8Calibrator;
 class IInt8EntropyCalibrator2;
 }
 
+namespace torch {
+namespace data {
+template<typename Example>
+class Iterator;
+}
+}
+
 namespace trtorch {
 namespace ptq {
 

cpp/api/include/trtorch/trtorch.h

@@ -12,9 +12,6 @@
 #include <vector>
 #include <memory>
 
-#include "torch/torch.h"
-#include "NvInfer.h"
-
 // Just include the .h?
 namespace torch {
 namespace jit {

cpp/ptq/main.cpp

@@ -41,8 +41,8 @@ int main(int argc, const char* argv[]) {
 
     std::string calibration_cache_file = "/tmp/vgg16_TRT_ptq_calibration.cache";
 
-    //auto calibrator = trtorch::ptq::make_int8_calibrator(std::move(calibration_dataloader), calibration_cache_file, true);
-    auto calibrator = trtorch::ptq::make_int8_cache_calibrator(calibration_cache_file);
+    auto calibrator = trtorch::ptq::make_int8_calibrator(std::move(calibration_dataloader), calibration_cache_file, true);
+    //auto calibrator = trtorch::ptq::make_int8_cache_calibrator(calibration_cache_file);
 
 
     std::vector<std::vector<int64_t>> input_shape = {{32, 3, 32, 32}};