File tree 1 file changed +6
-10
lines changed
libcudacxx/include/cuda/__ptx/instructions
1 file changed +6
-10
lines changed Original file line number Diff line number Diff line change 28
28
29
29
#include < nv/target> // __CUDA_MINIMUM_ARCH__ and friends
30
30
31
- #if defined(_LIBCUDACXX_HAS_NVFP16)
32
- # include < cuda_fp16.h>
33
- #endif // _LIBCUDACXX_HAS_NVFP16
34
-
35
- #if defined(_LIBCUDACXX_HAS_NVBF16)
36
- _CCCL_DIAG_PUSH
37
- _CCCL_DIAG_SUPPRESS_CLANG (" -Wunused-function" )
38
- # include < cuda_bf16.h>
39
- _CCCL_DIAG_POP
40
- #endif // _LIBCUDACXX_HAS_NVBF16
31
+ // Forward-declare __half and __nv_bfloat16. The cuda_fp16.h and cuda_bf16.h are
32
+ // expensive to include. The APIs use only pointers, so we do not have to define
33
+ // the types. If the user wants to use these types, it is their responsibility
34
+ // to include the headers.
35
+ struct __half ;
36
+ struct __nv_bfloat16 ;
41
37
42
38
_LIBCUDACXX_BEGIN_NAMESPACE_CUDA_PTX
43
39
You can’t perform that action at this time.
0 commit comments