mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

dla commented

Browse Source
This commit is contained in:
Francesco Gatti
2019-03-07 17:40:24 +01:00
parent 3714155809
commit a85367fa22
1 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/NetworkRT.cpp
+9 -7
View File
@@ -35,7 +35,7 @@ NetworkRT::NetworkRT(Network *net, const char *name) {
builderRT = createInferBuilder(loggerRT); builderRT = createInferBuilder(loggerRT);
std::cout<<"Float16 support: "<<builderRT->platformHasFastFp16()<<"\n"; std::cout<<"Float16 support: "<<builderRT->platformHasFastFp16()<<"\n";
std::cout<<"Int8 support: "<<builderRT->platformHasFastInt8()<<"\n"; std::cout<<"Int8 support: "<<builderRT->platformHasFastInt8()<<"\n";
std::cout<<"DLAs: "<<builderRT->getNbDLACores()<<"\n"; //std::cout<<"DLAs: "<<builderRT->getNbDLACores()<<"\n";
networkRT = builderRT->createNetwork(); networkRT = builderRT->createNetwork();
if(!fileExist(name)) { if(!fileExist(name)) {
@@ -51,14 +51,16 @@ NetworkRT::NetworkRT(Network *net, const char *name) {
dtRT = DataType::kHALF; dtRT = DataType::kHALF;
builderRT->setHalf2Mode(true); builderRT->setHalf2Mode(true);
} }
if(net->dla && builderRT->getNbDLACores() > 0) { /*
if(net->dla && builderRT->getNbDLACores() > 0) {
dtRT = DataType::kHALF; dtRT = DataType::kHALF;
builderRT->setFp16Mode(true); builderRT->setFp16Mode(true);
builderRT->allowGPUFallback(true); builderRT->allowGPUFallback(true);
builderRT->setDefaultDeviceType(DeviceType::kDLA); builderRT->setDefaultDeviceType(DeviceType::kDLA);
builderRT->setDLACore(0); builderRT->setDLACore(0);
} }
*/
//add input layer //add input layer
ITensor *input = networkRT->addInput("data", DataType::kFLOAT, ITensor *input = networkRT->addInput("data", DataType::kFLOAT,
DimsCHW{ dim.c, dim.h, dim.w}); DimsCHW{ dim.c, dim.h, dim.w});