Skip to content

Commit 789e1be

Browse files
committed
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]>
1 parent 8f94e08 commit 789e1be

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

Diff for: cpp/trtorchc/main.cpp

+3
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,10 @@
1313
#include "third_party/args/args.hpp"
1414
#include "torch/torch.h"
1515
#include "torch/script.h"
16+
17+
#include "trtorch/logging.h"
1618
#include "trtorch/trtorch.h"
19+
#include "trtorch/ptq.h"
1720

1821
bool checkRtol(const at::Tensor& diff, const std::vector<at::Tensor> inputs, float threshold) {
1922
double maxValue = 0.0;

0 commit comments

Comments
 (0)