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)) {