Unverified Commit 7c8ab626 authored by Jeffery (Zeyu) Zhao's avatar Jeffery (Zeyu) Zhao Committed by GitHub
Browse files

Fix libtorchvision.so not able to encode images by adding *_FOUND macros to CMakeLists.txt (#5547)



* Add *_FOUND macros to CMakeLists.txt

* Remove NVJPEG as it fails CI
Co-authored-by: default avatarNicolas Hug <contact@nicolas-hug.com>
parent 787aebc7
......@@ -26,6 +26,8 @@ endif()
find_package(Torch REQUIRED)
find_package(PNG REQUIRED)
find_package(JPEG REQUIRED)
add_definitions(-DJPEG_FOUND)
add_definitions(-DPNG_FOUND)
function(CUDA_CONVERT_FLAGS EXISTING_TARGET)
get_property(old_flags TARGET ${EXISTING_TARGET} PROPERTY INTERFACE_COMPILE_OPTIONS)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment