Rename files to .cu and fix IDX2C usage

This commit is contained in:
2024-02-21 20:03:04 +01:00
parent 15c0cd30f0
commit 035f3b053b
9 changed files with 72 additions and 36 deletions

View File

@@ -1,6 +1,6 @@
set(LAYER_SOURCES layers/dense.cpp)
set(LAYER_SOURCES layers/dense.cu)
add_library(CUDANet
utils/cuda_helper.cpp
utils/cuda_helper.cu
${LAYER_SOURCES}
)