File tree Expand file tree Collapse file tree 2 files changed +16
-6
lines changed Expand file tree Collapse file tree 2 files changed +16
-6
lines changed Original file line number Diff line number Diff line change 10
10
#define KERNEL_FLOAT_IS_CUDA (1)
11
11
#define KERNEL_FLOAT_DEVICE __forceinline__ __device__
12
12
13
- #ifdef __CUDA_ARCH__
13
+ // NVRTC cannot deal with __host__ annotations
14
+ #ifdef __CUDACC_RTC__
14
15
#define KERNEL_FLOAT_INLINE __forceinline__ __device__
16
+ #else
17
+ #define KERNEL_FLOAT_INLINE __forceinline__ __host__ __device__
18
+ #endif
19
+
20
+ #ifdef __CUDA_ARCH__
15
21
#define KERNEL_FLOAT_IS_DEVICE (1)
16
22
#else // __CUDA_ARCH__
17
- #define KERNEL_FLOAT_INLINE __forceinline__ __host__
18
23
#define KERNEL_FLOAT_IS_HOST (1)
19
24
#endif // __CUDA_ARCH__
20
25
#elif defined(__HIPCC__ )
Original file line number Diff line number Diff line change 16
16
17
17
// ================================================================================
18
18
// this file has been auto-generated, do not modify its contents!
19
- // date: 2025-09-15 12:39:02.709972
20
- // git hash: 81efb0fbfcc587343ba798590ba1541babe378d6
19
+ // date: 2025-09-15 12:44:05.768243
20
+ // git hash: 9b41485a27b669ea6f4aae118b4d251947608bf6
21
21
// ================================================================================
22
22
23
23
#ifndef KERNEL_FLOAT_MACROS_H
32
32
#define KERNEL_FLOAT_IS_CUDA (1 )
33
33
#define KERNEL_FLOAT_DEVICE __forceinline__ __device__
34
34
35
- #ifdef __CUDA_ARCH__
35
+ // NVRTC cannot deal with __host__ annotations
36
+ #ifdef __CUDACC_RTC__
36
37
#define KERNEL_FLOAT_INLINE __forceinline__ __device__
38
+ #else
39
+ #define KERNEL_FLOAT_INLINE __forceinline__ __host__ __device__
40
+ #endif
41
+
42
+ #ifdef __CUDA_ARCH__
37
43
#define KERNEL_FLOAT_IS_DEVICE (1 )
38
44
#else // __CUDA_ARCH__
39
- #define KERNEL_FLOAT_INLINE __forceinline__ __host__
40
45
#define KERNEL_FLOAT_IS_HOST (1 )
41
46
#endif // __CUDA_ARCH__
42
47
#elif defined(__HIPCC__)
You can’t perform that action at this time.
0 commit comments