bug fixes

posts/TensorRT-introduction/Makefile

@@ -25,7 +25,7 @@
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 CUDA_INSTALL_DIR=/usr/local/cuda
 
-CXXFLAGS=-std=c++11 -Wall -I$(CUDA_INSTALL_DIR)/include
+CXXFLAGS=-std=c++11 -Wall -I$(CUDA_INSTALL_DIR)/include -DONNX_ML=1
 LDFLAGS=-L$(CUDA_INSTALL_DIR)/lib64 -L$(CUDA_INSTALL_DIR)/lib64/stubs 
 LDLIBS=-Wl,--start-group -lnvinfer -lnvonnxparser -lcudart_static -lrt -ldl -lpthread -lonnx -lonnx_proto -lprotobuf -lstdc++ -lm -Wl,--end-group
 

posts/TensorRT-introduction/simpleOnnx.cpp

@@ -34,6 +34,8 @@
 #include <memory>
 #include <string>
 #include <vector>
+#include <cmath>
+#include <numeric>
 
 using namespace nvinfer1;
 using namespace std;

posts/TensorRT-introduction/simpleOnnx_1.cpp

@@ -34,6 +34,8 @@
 #include <memory>
 #include <string>
 #include <vector>
+#include <cmath>
+#include <numeric>
 
 using namespace nvinfer1;
 using namespace std;

posts/TensorRT-introduction/simpleOnnx_2.cpp

@@ -34,6 +34,8 @@
 #include <memory>
 #include <string>
 #include <vector>
+#include <cmath>
+#include <numeric>
 
 using namespace nvinfer1;
 using namespace std;