add CUDABlas related API tests

Author: youge325

CMakeLists.txt

@@ -76,6 +76,21 @@ list(REMOVE_ITEM TEST_SRC_FILES ${UNMATCH_FILES})
 file(GLOB_RECURSE TEST_BASE_FILES ${PROJECT_SOURCE_DIR}/src/*.cpp)
 set(PADDLE_TARGET_FOLDER ${CMAKE_BINARY_DIR}/paddle)
 
+# ---------------------------------------------------------------------------
+# CUDA Toolkit (needed for CUDA-specific test headers in the Torch build)
+# ---------------------------------------------------------------------------
+find_package(CUDAToolkit QUIET)
+if(CUDAToolkit_FOUND)
+  message(STATUS "Found CUDA Toolkit: ${CUDAToolkit_INCLUDE_DIRS}")
+  set(CUDA_INCLUDE_DIRS "${CUDAToolkit_INCLUDE_DIRS}")
+elseif(EXISTS "/usr/local/cuda/include")
+  set(CUDA_INCLUDE_DIRS "/usr/local/cuda/include")
+  message(STATUS "Using default CUDA include dir: ${CUDA_INCLUDE_DIRS}")
+else()
+  message(WARNING "CUDA headers not found; CUDA tests may not compile.")
+  set(CUDA_INCLUDE_DIRS "")
+endif()
+
 # ---------------------------------------------------------------------------
 # Build Torch test case
 # ---------------------------------------------------------------------------
@@ -87,8 +102,9 @@ set(TORCH_LIBRARIES "")
 file(GLOB_RECURSE TORCH_LIBRARIES "${TORCH_DIR}/lib/*.so"
      "${TORCH_DIR}/lib/*.a")
 
-set(TORCH_INCLUDE_DIR "${TORCH_DIR}/include"
-                      "${TORCH_DIR}/include/torch/csrc/api/include/")
+set(TORCH_INCLUDE_DIR
+    "${TORCH_DIR}/include" "${TORCH_DIR}/include/torch/csrc/api/include/"
+    "${CUDA_INCLUDE_DIRS}")
 
 set(TORCH_TARGET_FOLDER ${CMAKE_BINARY_DIR}/torch)
 set(BIN_PREFIX "torch_")
@@ -119,7 +135,8 @@ set(PADDLE_INCLUDE_DIR
     "${PADDLE_DIR}/include/third_party"
     "${PADDLE_DIR}/include/paddle/phi/api/include/compat/"
     "${PADDLE_DIR}/include/paddle/phi/api/include/compat/torch/csrc/api/include/"
-)
+    "${CUDA_INCLUDE_DIRS}"
+    "${CUDA_INCLUDE_DIRS}/cccl")
 
 set(PADDLE_LIBRARIES
     "${PADDLE_DIR}/base/libpaddle.so" "${PADDLE_DIR}/libs/libcommon.so"

cmake/build.cmake

@@ -23,6 +23,20 @@ function(
     message(STATUS "include dir: ${INCLUDE_DIR}")
     target_compile_definitions(${_test_name}
                                PRIVATE USE_PADDLE_API=${USE_PADDLE_API})
+    if(${USE_PADDLE_API})
+      # Paddle's CUDA compat headers (CUDAContextLight.h, CUDAFunctions.h)
+      # require PADDLE_WITH_CUDA to be defined so that GPU type aliases
+      # (gpuStream_t, cudaDeviceProp) are resolved via cuda_runtime.h.
+      target_compile_definitions(${_test_name} PRIVATE PADDLE_WITH_CUDA)
+      # Link libcudart for CUDA runtime symbols used by the Paddle CUDA compat
+      # layer.
+      if(TARGET CUDA::cudart)
+        target_link_libraries(${_test_name} CUDA::cudart)
+      elseif(EXISTS "/usr/local/cuda/lib64/libcudart.so")
+        target_link_libraries(${_test_name}
+                              "/usr/local/cuda/lib64/libcudart.so")
+      endif()
+    endif()
     message(STATUS "USE_PADDLE_API: ${USE_PADDLE_API}")
     add_test(NAME ${_test_name} COMMAND ${_test_name})
     set_tests_properties(${_test_name} PROPERTIES TIMEOUT 5)