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

Fixed tensorrt8 branch to work jetpack 4.5 and tensorrt7

Browse Source 
Signed-off-by: perseusdg <f20180523@goa.bits-pilani.ac.in>
This commit was merged in pull request #278.
This commit is contained in:
Harshvardhan Chandirasekar
2022-03-16 18:09:32 +05:30
committed by perseusdg
parent 480b5a9c5a
commit 40266a6c32
3 changed files with 17 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/NetworkRT.cpp
+8 -1
View File
@@ -834,7 +834,7 @@ IPluginV2Layer* NetworkRT::convert_layer(ITensor *input, Yolo *l) {
return lRT;
}
IResizeLayer* NetworkRT::convert_layer(ITensor *input, Upsample *l) {
ILayer* NetworkRT::convert_layer(ITensor *input, Upsample *l) {
#if NV_TENSORRT_MAJOR < 8
auto creator = getPluginRegistry()->getPluginCreator("UpSample_tkDNN","1");
@@ -1008,6 +1008,7 @@ bool NetworkRT::deserialize(const char *filename) {
return true;
}
#if NV_TENSORRT_MAJOR > 7
void NetworkRT::destroy() {
delete contextRT;
if(builderActive) {
@@ -1015,5 +1016,11 @@ void NetworkRT::destroy() {
delete builderRT;
}
}
#elif NV_TENSORRT_MAJOR <=7
void NetworkRT::destroy() {
}
#endif
}}