Use IDX2C macro properly

2025-11-06 17:54:27 +00:00 · 2024-02-19 22:26:54 +01:00
parent 4b6fff9bfd
commit 02fc9e4e8b
3 changed files with 27 additions and 15 deletions
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -13,6 +13,7 @@ set(LIBRARY_SOURCES
    src/layers/dense.cpp
 )

+set(CMAKE_CUDA_ARCHITECTURES 75)
 set(CMAKE_EXPORT_COMPILE_COMMANDS ON)
 set(CUDA_NVCC_FLAGS ${CUDA_NVCC_FLAGS} -arch=sm_75)