From fb1208cccbf2d9ea15319a8a5697960cb758f71e Mon Sep 17 00:00:00 2001 From: dusty-nv Date: Thu, 20 Sep 2018 17:08:00 +0000 Subject: [PATCH] use setDLACore() --- tensorNet.cpp | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tensorNet.cpp b/tensorNet.cpp index f5ac8a80e..b3df7c5fe 100644 --- a/tensorNet.cpp +++ b/tensorNet.cpp @@ -548,9 +548,15 @@ bool tensorNet::LoadNetwork( const char* prototxt_path, const char* model_path, #if NV_TENSORRT_MAJOR >= 5 // if using DLA, set the desired core before deserialization occurs if( device == DEVICE_DLA_0 ) + { + printf(LOG_TRT "device %s, enabling DLA core 0\n", deviceTypeToStr(device)); infer->setDLACore(0); + } else if( device == DEVICE_DLA_1 ) + { + printf(LOG_TRT "device %s, enabling DLA core 1\n", deviceTypeToStr(device)); infer->setDLACore(1); + } #endif #if NV_TENSORRT_MAJOR > 1