Skip to content

Commit a53bea7

Browse files
committed
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]>
1 parent 26c3048 commit a53bea7

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

Diff for: tests/accuracy/test_int8_accuracy.cpp

+1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
#include "accuracy_test.h"
22
#include "datasets/cifar10.h"
33
#include "torch/torch.h"
4+
#include "trtorch/ptq.h"
45

56
TEST_P(AccuracyTests, FP16AccuracyIsClose) {
67
auto calibration_dataset = datasets::CIFAR10("tests/accuracy/datasets/data/cifar-10-batches-bin/", datasets::CIFAR10::Mode::kTest)

0 commit comments

Comments
 (0)