bugfix: resolve gtest compilation failures.

Author: yingxudeng

cmake/cc_test.cmake

@@ -69,6 +69,14 @@ function(cc_test)
     PRIVATE ${CC_TEST_LINKOPTS}
   )
 
+  if(USE_NPU)
+    set(COMMON_LIBS Python::Python torch_npu torch_python)
+  endif()
+
+  if(USE_NPU AND DEFINED COMMON_LIBS)
+    target_link_libraries(${CC_TEST_NAME} PRIVATE ${COMMON_LIBS})
+  endif()
+
   add_dependencies(all_tests ${CC_TEST_NAME})
 
   gtest_add_tests(

setup.py

@@ -17,7 +17,7 @@
 from setuptools.command.bdist_wheel import bdist_wheel
 from setuptools.command.build_ext import build_ext
 
-BUILD_TEST_FILE = False
+BUILD_TEST_FILE = True
 BUILD_EXPORT = True
 
 # get cpu architecture

xllm/core/layers/common/CMakeLists.txt

@@ -43,6 +43,7 @@ cc_library(
     torch
 )
 
+if(NOT USE_NPU)
 # Add test for DenseMLP
 cc_test(
   NAME
@@ -76,3 +77,4 @@ cc_test(
     torch
     GTest::gtest_main
 )
+endif()