diff --git a/paddle/fluid/inference/tensorrt/trt_int8_calibrator.h b/paddle/fluid/inference/tensorrt/trt_int8_calibrator.h index 81ba9c7032c2022c6fc3c4421692564d7f6b8740..13f6e7ad011833991262a556978f08494d230949 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"