amirstan_plugin
amirstan_plugin copied to clipboard
some issues when build with win10 use clion cuda11.1 tensorrt 8.0
batchedNMSPlugin/kernel.h line 32 std::max->std::fmax; batchedNMSPlugin batchedNMSInference.cpp and batchedNMSPlugin.h nmsInference is not the same; torchCumPlugin torchCumPlugin.cpp line 58 or->||;
cmakeslist
- find_package(CUDA REQUIRED) changed to set(CMAKE_SIZEOF_VOID_P 8) FIND_PACKAGE(CUDA REQUIRED);
- add libraries when use cublas:target_link_libraries(${SHARED_TARGET or STATIC_TARGET} ${CUDA_cublas_LIBRARY})
Thanks for the notification! Would you like to create a PR to fix it?