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:
@@ -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;
|
||||
}
|
||||
|
||||
@@ -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;
|
||||
}
|
||||
|
||||
@@ -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;
|
||||
}
|
||||
|
||||
@@ -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;
|
||||
}
|
||||
@@ -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;
|
||||
}
|
||||
|
||||
@@ -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;
|
||||
}
|
||||
|
||||
@@ -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;
|
||||
}
|
||||
|
||||
@@ -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;
|
||||
}
|
||||
|
||||
@@ -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;
|
||||
}
|
||||
|
||||
@@ -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;
|
||||
}
|
||||
|
||||
@@ -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;
|
||||
}
|
||||
|
||||
@@ -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;
|
||||
}
|
||||
|
||||
@@ -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;
|
||||
}
|
||||
|
||||
@@ -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;
|
||||
}
|
||||
|
||||
@@ -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;
|
||||
}
|
||||
|
||||
@@ -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;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user