File tree 3 files changed +3
-3
lines changed
3 files changed +3
-3
lines changed Original file line number Diff line number Diff line change 11
11
#include < caffe2/core/hip/common_gpu.h>
12
12
#endif
13
13
14
- #if defined(__CUDA_ARCH__) || defined(__HIP_DEVICE_COMPILE__)
14
+ #if defined(__CUDA_ARCH__) || defined(__HIP_DEVICE_COMPILE__) || defined(__HIP__)
15
15
#define CONVERSIONS_DECL __host__ __device__ inline
16
16
#else
17
17
#define CONVERSIONS_DECL inline
Original file line number Diff line number Diff line change 7
7
#include < cstdio>
8
8
#include < cstdlib>
9
9
10
- #if defined(__CUDA_ARCH__) || defined(__HIP_ARCH__)
10
+ #if defined(__CUDA_ARCH__) || defined(__HIP_ARCH__) || defined(__HIP__)
11
11
#define FIXED_DIVISOR_DECL inline __host__ __device__
12
12
#else
13
13
#define FIXED_DIVISOR_DECL inline
Original file line number Diff line number Diff line change 3
3
4
4
#include " caffe2/core/common.h"
5
5
6
- #if defined(__CUDA_ARCH__) || defined(__HIP_DEVICE_COMPILE__)
6
+ #if defined(__CUDA_ARCH__) || defined(__HIP_DEVICE_COMPILE__) || defined(__HIP__)
7
7
#define MATH_UTILS_DECL inline __host__ __device__
8
8
#else
9
9
#define MATH_UTILS_DECL inline
You can’t perform that action at this time.
0 commit comments