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.

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

add option to specify CUDA arch #2940

Closed
cydrain opened this issue Jul 21, 2020 · 0 comments · Fixed by #2950
Closed

add option to specify CUDA arch #2940

cydrain opened this issue Jul 21, 2020 · 0 comments · Fixed by #2950
Assignees
Labels
kind/enhancement Issues or changes related to enhancement

Comments

@cydrain
Copy link
Contributor

cydrain commented Jul 21, 2020

Please state you issue using the following template and, most importantly, in English.

Is your feature request related to a problem? Please describe.
GPU FAISS building is timing consuming, if we only build GPU FAISS with specified CUDA arch,
we can save a lot of building time.

Describe the solution you'd like
Add an option in build.sh to specify the CUDA arch we can support.
Proposal 1) add option "-cuda-arch=-gencode=arch=compute_61,code=sm_61 -gencode=arch=compute_75,code=sm_75"
Proposal 2) add option "-ci_only=true"
without above options, 4 CUDA arch will be specified.

@cydrain cydrain added the kind/enhancement Issues or changes related to enhancement label Jul 21, 2020
@cydrain cydrain added this to To do in Improvement Request via automation Jul 21, 2020
@wxyucs wxyucs assigned XuanYang-cn and unassigned wxyucs Jul 21, 2020
Improvement Request automation moved this from To do to Done Jul 23, 2020
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
kind/enhancement Issues or changes related to enhancement
Projects
No open projects
Development

Successfully merging a pull request may close this issue.

3 participants