Skip to content

Commit 306ee3f

Browse files
committed
Fix: bf16 requires Ampere
1 parent 97991fd commit 306ee3f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

less_slow.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@
6868
#if (__CUDA_ARCH__ >= 700)
6969
#include <cuda_fp16.h> // `half` type
7070
#endif
71-
#if (__CUDA_ARCH__ >= 750)
71+
#if (__CUDA_ARCH__ >= 800)
7272
#include <cuda_bf16.h> // `__nv_bfloat16` type
7373
#endif
7474
#if (__CUDA_ARCH__ >= 900)

0 commit comments

Comments
 (0)