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

Adding support for building BERT plugins with GPU_ARCHS specified #255

Merged
merged 1 commit into from
Dec 5, 2019
Merged

Adding support for building BERT plugins with GPU_ARCHS specified #255

merged 1 commit into from
Dec 5, 2019

Conversation

kevinch-nv
Copy link
Collaborator

  • Support GPU_ARCHS when compiling plugin cuda files
  • Updated README to reflect expected usage of GPU_ARCHS
  • Moved Bert-plugin SM check logic to top level cmakefile
  • Change Plugin top level cmakefile and individual plugin cmakefiles to glob .cpp and cu sources separately, and provide gencode flags when compiling cu files.
  • Updated bert plugin known issues to document architecture requirement

@kevinch-nv kevinch-nv merged commit 98ab844 into NVIDIA:master Dec 5, 2019
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@rajeevsrao rajeevsrao rajeevsrao approved these changes

@pranavm-nvidia pranavm-nvidia pranavm-nvidia 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.
3 participants
@kevinch-nv @rajeevsrao @pranavm-nvidia