diff --git a/cpp/trtorchc/main.cpp b/cpp/trtorchc/main.cpp index 5dab59a4ea..cd971f8d8e 100644 --- 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 inputs, float threshold) { double maxValue = 0.0;