CUDA_NVCC_FLAGS(
    --expt-relaxed-constexpr
    -gencode arch=compute_35,code=sm_35
    -gencode arch=compute_50,code=compute_50
    -gencode arch=compute_52,code=sm_52
    -gencode arch=compute_60,code=compute_60
    -gencode arch=compute_61,code=compute_61
    -gencode arch=compute_61,code=sm_61
    -gencode arch=compute_70,code=sm_70
    -gencode arch=compute_70,code=compute_70
    --ptxas-options=-v
    -lineinfo
    --use_fast_math
)

IF (CUDA_VERSION VERSION_GE 11)
    CUDA_NVCC_FLAGS(
        -gencode=arch=compute_80,code=sm_80
    )
ENDIF()
