fix win error

test=develop

fix win error
test=develop
c1264e99 · nhzlx · 4e3522e5 · c1264e99
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/inference/tensorrt/trt_int8_calibrator.h paddle/fluid/inference/tensorrt/trt_int8_calibrator.h +1 -1

未找到文件。
--- 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 <cuda_runtime_api.h>
 #include <atomic>
 #include <memory>
 #include <mutex>
@@ -23,6 +22,7 @@
 #include <vector>

 #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"