fix python

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Nov 13, 2024 · c455b00 · c455b00
1 parent 3c5199d
commit c455b00
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/source/CMakeLists.txt b/source/CMakeLists.txt
@@ -251,7 +251,10 @@ if(ENABLE_PYTORCH AND NOT DEEPMD_C_ROOT)
   list(APPEND BACKEND_LIBRARY_PATH ${PyTorch_LIBRARY_PATH})
   list(APPEND BACKEND_INCLUDE_DIRS ${TORCH_INCLUDE_DIRS})
 endif()
-if(ENABLE_JAX AND NOT DEEPMD_C_ROOT)
+if(ENABLE_JAX
+   AND BUILD_CPP_IF
+   AND NOT DEEPMD_C_ROOT)
+  # no way to find it using Python
   find_package(TensorFlowC REQUIRED MODULE)
   if(DEFINED TENSORFLOWC_LIBRARY)
     list(APPEND BACKEND_LIBRARY_PATH ${TENSORFLOWC_LIBRARY})