Skip to content

Commit cff4211

Browse files
authored
Merge pull request #138 from NVIDIA/fix_trtorchc
fix(//cpp/trtorchc): Refactor trtorchc to use new C++ API
2 parents 8f94e08 + 789e1be commit cff4211

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

cpp/trtorchc/main.cpp

Lines changed: 3 additions & 0 deletions
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)