PetaVision / OpenPV

PetaVision is a C++ library for designing and deploying large-scale neurally-inspired computational models.
http://petavision.github.io
Eclipse Public License 1.0
40 stars 13 forks source link

Adds PV_CUDA_ARCHITECTURE cmake variable #312

Closed peteschultz closed 3 years ago

peteschultz commented 3 years ago

This pull request removes the hardcoded gpu-architecture setting from the PVConfigProject.cmake file and provides a CMake variable to control the setting.

PV_CUDA_ARCHITECTURE is a string that can be set to an NVIDIA compute capability, such as 7.0, or an NVIDIA microarchitecture name, such as Volta. If set to Auto, it will detect the compute capability of the current GPU and use that. If left blank, it will use the default setting for the installed version of the nvcc compiler.

The CMake variable initializes to Auto.