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

Compile on tx2

Browse Source 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
This commit is contained in:
Micaela Verucchi
2020-04-08 16:23:08 +02:00
parent 35e86ca5cb
commit 6d1fda0c21
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/NetworkRT.cpp
+4
View File
@@ -36,7 +36,9 @@ 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";
#if NV_TENSORRT_MAJOR >= 5
std::cout<<"DLAs: "<<builderRT->getNbDLACores()<<"\n"; std::cout<<"DLAs: "<<builderRT->getNbDLACores()<<"\n";
#endif
networkRT = builderRT->createNetwork(); networkRT = builderRT->createNetwork();
#if NV_TENSORRT_MAJOR >= 6 #if NV_TENSORRT_MAJOR >= 6
configRT = builderRT->createBuilderConfig(); configRT = builderRT->createBuilderConfig();
@@ -66,6 +68,7 @@ NetworkRT::NetworkRT(Network *net, const char *name) {
configRT->setFlag(BuilderFlag::kFP16); configRT->setFlag(BuilderFlag::kFP16);
#endif #endif
} }
#if NV_TENSORRT_MAJOR >= 5
if(net->dla && builderRT->getNbDLACores() > 0) { if(net->dla && builderRT->getNbDLACores() > 0) {
dtRT = DataType::kHALF; dtRT = DataType::kHALF;
builderRT->setFp16Mode(true); builderRT->setFp16Mode(true);
@@ -73,6 +76,7 @@ NetworkRT::NetworkRT(Network *net, const char *name) {
builderRT->setDefaultDeviceType(DeviceType::kDLA); builderRT->setDefaultDeviceType(DeviceType::kDLA);
builderRT->setDLACore(0); builderRT->setDLACore(0);
} }
#endif
#if NV_TENSORRT_MAJOR >= 6 #if NV_TENSORRT_MAJOR >= 6
if(net->int8 && builderRT->platformHasFastInt8()){ if(net->int8 && builderRT->platformHasFastInt8()){
// dtRT = DataType::kINT8; // dtRT = DataType::kINT8;