tkDNN works with trt8!!,need to test int8 and mobilenet,dla_cnet (fps seems to be a bit low 350 on trt8 compared to 396 on trt7)

This commit is contained in:
perseusdg
2021-09-07 22:00:28 +05:30
parent 9fa116ce4a
commit 03473743c4
30 changed files with 43 additions and 16 deletions
+1
View File
@@ -29,6 +29,7 @@ int main() {
int ret = testInference(input_bins, output_bins, net, netRT);
net->releaseLayers();
delete net;
netRT->destroy();
delete netRT;
return ret;
}
@@ -29,6 +29,7 @@ int main() {
int ret = testInference(input_bins, output_bins, net, netRT);
net->releaseLayers();
delete net;
netRT->destroy();
delete netRT;
return ret;
}
+1
View File
@@ -27,6 +27,7 @@ int main() {
int ret = testInference(input_bins, output_bins, net, netRT);
net->releaseLayers();
delete net;
netRT->destroy();
delete netRT;
return ret;
}
+1
View File
@@ -28,6 +28,7 @@ int main() {
int ret = testInference(input_bins, output_bins, net, netRT);
net->releaseLayers();
delete net;
netRT->destroy();
delete netRT;
return ret;
}
+1
View File
@@ -29,6 +29,7 @@ int main() {
int ret = testInference(input_bins, output_bins, net, netRT);
net->releaseLayers();
delete net;
netRT->destroy();
delete netRT;
return ret;
}
+1
View File
@@ -41,6 +41,7 @@ int main() {
int ret = testInference(input_bins, output_bins, net, netRT);
net->releaseLayers();
delete net;
netRT->destroy();
delete netRT;
return ret;
}
+1
View File
@@ -29,6 +29,7 @@ int main() {
int ret = testInference(input_bins, output_bins, net, netRT);
net->releaseLayers();
delete net;
netRT->destroy();
delete netRT;
return ret;
}
+1
View File
@@ -29,6 +29,7 @@ int main() {
int ret = testInference(input_bins, output_bins, net, netRT);
net->releaseLayers();
delete net;
netRT->destroy();
delete netRT;
return ret;
}
+1
View File
@@ -29,6 +29,7 @@ int main() {
int ret = testInference(input_bins, output_bins, net, netRT);
net->releaseLayers();
delete net;
netRT->destroy();
delete netRT;
return ret;
}
+1
View File
@@ -28,6 +28,7 @@ int main() {
int ret = testInference(input_bins, output_bins, net, netRT);
net->releaseLayers();
delete net;
netRT->destroy();
delete netRT;
return ret;
}
+1
View File
@@ -41,6 +41,7 @@ int main() {
int ret = testInference(input_bins, output_bins, net, netRT);
net->releaseLayers();
delete net;
netRT->destroy();
delete netRT;
return ret;
}
+1
View File
@@ -29,6 +29,7 @@ int main() {
int ret = testInference(input_bins, output_bins, net, netRT);
net->releaseLayers();
delete net;
netRT->destroy();
delete netRT;
return ret;
}
+1
View File
@@ -29,6 +29,7 @@ int main() {
int ret = testInference(input_bins, output_bins, net, netRT);
net->releaseLayers();
delete net;
netRT->destroy();
delete netRT;
return ret;
}
+1
View File
@@ -29,6 +29,7 @@ int main() {
int ret = testInference(input_bins, output_bins, net, netRT);
net->releaseLayers();
delete net;
netRT->destroy();
delete netRT;
return ret;
}
+1
View File
@@ -42,6 +42,7 @@ int main() {
int ret = testInference(input_bins, output_bins, net, netRT);
net->releaseLayers();
delete net;
netRT->destroy();
delete netRT;
return ret;
}
+1
View File
@@ -29,6 +29,7 @@ int main() {
int ret = testInference(input_bins, output_bins, net, netRT);
net->releaseLayers();
delete net;
netRT->destroy();
delete netRT;
return ret;
}
+1
View File
@@ -29,6 +29,7 @@ int main() {
int ret = testInference(input_bins, output_bins, net, netRT);
net->releaseLayers();
delete net;
netRT->destroy();
delete netRT;
return ret;
}
+1
View File
@@ -29,6 +29,7 @@ int main() {
int ret = testInference(input_bins, output_bins, net, netRT);
net->releaseLayers();
delete net;
netRT->destroy();
delete netRT;
return ret;
}
+1
View File
@@ -29,6 +29,7 @@ int main() {
int ret = testInference(input_bins, output_bins, net, netRT);
net->releaseLayers();
delete net;
netRT->destroy();
delete netRT;
return ret;
}
+1
View File
@@ -40,6 +40,7 @@ int main() {
int ret = testInference(input_bins, output_bins, net, netRT);
net->releaseLayers();
delete net;
netRT->destroy();
delete netRT;
return ret;
}
+1
View File
@@ -31,6 +31,7 @@ int main() {
int ret = testInference(input_bins, output_bins, net, netRT);
net->releaseLayers();
delete net;
netRT->destroy();
delete netRT;
return ret;
}
@@ -542,5 +542,6 @@ int main()
ret_cudnn_tensorrt |= checkResult(conf->output_dim.tot(), conf->dstData, rt_out3) == 0 ? 0 : ERROR_CUDNNvsTENSORRT;
ret_cudnn_tensorrt |= checkResult(loc->output_dim.tot(), loc->dstData, rt_out4) == 0 ? 0 : ERROR_CUDNNvsTENSORRT;
netRT.destroy();
return ret_cudnn | ret_tensorrt | ret_cudnn_tensorrt;
}