Add GPUBackend

2025-08-01 14:20:32 +00:00
parent 404fbd3c02
commit b7c13be6c0
17 changed files with 233 additions and 11 deletions
--- a/src/CMakeLists.txt
+++ b/src/CMakeLists.txt
@@ -9,3 +9,12 @@ if(USE_MKL)
  target_sources(trigdx PRIVATE mkl.cpp)
  target_link_libraries(trigdx PRIVATE MKL::MKL)
 endif()
+
+if(USE_GPU)
+  enable_language(CUDA)
+  find_package(CUDAToolkit REQUIRED)
+  add_library(gpu SHARED gpu/gpu.cu)
+  target_sources(trigdx PRIVATE gpu.cpp)
+  target_link_libraries(trigdx PRIVATE CUDA::cudart)
+  target_link_libraries(trigdx PRIVATE gpu)
+endif()