Skip to content

GitLab

"src/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "2b23ec82e898aa1f0e172da6ef054631634b643d"
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
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
CMakeLists.txt
View file @ 7c8ab626
...@@ -26,6 +26,8 @@ endif() ...@@ -26,6 +26,8 @@ endif()
find_package(Torch REQUIRED) find_package(Torch REQUIRED)
find_package(PNG REQUIRED) find_package(PNG REQUIRED)
find_package(JPEG REQUIRED) find_package(JPEG REQUIRED)
add_definitions(-DJPEG_FOUND)
add_definitions(-DPNG_FOUND)
function(CUDA_CONVERT_FLAGS EXISTING_TARGET) function(CUDA_CONVERT_FLAGS EXISTING_TARGET)
get_property(old_flags TARGET ${EXISTING_TARGET} PROPERTY INTERFACE_COMPILE_OPTIONS) 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