diff --git a/vstrt/win32.cpp b/vstrt/win32.cpp index b984803..1dd4ffc 100644 --- a/vstrt/win32.cpp +++ b/vstrt/win32.cpp @@ -12,14 +12,14 @@ #include -#if NV_TENSORRT_VERSION >= 100100 +#if NV_TENSORRT_VERSION >= 100001 #define CONCAT_VERSION(name, version) (name "_" #version ".dll") -#endif // NV_TENSORRT_VERSION >= 100100 +#endif // NV_TENSORRT_VERSION >= 100001 namespace { std::vector dlls = { // This list must be sorted by dependency. -#if NV_TENSORRT_VERSION >= 100100 +#if NV_TENSORRT_VERSION >= 100001 #ifdef USE_NVINFER_PLUGIN // nvinfer_plugin dependencies CONCAT_VERSION(L"nvinfer", NV_TENSORRT_MAJOR), @@ -27,7 +27,7 @@ std::vector dlls = { #endif // USE_NVINFER_PLUGIN // Finally, nvinfer again. CONCAT_VERSION(L"nvinfer", NV_TENSORRT_MAJOR), // must be the last -#else // NV_TENSORRT_VERSION >= 100100 +#else // NV_TENSORRT_VERSION >= 100001 #ifdef USE_NVINFER_PLUGIN // nvinfer_plugin dependencies L"nvinfer.dll", @@ -35,7 +35,7 @@ std::vector dlls = { #endif // USE_NVINFER_PLUGIN // Finally, nvinfer again. L"nvinfer.dll", // must be the last -#endif // NV_TENSORRT_VERSION >= 100100 +#endif // NV_TENSORRT_VERSION >= 100001 }; namespace fs = std::filesystem;