support check

2017-08-14 11:48:29 +02:00
parent 6110fffbb5
commit fc9fb4f153
1 changed files with 3 additions and 1 deletions
@@ -33,6 +33,8 @@ NetworkRT::NetworkRT(Network *net, const char *name) {
    std::cout<<"New NetworkRT (TensorRT v"<<rt_ver<<")\n";
  
    builderRT = createInferBuilder(loggerRT);
+    std::cout<<"Float16 support: "<<builderRT->platformHasFastFp16()<<"\n";
+	std::cout<<"Int8 support: "<<builderRT->platformHasFastInt8()<<"\n";
    networkRT = builderRT->createNetwork();
    dtRT = DataType::kFLOAT;