Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Enhance cmake to support specifying CUDA_ARCH_NAME to Ampere. #31923

Merged
merged 3 commits into from
Mar 30, 2021
Merged

Enhance cmake to support specifying CUDA_ARCH_NAME to Ampere. #31923

merged 3 commits into from
Mar 30, 2021

Conversation

Xreki
Copy link
Contributor

@Xreki Xreki commented Mar 29, 2021 •
edited
Loading

PR types

Others

PR changes

Others

Describe

  • 支持设置cmake选项：CUDA_ARCH_NAME=Ampere，便于交叉编译适用于Ampere架构的二进制代码。设置了CUDA_ARCH_NAME=Ampere之后，cmake输出如下：
-- CUDA detected: 11.0.221
-- NVCC_FLAGS_EXTRA:  -gencode arch=compute_80,code=sm_80
-- Current cuDNN header is /usr/include/cudnn_version.h Current cuDNN version is v8.1.1.

@Xreki Xreki changed the title Enhance cmake to support psecifying CUDA_ARCH_NAME to Ampere. Enhance cmake to support specifying CUDA_ARCH_NAME to Ampere. Mar 30, 2021
@Xreki Xreki requested a review from luotao1 March 30, 2021 03:44
luotao1
luotao1 approved these changes Mar 30, 2021
View reviewed changes
Copy link
Contributor

@luotao1 luotao1 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@Xreki Xreki merged commit e50bc2c into PaddlePaddle:develop Mar 30, 2021
@Xreki Xreki deleted the cmake/cuda11 branch March 30, 2021 06:18
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@luotao1 luotao1 luotao1 approved these changes

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@Xreki @luotao1