Fix compilation bug with CUDA 12.1 (#949) · NVIDIA/TransformerEngine@744624d (original) (raw)
Navigation Menu
- GitHub Copilot Write better code with AI
- GitHub Models New Manage and compare prompts
- GitHub Advanced Security Find and fix vulnerabilities
- Actions Automate any workflow
- Codespaces Instant dev environments
- Issues Plan and track work
- Code Review Manage code changes
- Discussions Collaborate outside of code
- Code Search Find more, search less
- Explore
- Pricing
Provide feedback
Saved searches
Use saved searches to filter your results more quickly
Appearance settings
Commit 744624d
and
authored
Fix compilation bug with CUDA 12.1 (#949)
* fix compilation Signed-off-by: Edenzzzz wtan45@wisc.edu * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ciSigned-off-by: Edenzzzz wtan45@wisc.edu * Fix CUDA 12.1 compilation bug Signed-off-by: Wenxuan(Eden) Tan wtan45@wisc.edu Signed-off-by: Edenzzzz wtan45@wisc.edu --------- Signed-off-by: Edenzzzz wtan45@wisc.edu Signed-off-by: Wenxuan(Eden) Tan wtan45@wisc.edu Co-authored-by: Edenzzzz wtan45@wisc.edu Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
File tree
1 file changed
lines changed
- transformer_engine/pytorch/csrc/userbuffers
1 file changed
lines changed
Lines changed: 1 addition & 1 deletion
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -5,6 +5,7 @@ | ||
5 | 5 | ************************************************************************/ |
6 | 6 | |
7 | 7 | #include <cuda.h> |
8 | +#include <cuda_fp8.h> | |
8 | 9 | #include <cuda_runtime.h> |
9 | 10 | |
10 | 11 | #if __CUDA_ARCH__ >= 800 |
@@ -15,7 +16,6 @@ | ||
15 | 16 | #endif |
16 | 17 | |
17 | 18 | #include <assert.h> |
18 | -#include <cuda_fp8.h> | |
19 | 19 | #include <stdio.h> |
20 | 20 | #include <unistd.h> |
21 | 21 |