From c22219ad16ced13d25326311302152ae855de900 Mon Sep 17 00:00:00 2001 From: Francesco Gatti Date: Wed, 6 Mar 2019 13:02:39 +0100 Subject: [PATCH] DLA number print --- src/NetworkRT.cpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/NetworkRT.cpp b/src/NetworkRT.cpp index 671df8b..93dd321 100644 --- a/src/NetworkRT.cpp +++ b/src/NetworkRT.cpp @@ -34,7 +34,8 @@ NetworkRT::NetworkRT(Network *net, const char *name) { builderRT = createInferBuilder(loggerRT); std::cout<<"Float16 support: "<platformHasFastFp16()<<"\n"; - std::cout<<"Int8 support: "<platformHasFastInt8()<<"\n"; + std::cout<<"Int8 support: "<platformHasFastInt8()<<"\n"; + std::cout<<"DLAs: "<getNbDLACores()<<"\n"; networkRT = builderRT->createNetwork(); if(!fileExist(name)) {