Fix compilation bug on CUDA 12.9 (#72978)

risemeup1 · web-flow · commit a4ae33fce5c2 · 2025-05-29T10:54:17.000+08:00
* fix compile bug

* fix
diff --git a/cmake/cinn/core.cmake b/cmake/cinn/core.cmake
@@ -73,8 +73,12 @@ function(cinn_cc_test TARGET_NAME)
     add_executable(${TARGET_NAME} ${cinn_cc_test_SRCS})
     get_property(os_dependency_modules GLOBAL PROPERTY OS_DEPENDENCY_MODULES)
     target_link_libraries(${TARGET_NAME} ${os_dependency_modules}
-                          cinn_gtest_main gtest glog ${cinn_cc_test_DEPS})
-    add_dependencies(${TARGET_NAME} cinn_gtest_main gtest glog
+                          paddle_gtest_main gtest glog ${cinn_cc_test_DEPS})
+    if(WITH_SHARED_PHI)
+      target_link_libraries(${TARGET_NAME} -Wl,--as-needed phi_core phi_gpu
+                            -Wl,--no-as-needed)
+    endif()
+    add_dependencies(${TARGET_NAME} paddle_gtest_main gtest glog
                      ${cinn_cc_test_DEPS})
 
     add_test(
@@ -159,13 +163,18 @@ function(cinn_nv_test TARGET_NAME)
     target_link_libraries(
       ${TARGET_NAME}
       ${cinn_nv_test_DEPS}
-      cinn_gtest_main
+      paddle_gtest_main
       gtest
       ${os_dependency_modules}
       ${CUDNN_LIBRARY}
       ${CUBLAS_LIBRARIES}
       ${CUDA_LIBRARIES})
-    add_dependencies(${TARGET_NAME} ${cinn_nv_test_DEPS} cinn_gtest_main gtest)
+    if(WITH_SHARED_PHI)
+      target_link_libraries(${TARGET_NAME} -Wl,--as-needed phi_core phi_gpu
+                            -Wl,--no-as-needed)
+    endif()
+    add_dependencies(${TARGET_NAME} ${cinn_nv_test_DEPS} paddle_gtest_main
+                     gtest)
     common_link(${TARGET_NAME})
     add_test(
       NAME ${TARGET_NAME}
diff --git a/cmake/generic.cmake b/cmake/generic.cmake
@@ -599,8 +599,9 @@ function(paddle_test_build TARGET_NAME)
       target_link_libraries(${TARGET_NAME} ${PYTHON_LIBRARIES})
     endif()
     if(WITH_CINN)
-      target_link_libraries(${TARGET_NAME} $<TARGET_LINKER_FILE:cinnapi>
-                            cinn_transforms)
+      target_link_libraries(${TARGET_NAME} -Wl,--as-needed cinnapi
+                            -Wl,--no-as-needed)
+      target_link_libraries(${TARGET_NAME} cinn_transforms)
       add_dependencies(${TARGET_NAME} cinnapi)
     endif()
     if(WITH_XPU)
