Allows JNI to be compiled on headless jdk (deepjavalibrary#2098)

KexinFeng · Oct 26, 2022 · 37e2b2e · 37e2b2e
1 parent 7a407ba
commit 37e2b2e
Show file tree

Hide file tree

Showing 6 changed files with 13 additions and 0 deletions.
diff --git a/engines/dlr/dlr-native/CMakeLists.txt b/engines/dlr/dlr-native/CMakeLists.txt
@@ -4,6 +4,8 @@ project(djl_dlr)
 set(CMAKE_CXX_STANDARD 14)
 set(CMAKE_CXX_STANDARD_REQUIRED ON)
 
+set(JAVA_AWT_LIBRARY NotNeeded)
+set(JAVA_AWT_INCLUDE_PATH NotNeeded)
 find_package(JNI REQUIRED)
 
 add_subdirectory(neo-ai-dlr)

diff --git a/engines/paddlepaddle/paddlepaddle-native/CMakeLists.txt b/engines/paddlepaddle/paddlepaddle-native/CMakeLists.txt
@@ -4,6 +4,8 @@ project(djl_paddle)
 set(CMAKE_CXX_STANDARD 11)
 set(CMAKE_CXX_STANDARD_REQUIRED ON)
 
+set(JAVA_AWT_LIBRARY NotNeeded)
+set(JAVA_AWT_INCLUDE_PATH NotNeeded)
 find_package(JNI REQUIRED)
 find_library(PADDLE_LIBRARY
         NAMES paddle_inference

diff --git a/engines/pytorch/pytorch-native/CMakeLists.txt b/engines/pytorch/pytorch-native/CMakeLists.txt
@@ -11,6 +11,9 @@ endif()
 if(NOT BUILD_ANDROID)
     message("Building djl_torch with the host...")
     find_package(Torch REQUIRED)
+
+    set(JAVA_AWT_LIBRARY NotNeeded)
+    set(JAVA_AWT_INCLUDE_PATH NotNeeded)
     find_package(JNI REQUIRED)
 
     set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${TORCH_CXX_FLAGS}")

diff --git a/engines/tensorrt/CMakeLists.txt b/engines/tensorrt/CMakeLists.txt
@@ -4,6 +4,8 @@ project(djl_trt)
 set(CMAKE_CXX_STANDARD 14)
 set(CMAKE_CXX_STANDARD_REQUIRED ON)
 
+set(JAVA_AWT_LIBRARY NotNeeded)
+set(JAVA_AWT_INCLUDE_PATH NotNeeded)
 find_package(JNI REQUIRED)
 #find_library(TRT_ONNX_PARSER
 #        NAMES nvonnxparser

diff --git a/extensions/fasttext/CMakeLists.txt b/extensions/fasttext/CMakeLists.txt
@@ -4,6 +4,8 @@ project(jni_fasttext)
 set(CMAKE_CXX_STANDARD 11)
 set(CMAKE_CXX_STANDARD_REQUIRED ON)
 
+set(JAVA_AWT_LIBRARY NotNeeded)
+set(JAVA_AWT_INCLUDE_PATH NotNeeded)
 find_package(JNI REQUIRED)
 find_path(UTILS_INCLUDE_DIR NAMES djl/utils.h PATHS ${PROJECT_SOURCE_DIR}/../../api/src/main/native REQUIRED)
 add_subdirectory(fastText)

diff --git a/extensions/sentencepiece/CMakeLists.txt b/extensions/sentencepiece/CMakeLists.txt
@@ -17,6 +17,8 @@ if (MSVC)
   add_definitions("/wd4267 /wd4244 /wd4305 /Zc:strictStrings /utf-8")
 endif()
 
+set(JAVA_AWT_LIBRARY NotNeeded)
+set(JAVA_AWT_INCLUDE_PATH NotNeeded)
 find_package(JNI REQUIRED)
 find_path(UTILS_INCLUDE_DIR NAMES djl/utils.h PATHS ${PROJECT_SOURCE_DIR}/../../api/src/main/native REQUIRED)