feat: update Docker image to support ATB, ACLNN fused ops and ProcessGroupHCCL. (#443)

Author: yingxudeng

cmake/cc_test.cmake

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

xllm/core/kernels/npu/impl/CMakeLists.txt

@@ -27,83 +27,3 @@ cc_library(
     torch
     torch_npu
 )
-
-cc_test(
-  NAME
-    npu_rms_norm_test
-  SRCS
-    npu_rms_norm_test.cpp
-  DEPS
-    :npu_kernels_impl
-    GTest::gtest
-    GTest::gtest_main
-    xllm_kernels
-    c_sec
-    atb
-    opapi
-    spdlog::spdlog
-)
-
-cc_test(
-  NAME
-    npu_linear_test
-  SRCS
-    npu_linear_test.cpp
-  DEPS
-    :npu_kernels_impl
-    GTest::gtest
-    GTest::gtest_main
-    xllm_kernels
-    c_sec
-    atb
-    opapi
-    spdlog::spdlog
-)
-
-cc_test(
-  NAME
-    npu_split_test
-  SRCS
-    npu_split_test.cpp
-  DEPS
-    :npu_kernels_impl
-    GTest::gtest
-    GTest::gtest_main
-    xllm_kernels
-    c_sec
-    atb
-    opapi
-    spdlog::spdlog
-)
-
-cc_test(
-  NAME
-    npu_rope_impl_test
-  SRCS
-    npu_rope_impl_test.cpp
-  DEPS
-    :npu_kernels_impl
-    GTest::gtest
-    GTest::gtest_main
-    xllm_kernels
-    c_sec
-    atb
-    opapi
-    spdlog::spdlog
-)
-
-cc_test(
-  NAME
-    npu_sample_model_test
-  SRCS
-    npu_sample_model_test.cpp
-  DEPS
-    :npu_kernels_impl
-    GTest::gtest
-    GTest::gtest_main
-    xllm_kernels
-    c_sec
-    atb
-    opapi
-    spdlog::spdlog
-)