From c1264e99f3a213b7cb8ff659aa998e17f47ef937 Mon Sep 17 00:00:00 2001 From: nhzlx Date: Wed, 9 Jan 2019 10:20:49 +0000 Subject: [PATCH] fix win error test=develop --- paddle/fluid/inference/tensorrt/trt_int8_calibrator.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/fluid/inference/tensorrt/trt_int8_calibrator.h b/paddle/fluid/inference/tensorrt/trt_int8_calibrator.h index 81ba9c7032..13f6e7ad01 100644 --- a/paddle/fluid/inference/tensorrt/trt_int8_calibrator.h +++ b/paddle/fluid/inference/tensorrt/trt_int8_calibrator.h @@ -13,7 +13,6 @@ // limitations under the License. #pragma once -#include #include #include #include @@ -23,6 +22,7 @@ #include #include "NvInfer.h" +#include "cuda_runtime_api.h" #include "paddle/fluid/framework/tensor.h" #include "paddle/fluid/inference/tensorrt/engine.h" #include "paddle/fluid/platform/place.h" -- GitLab