diff options
Diffstat (limited to 'caffe2/utils/fixed_divisor.h')
| -rw-r--r-- | caffe2/utils/fixed_divisor.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/caffe2/utils/fixed_divisor.h b/caffe2/utils/fixed_divisor.h index 3c7aaf016a..64e84fb7da 100644 --- a/caffe2/utils/fixed_divisor.h +++ b/caffe2/utils/fixed_divisor.h @@ -5,7 +5,7 @@ #include <cstdio> #include <cstdlib> -#if defined(__CUDA_ARCH__) || defined(__HIP_ARCH__) +#if defined(__CUDA_ARCH__) || defined(__HIP_ARCH__) || defined(__HIP__) #define FIXED_DIVISOR_DECL inline __host__ __device__ #else #define FIXED_DIVISOR_DECL inline |