From e0af522dac4a7f7cb8a14d30d25e51275a206b7c Mon Sep 17 00:00:00 2001
From: thoumine <nathan.thoumine@cea.fr>
Date: Mon, 29 Jan 2024 08:49:25 +0000
Subject: [PATCH] Fix includes in Graph.cpp and BatchSteam.hpp
---
.../static/tensorrt_8.6/include/BatchStream.hpp | 4 +---
aidge_export_tensorrt/static/tensorrt_8.6/src/Graph.cpp | 3 +++
2 files changed, 4 insertions(+), 3 deletions(-)
diff --git a/aidge_export_tensorrt/static/tensorrt_8.6/include/BatchStream.hpp b/aidge_export_tensorrt/static/tensorrt_8.6/include/BatchStream.hpp
index 46f10bd..50aa8c7 100755
--- a/aidge_export_tensorrt/static/tensorrt_8.6/include/BatchStream.hpp
+++ b/aidge_export_tensorrt/static/tensorrt_8.6/include/BatchStream.hpp
@@ -2,7 +2,6 @@
#define BATCH_STREAM_H
#include "NvInfer.h"
-#include "dnn_utils.hpp"
#include <algorithm>
#include <assert.h>
#include <stdio.h>
@@ -67,8 +66,7 @@ public:
}
int x = _batchCount;
- for (int i = 0; i < skipCount; i++)
- next();
+ for (std::size_t i = 0; i < skipCount; ++i) next();
_batchCount = x;
}
diff --git a/aidge_export_tensorrt/static/tensorrt_8.6/src/Graph.cpp b/aidge_export_tensorrt/static/tensorrt_8.6/src/Graph.cpp
index 61308b9..6016925 100644
--- a/aidge_export_tensorrt/static/tensorrt_8.6/src/Graph.cpp
+++ b/aidge_export_tensorrt/static/tensorrt_8.6/src/Graph.cpp
@@ -1,6 +1,9 @@
#include "Graph.hpp"
#include <fstream>
#include <sstream>
+#include "BatchStream.hpp"
+#include "IInt8EntropyCalibrator.hpp"
+#include <dirent.h>
Graph::Graph(std::string const& filePath,
unsigned int device_id,
--
GitLab