From 6d1fda0c216dacfd9335f49ca7c66d7c8e385f6c Mon Sep 17 00:00:00 2001
From: Micaela Verucchi <micaelaverucchi@gmail.com>
Date: Wed, 8 Apr 2020 16:23:08 +0200
Subject: [PATCH] Compile on tx2

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
---
 src/NetworkRT.cpp | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/NetworkRT.cpp b/src/NetworkRT.cpp
index bf74010..da58ddf 100644
--- a/src/NetworkRT.cpp
+++ b/src/NetworkRT.cpp
@@ -36,7 +36,9 @@ NetworkRT::NetworkRT(Network *net, const char *name) {
     builderRT = createInferBuilder(loggerRT);
     std::cout<<"Float16 support: "<<builderRT->platformHasFastFp16()<<"\n";
     std::cout<<"Int8 support: "<<builderRT->platformHasFastInt8()<<"\n";
+#if NV_TENSORRT_MAJOR >= 5
     std::cout<<"DLAs: "<<builderRT->getNbDLACores()<<"\n";
+#endif
     networkRT = builderRT->createNetwork();
 #if NV_TENSORRT_MAJOR >= 6                
         configRT = builderRT->createBuilderConfig();
@@ -66,6 +68,7 @@ NetworkRT::NetworkRT(Network *net, const char *name) {
             configRT->setFlag(BuilderFlag::kFP16);
 #endif
         }
+#if NV_TENSORRT_MAJOR >= 5
         if(net->dla && builderRT->getNbDLACores() > 0) {
             dtRT = DataType::kHALF;
             builderRT->setFp16Mode(true);
@@ -73,6 +76,7 @@ NetworkRT::NetworkRT(Network *net, const char *name) {
             builderRT->setDefaultDeviceType(DeviceType::kDLA);
             builderRT->setDLACore(0);
         }
+#endif
 #if NV_TENSORRT_MAJOR >= 6                
         if(net->int8 && builderRT->platformHasFastInt8()){
             // dtRT = DataType::kINT8;