Skip to content

Commit fa950ad

Browse files
authored
Merge pull request #198 from NVIDIA/fix_int8_test
fix(//tests/accuracy): Fix int8 accuracy test for new PTQ api
2 parents 26c3048 + a53bea7 commit fa950ad

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/accuracy/test_int8_accuracy.cpp

Lines changed: 1 addition & 0 deletions
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)