Add activations test

This commit is contained in:
2024-02-27 20:19:17 +01:00
parent 48ba09b28d
commit b1eb8b5806
6 changed files with 54 additions and 21 deletions

View File

@@ -10,7 +10,7 @@ include_directories(${CUDAToolkit_INCLUDE_DIRS})
# Add project source files for the library
set(LIBRARY_SOURCES
src/utils/cuda_helper.cu
src/functions/activations.cu
src/kernels/activations.cu
src/layers/dense.cu
)
@@ -28,7 +28,7 @@ target_link_libraries(${PROJECT_NAME} CUDA::cublas CUDA::cudart)
target_include_directories(${PROJECT_NAME} PUBLIC
${CMAKE_CURRENT_SOURCE_DIR}/include
${CMAKE_CURRENT_SOURCE_DIR}/include/utils
${CMAKE_CURRENT_SOURCE_DIR}/include/functions
${CMAKE_CURRENT_SOURCE_DIR}/include/kernels
${CMAKE_CURRENT_SOURCE_DIR}/include/layers
${CMAKE_CURRENT_SOURCE_DIR}/src
)