Added CMake code to generate the correct flags for CUDA compute capability