fix(//cpp/trtorchc): Refactor trtorchc to use new C++ API

Addresses issue #132 Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
pytorch · Jul 16, 2020 · 789e1be · 789e1be
1 parent 8f94e08
commit 789e1be
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/cpp/trtorchc/main.cpp b/cpp/trtorchc/main.cpp
@@ -13,7 +13,10 @@
 #include "third_party/args/args.hpp"
 #include "torch/torch.h"
 #include "torch/script.h"
+
+#include "trtorch/logging.h"
 #include "trtorch/trtorch.h"
+#include "trtorch/ptq.h"
 
 bool checkRtol(const at::Tensor& diff, const std::vector<at::Tensor> inputs, float threshold) {
     double maxValue = 0.0;