fix(//tests/accuracy): Fix int8 accuracy test for new PTQ api

Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
pytorch · Oct 23, 2020 · a53bea7 · a53bea7
1 parent 26c3048
commit a53bea7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/accuracy/test_int8_accuracy.cpp b/tests/accuracy/test_int8_accuracy.cpp
@@ -1,6 +1,7 @@
 #include "accuracy_test.h"
 #include "datasets/cifar10.h"
 #include "torch/torch.h"
+#include "trtorch/ptq.h"
 
 TEST_P(AccuracyTests, FP16AccuracyIsClose) {
     auto calibration_dataset = datasets::CIFAR10("tests/accuracy/datasets/data/cifar-10-batches-bin/", datasets::CIFAR10::Mode::kTest)