rapids_cuda_set_architectures

添加于 v21.06.00 版本。

基于请求的模式设置 CMAKE_CUDA_ARCHITECTURES

rapids_cuda_set_architectures( (NATIVE|RAPIDS) )

建立将编译哪些 CUDA 架构，覆盖任何现有的 CMAKE_CUDA_ARCHITECTURES 值。

此函数应很少使用，因为 rapids_cuda_init_architectures() 允许在配置项目时使用 CMAKE_CUDA_ARCHITECTURES 的预期工作流程。如果由于某种原因您的项目无法使用 rapids_cuda_init_architectures()，则可以直接使用 rapids_cuda_set_architectures()。

注意

此函数由 rapids_cuda_init_architectures() 自动调用

NATIVE 或 ""

当作为 CMAKE_CUDA_ARCHITECTURES 或 ENV{CUDAARCHS} 的值传递时，将编译当前机器上存在的所有 GPU 架构。

RAPIDS, ALL，或在 CMAKE_CUDA_ARCHITECTURES 和 ENV{CUDAARCHS} 中没有值时

当作为 CMAKE_CUDA_ARCHITECTURES 或 ENV{CUDAARCHS} 的值传递时，将编译所有支持的 RAPIDS GPU 架构。

结果变量

CMAKE_CUDA_ARCHITECTURES

将作为一个局部变量存在，并被设置为应编译的架构列表。将覆盖任何现有值。

添加于 v24.08.00 版本：如果它还不是缓存变量，将被添加为一个缓存变量。