diff --git a/src/NetworkRT.cpp b/src/NetworkRT.cpp index 2fce5e6..2a73efa 100644 --- a/src/NetworkRT.cpp +++ b/src/NetworkRT.cpp @@ -35,7 +35,7 @@ NetworkRT::NetworkRT(Network *net, const char *name) { builderRT = createInferBuilder(loggerRT); std::cout<<"Float16 support: "<platformHasFastFp16()<<"\n"; std::cout<<"Int8 support: "<platformHasFastInt8()<<"\n"; - std::cout<<"DLAs: "<getNbDLACores()<<"\n"; + //std::cout<<"DLAs: "<getNbDLACores()<<"\n"; networkRT = builderRT->createNetwork(); if(!fileExist(name)) { @@ -51,14 +51,16 @@ NetworkRT::NetworkRT(Network *net, const char *name) { dtRT = DataType::kHALF; builderRT->setHalf2Mode(true); } - if(net->dla && builderRT->getNbDLACores() > 0) { + /* + if(net->dla && builderRT->getNbDLACores() > 0) { dtRT = DataType::kHALF; - builderRT->setFp16Mode(true); - builderRT->allowGPUFallback(true); + builderRT->setFp16Mode(true); + builderRT->allowGPUFallback(true); builderRT->setDefaultDeviceType(DeviceType::kDLA); - builderRT->setDLACore(0); - } - + builderRT->setDLACore(0); + } + */ + //add input layer ITensor *input = networkRT->addInput("data", DataType::kFLOAT, DimsCHW{ dim.c, dim.h, dim.w});