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: Nicolas Hug <contact@nicolas-hug.com>

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: Nicolas Hug <contact@nicolas-hug.com>
7c8ab626 · Jeffery (Zeyu) Zhao · GitHub · 787aebc7 · 7c8ab626
Unverified Commit 7c8ab626 authored May 03, 2022 by Jeffery (Zeyu) Zhao Committed by GitHub May 03, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

CMakeLists.txt CMakeLists.txt +2 -0

No files found.
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -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)