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

support check

Browse Source
This commit is contained in:
Francesco Gatti
2017-08-14 11:48:29 +02:00
parent 6110fffbb5
commit fc9fb4f153
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/NetworkRT.cpp
+3 -1
View File
@@ -31,8 +31,10 @@ NetworkRT::NetworkRT(Network *net, const char *name) {
float(NV_TENSORRT_MINOR)/10 + float(NV_TENSORRT_MINOR)/10 +
float(NV_TENSORRT_PATCH)/100; float(NV_TENSORRT_PATCH)/100;
std::cout<<"New NetworkRT (TensorRT v"<<rt_ver<<")\n"; std::cout<<"New NetworkRT (TensorRT v"<<rt_ver<<")\n";
builderRT = createInferBuilder(loggerRT); builderRT = createInferBuilder(loggerRT);
std::cout<<"Float16 support: "<<builderRT->platformHasFastFp16()<<"\n";
std::cout<<"Int8 support: "<<builderRT->platformHasFastInt8()<<"\n";
networkRT = builderRT->createNetwork(); networkRT = builderRT->createNetwork();
dtRT = DataType::kFLOAT; dtRT = DataType::kFLOAT;