Skip to content

Commit e5d9a97

Browse files
authored
Fix cuda compile error with bf16
Differential Revision: D73562284 Pull Request resolved: #2122
1 parent 72e3c11 commit e5d9a97

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

torchao/csrc/cuda/fp6_llm/utils_parallel_dequant.cuh

-2
Original file line numberDiff line numberDiff line change
@@ -25,9 +25,7 @@
2525

2626
#include <cuda.h>
2727
#include <cuda_fp16.h>
28-
#if !defined(__CUDA_ARCH__) || __CUDA_ARCH__ >= 800
2928
#include <cuda_bf16.h>
30-
#endif
3129
#include <cuda_runtime.h>
3230

3331
/*

0 commit comments

Comments
 (0)