dla

2019-03-06 16:39:36 +01:00
parent c22219ad16
commit 3714155809
3 changed files with 17 additions and 2 deletions
@@ -51,6 +51,13 @@ NetworkRT::NetworkRT(Network *net, const char *name) {
            dtRT = DataType::kHALF;
            builderRT->setHalf2Mode(true);
        }
+	if(net->dla && builderRT->getNbDLACores() > 0) {
+            dtRT = DataType::kHALF;
+	    builderRT->setFp16Mode(true);
+	    builderRT->allowGPUFallback(true);
+            builderRT->setDefaultDeviceType(DeviceType::kDLA);
+	    builderRT->setDLACore(0);
+	}

        //add input layer
        ITensor *input = networkRT->addInput("data", DataType::kFLOAT,