Fix compilation bug with CUDA 12.1 (#949) · NVIDIA/TransformerEngine@744624d (original) (raw)

Skip to content

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Sign up

Appearance settings

Commit 744624d

Edenzzzzpre-commit-ci[bot]

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

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