vstrt/win32.cpp: update dummy() signature

AmusementClub · Dec 2, 2023 · 9b73fa0 · 9b73fa0
1 parent 2b7c1cb
commit 9b73fa0
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/vstrt/win32.cpp b/vstrt/win32.cpp
@@ -10,6 +10,8 @@
 
 #include <iostream>
 
+#include <NvInferVersion.h>
+
 namespace {
 std::vector<std::wstring> dlls = {
 	// This list must be sorted by dependency.
@@ -63,7 +65,11 @@ FARPROC loadDLLs() {
 	return (FARPROC)h;
 }
 
+#if ((NV_TENSORRT_MAJOR * 1000) + (NV_TENSORRT_MINOR * 100) + NV_TENSORRT_PATCH) == 9100 && defined(_WIN32)
+static void * dummy() { // mimic getPluginRegistry
+#else
 static int dummy() { // mimic getInferLibVersion
+#endif
 	return 0;
 }