search

yrmo / cudagrad

A tensor-valued autograd engine for Python
MIT License
0 stars 0 forks source link

Unify CMakeLists #11

Open yrmo opened 6 months ago

yrmo commented 2 months ago 
$ grep -Fx -f CMakeLists.txt  CMakeListsCUDA.txt 
set(CMAKE_CXX_STANDARD 17)
set(CMAKE_CUDA_ARCHITECTURES "86")
enable_language(CUDA)
include_directories(${CMAKE_CURRENT_SOURCE_DIR}/cccl/thrust)
include_directories(${CMAKE_CURRENT_SOURCE_DIR}/cccl/libcudacxx/include)
include_directories(${CMAKE_CURRENT_SOURCE_DIR}/cccl/cub)
yrmo commented 2 months ago

Make the test CMakeList.txt goes in /cudagrad/tests/