fix win error

test=develop
6 years ago · c1264e99f3
parent 4e3522e5b4
commit c1264e99f3
1 changed files with 1 additions and 1 deletions
--- 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"