Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1 | /*===---- __clang_cuda_device_functions.h - CUDA runtime support -----------=== |
| 2 | * |
Logan Chien | df4f766 | 2019-09-04 16:45:23 -0700 | [diff] [blame] | 3 | * Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
| 4 | * See https://llvm.org/LICENSE.txt for license information. |
| 5 | * SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 6 | * |
| 7 | *===-----------------------------------------------------------------------=== |
| 8 | */ |
| 9 | |
| 10 | #ifndef __CLANG_CUDA_DEVICE_FUNCTIONS_H__ |
| 11 | #define __CLANG_CUDA_DEVICE_FUNCTIONS_H__ |
| 12 | |
Sasha Smundak | 0fc590b | 2020-10-07 08:11:59 -0700 | [diff] [blame] | 13 | #ifndef __OPENMP_NVPTX__ |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 14 | #if CUDA_VERSION < 9000 |
| 15 | #error This file is intended to be used with CUDA-9+ only. |
| 16 | #endif |
Logan Chien | df4f766 | 2019-09-04 16:45:23 -0700 | [diff] [blame] | 17 | #endif |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 18 | |
| 19 | // __DEVICE__ is a helper macro with common set of attributes for the wrappers |
| 20 | // we implement in this file. We need static in order to avoid emitting unused |
| 21 | // functions and __forceinline__ helps inlining these wrappers at -O1. |
| 22 | #pragma push_macro("__DEVICE__") |
Sasha Smundak | 0fc590b | 2020-10-07 08:11:59 -0700 | [diff] [blame] | 23 | #ifdef __OPENMP_NVPTX__ |
| 24 | #define __DEVICE__ static __attribute__((always_inline, nothrow)) |
Logan Chien | df4f766 | 2019-09-04 16:45:23 -0700 | [diff] [blame] | 25 | #else |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 26 | #define __DEVICE__ static __device__ __forceinline__ |
Logan Chien | df4f766 | 2019-09-04 16:45:23 -0700 | [diff] [blame] | 27 | #endif |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 28 | |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 29 | __DEVICE__ int __all(int __a) { return __nvvm_vote_all(__a); } |
| 30 | __DEVICE__ int __any(int __a) { return __nvvm_vote_any(__a); } |
| 31 | __DEVICE__ unsigned int __ballot(int __a) { return __nvvm_vote_ballot(__a); } |
| 32 | __DEVICE__ unsigned int __brev(unsigned int __a) { return __nv_brev(__a); } |
| 33 | __DEVICE__ unsigned long long __brevll(unsigned long long __a) { |
| 34 | return __nv_brevll(__a); |
| 35 | } |
Logan Chien | df4f766 | 2019-09-04 16:45:23 -0700 | [diff] [blame] | 36 | #if defined(__cplusplus) |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 37 | __DEVICE__ void __brkpt() { __asm__ __volatile__("brkpt;"); } |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 38 | __DEVICE__ void __brkpt(int __a) { __brkpt(); } |
Logan Chien | df4f766 | 2019-09-04 16:45:23 -0700 | [diff] [blame] | 39 | #else |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 40 | __DEVICE__ void __attribute__((overloadable)) __brkpt(void) { |
| 41 | __asm__ __volatile__("brkpt;"); |
| 42 | } |
Logan Chien | df4f766 | 2019-09-04 16:45:23 -0700 | [diff] [blame] | 43 | __DEVICE__ void __attribute__((overloadable)) __brkpt(int __a) { __brkpt(); } |
| 44 | #endif |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 45 | __DEVICE__ unsigned int __byte_perm(unsigned int __a, unsigned int __b, |
| 46 | unsigned int __c) { |
| 47 | return __nv_byte_perm(__a, __b, __c); |
| 48 | } |
| 49 | __DEVICE__ int __clz(int __a) { return __nv_clz(__a); } |
| 50 | __DEVICE__ int __clzll(long long __a) { return __nv_clzll(__a); } |
| 51 | __DEVICE__ float __cosf(float __a) { return __nv_fast_cosf(__a); } |
| 52 | __DEVICE__ double __dAtomicAdd(double *__p, double __v) { |
| 53 | return __nvvm_atom_add_gen_d(__p, __v); |
| 54 | } |
| 55 | __DEVICE__ double __dAtomicAdd_block(double *__p, double __v) { |
| 56 | return __nvvm_atom_cta_add_gen_d(__p, __v); |
| 57 | } |
| 58 | __DEVICE__ double __dAtomicAdd_system(double *__p, double __v) { |
| 59 | return __nvvm_atom_sys_add_gen_d(__p, __v); |
| 60 | } |
| 61 | __DEVICE__ double __dadd_rd(double __a, double __b) { |
| 62 | return __nv_dadd_rd(__a, __b); |
| 63 | } |
| 64 | __DEVICE__ double __dadd_rn(double __a, double __b) { |
| 65 | return __nv_dadd_rn(__a, __b); |
| 66 | } |
| 67 | __DEVICE__ double __dadd_ru(double __a, double __b) { |
| 68 | return __nv_dadd_ru(__a, __b); |
| 69 | } |
| 70 | __DEVICE__ double __dadd_rz(double __a, double __b) { |
| 71 | return __nv_dadd_rz(__a, __b); |
| 72 | } |
| 73 | __DEVICE__ double __ddiv_rd(double __a, double __b) { |
| 74 | return __nv_ddiv_rd(__a, __b); |
| 75 | } |
| 76 | __DEVICE__ double __ddiv_rn(double __a, double __b) { |
| 77 | return __nv_ddiv_rn(__a, __b); |
| 78 | } |
| 79 | __DEVICE__ double __ddiv_ru(double __a, double __b) { |
| 80 | return __nv_ddiv_ru(__a, __b); |
| 81 | } |
| 82 | __DEVICE__ double __ddiv_rz(double __a, double __b) { |
| 83 | return __nv_ddiv_rz(__a, __b); |
| 84 | } |
| 85 | __DEVICE__ double __dmul_rd(double __a, double __b) { |
| 86 | return __nv_dmul_rd(__a, __b); |
| 87 | } |
| 88 | __DEVICE__ double __dmul_rn(double __a, double __b) { |
| 89 | return __nv_dmul_rn(__a, __b); |
| 90 | } |
| 91 | __DEVICE__ double __dmul_ru(double __a, double __b) { |
| 92 | return __nv_dmul_ru(__a, __b); |
| 93 | } |
| 94 | __DEVICE__ double __dmul_rz(double __a, double __b) { |
| 95 | return __nv_dmul_rz(__a, __b); |
| 96 | } |
| 97 | __DEVICE__ float __double2float_rd(double __a) { |
| 98 | return __nv_double2float_rd(__a); |
| 99 | } |
| 100 | __DEVICE__ float __double2float_rn(double __a) { |
| 101 | return __nv_double2float_rn(__a); |
| 102 | } |
| 103 | __DEVICE__ float __double2float_ru(double __a) { |
| 104 | return __nv_double2float_ru(__a); |
| 105 | } |
| 106 | __DEVICE__ float __double2float_rz(double __a) { |
| 107 | return __nv_double2float_rz(__a); |
| 108 | } |
| 109 | __DEVICE__ int __double2hiint(double __a) { return __nv_double2hiint(__a); } |
| 110 | __DEVICE__ int __double2int_rd(double __a) { return __nv_double2int_rd(__a); } |
| 111 | __DEVICE__ int __double2int_rn(double __a) { return __nv_double2int_rn(__a); } |
| 112 | __DEVICE__ int __double2int_ru(double __a) { return __nv_double2int_ru(__a); } |
| 113 | __DEVICE__ int __double2int_rz(double __a) { return __nv_double2int_rz(__a); } |
| 114 | __DEVICE__ long long __double2ll_rd(double __a) { |
| 115 | return __nv_double2ll_rd(__a); |
| 116 | } |
| 117 | __DEVICE__ long long __double2ll_rn(double __a) { |
| 118 | return __nv_double2ll_rn(__a); |
| 119 | } |
| 120 | __DEVICE__ long long __double2ll_ru(double __a) { |
| 121 | return __nv_double2ll_ru(__a); |
| 122 | } |
| 123 | __DEVICE__ long long __double2ll_rz(double __a) { |
| 124 | return __nv_double2ll_rz(__a); |
| 125 | } |
| 126 | __DEVICE__ int __double2loint(double __a) { return __nv_double2loint(__a); } |
| 127 | __DEVICE__ unsigned int __double2uint_rd(double __a) { |
| 128 | return __nv_double2uint_rd(__a); |
| 129 | } |
| 130 | __DEVICE__ unsigned int __double2uint_rn(double __a) { |
| 131 | return __nv_double2uint_rn(__a); |
| 132 | } |
| 133 | __DEVICE__ unsigned int __double2uint_ru(double __a) { |
| 134 | return __nv_double2uint_ru(__a); |
| 135 | } |
| 136 | __DEVICE__ unsigned int __double2uint_rz(double __a) { |
| 137 | return __nv_double2uint_rz(__a); |
| 138 | } |
| 139 | __DEVICE__ unsigned long long __double2ull_rd(double __a) { |
| 140 | return __nv_double2ull_rd(__a); |
| 141 | } |
| 142 | __DEVICE__ unsigned long long __double2ull_rn(double __a) { |
| 143 | return __nv_double2ull_rn(__a); |
| 144 | } |
| 145 | __DEVICE__ unsigned long long __double2ull_ru(double __a) { |
| 146 | return __nv_double2ull_ru(__a); |
| 147 | } |
| 148 | __DEVICE__ unsigned long long __double2ull_rz(double __a) { |
| 149 | return __nv_double2ull_rz(__a); |
| 150 | } |
| 151 | __DEVICE__ long long __double_as_longlong(double __a) { |
| 152 | return __nv_double_as_longlong(__a); |
| 153 | } |
| 154 | __DEVICE__ double __drcp_rd(double __a) { return __nv_drcp_rd(__a); } |
| 155 | __DEVICE__ double __drcp_rn(double __a) { return __nv_drcp_rn(__a); } |
| 156 | __DEVICE__ double __drcp_ru(double __a) { return __nv_drcp_ru(__a); } |
| 157 | __DEVICE__ double __drcp_rz(double __a) { return __nv_drcp_rz(__a); } |
| 158 | __DEVICE__ double __dsqrt_rd(double __a) { return __nv_dsqrt_rd(__a); } |
| 159 | __DEVICE__ double __dsqrt_rn(double __a) { return __nv_dsqrt_rn(__a); } |
| 160 | __DEVICE__ double __dsqrt_ru(double __a) { return __nv_dsqrt_ru(__a); } |
| 161 | __DEVICE__ double __dsqrt_rz(double __a) { return __nv_dsqrt_rz(__a); } |
| 162 | __DEVICE__ double __dsub_rd(double __a, double __b) { |
| 163 | return __nv_dsub_rd(__a, __b); |
| 164 | } |
| 165 | __DEVICE__ double __dsub_rn(double __a, double __b) { |
| 166 | return __nv_dsub_rn(__a, __b); |
| 167 | } |
| 168 | __DEVICE__ double __dsub_ru(double __a, double __b) { |
| 169 | return __nv_dsub_ru(__a, __b); |
| 170 | } |
| 171 | __DEVICE__ double __dsub_rz(double __a, double __b) { |
| 172 | return __nv_dsub_rz(__a, __b); |
| 173 | } |
| 174 | __DEVICE__ float __exp10f(float __a) { return __nv_fast_exp10f(__a); } |
| 175 | __DEVICE__ float __expf(float __a) { return __nv_fast_expf(__a); } |
| 176 | __DEVICE__ float __fAtomicAdd(float *__p, float __v) { |
| 177 | return __nvvm_atom_add_gen_f(__p, __v); |
| 178 | } |
| 179 | __DEVICE__ float __fAtomicAdd_block(float *__p, float __v) { |
| 180 | return __nvvm_atom_cta_add_gen_f(__p, __v); |
| 181 | } |
| 182 | __DEVICE__ float __fAtomicAdd_system(float *__p, float __v) { |
| 183 | return __nvvm_atom_sys_add_gen_f(__p, __v); |
| 184 | } |
| 185 | __DEVICE__ float __fAtomicExch(float *__p, float __v) { |
| 186 | return __nv_int_as_float( |
| 187 | __nvvm_atom_xchg_gen_i((int *)__p, __nv_float_as_int(__v))); |
| 188 | } |
| 189 | __DEVICE__ float __fAtomicExch_block(float *__p, float __v) { |
| 190 | return __nv_int_as_float( |
| 191 | __nvvm_atom_cta_xchg_gen_i((int *)__p, __nv_float_as_int(__v))); |
| 192 | } |
| 193 | __DEVICE__ float __fAtomicExch_system(float *__p, float __v) { |
| 194 | return __nv_int_as_float( |
| 195 | __nvvm_atom_sys_xchg_gen_i((int *)__p, __nv_float_as_int(__v))); |
| 196 | } |
| 197 | __DEVICE__ float __fadd_rd(float __a, float __b) { |
| 198 | return __nv_fadd_rd(__a, __b); |
| 199 | } |
| 200 | __DEVICE__ float __fadd_rn(float __a, float __b) { |
| 201 | return __nv_fadd_rn(__a, __b); |
| 202 | } |
| 203 | __DEVICE__ float __fadd_ru(float __a, float __b) { |
| 204 | return __nv_fadd_ru(__a, __b); |
| 205 | } |
| 206 | __DEVICE__ float __fadd_rz(float __a, float __b) { |
| 207 | return __nv_fadd_rz(__a, __b); |
| 208 | } |
| 209 | __DEVICE__ float __fdiv_rd(float __a, float __b) { |
| 210 | return __nv_fdiv_rd(__a, __b); |
| 211 | } |
| 212 | __DEVICE__ float __fdiv_rn(float __a, float __b) { |
| 213 | return __nv_fdiv_rn(__a, __b); |
| 214 | } |
| 215 | __DEVICE__ float __fdiv_ru(float __a, float __b) { |
| 216 | return __nv_fdiv_ru(__a, __b); |
| 217 | } |
| 218 | __DEVICE__ float __fdiv_rz(float __a, float __b) { |
| 219 | return __nv_fdiv_rz(__a, __b); |
| 220 | } |
| 221 | __DEVICE__ float __fdividef(float __a, float __b) { |
| 222 | return __nv_fast_fdividef(__a, __b); |
| 223 | } |
| 224 | __DEVICE__ int __ffs(int __a) { return __nv_ffs(__a); } |
| 225 | __DEVICE__ int __ffsll(long long __a) { return __nv_ffsll(__a); } |
| 226 | __DEVICE__ int __finite(double __a) { return __nv_isfinited(__a); } |
| 227 | __DEVICE__ int __finitef(float __a) { return __nv_finitef(__a); } |
Logan Chien | df4f766 | 2019-09-04 16:45:23 -0700 | [diff] [blame] | 228 | #ifdef _MSC_VER |
| 229 | __DEVICE__ int __finitel(long double __a); |
| 230 | #endif |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 231 | __DEVICE__ int __float2int_rd(float __a) { return __nv_float2int_rd(__a); } |
| 232 | __DEVICE__ int __float2int_rn(float __a) { return __nv_float2int_rn(__a); } |
| 233 | __DEVICE__ int __float2int_ru(float __a) { return __nv_float2int_ru(__a); } |
| 234 | __DEVICE__ int __float2int_rz(float __a) { return __nv_float2int_rz(__a); } |
| 235 | __DEVICE__ long long __float2ll_rd(float __a) { return __nv_float2ll_rd(__a); } |
| 236 | __DEVICE__ long long __float2ll_rn(float __a) { return __nv_float2ll_rn(__a); } |
| 237 | __DEVICE__ long long __float2ll_ru(float __a) { return __nv_float2ll_ru(__a); } |
| 238 | __DEVICE__ long long __float2ll_rz(float __a) { return __nv_float2ll_rz(__a); } |
| 239 | __DEVICE__ unsigned int __float2uint_rd(float __a) { |
| 240 | return __nv_float2uint_rd(__a); |
| 241 | } |
| 242 | __DEVICE__ unsigned int __float2uint_rn(float __a) { |
| 243 | return __nv_float2uint_rn(__a); |
| 244 | } |
| 245 | __DEVICE__ unsigned int __float2uint_ru(float __a) { |
| 246 | return __nv_float2uint_ru(__a); |
| 247 | } |
| 248 | __DEVICE__ unsigned int __float2uint_rz(float __a) { |
| 249 | return __nv_float2uint_rz(__a); |
| 250 | } |
| 251 | __DEVICE__ unsigned long long __float2ull_rd(float __a) { |
| 252 | return __nv_float2ull_rd(__a); |
| 253 | } |
| 254 | __DEVICE__ unsigned long long __float2ull_rn(float __a) { |
| 255 | return __nv_float2ull_rn(__a); |
| 256 | } |
| 257 | __DEVICE__ unsigned long long __float2ull_ru(float __a) { |
| 258 | return __nv_float2ull_ru(__a); |
| 259 | } |
| 260 | __DEVICE__ unsigned long long __float2ull_rz(float __a) { |
| 261 | return __nv_float2ull_rz(__a); |
| 262 | } |
| 263 | __DEVICE__ int __float_as_int(float __a) { return __nv_float_as_int(__a); } |
| 264 | __DEVICE__ unsigned int __float_as_uint(float __a) { |
| 265 | return __nv_float_as_uint(__a); |
| 266 | } |
| 267 | __DEVICE__ double __fma_rd(double __a, double __b, double __c) { |
| 268 | return __nv_fma_rd(__a, __b, __c); |
| 269 | } |
| 270 | __DEVICE__ double __fma_rn(double __a, double __b, double __c) { |
| 271 | return __nv_fma_rn(__a, __b, __c); |
| 272 | } |
| 273 | __DEVICE__ double __fma_ru(double __a, double __b, double __c) { |
| 274 | return __nv_fma_ru(__a, __b, __c); |
| 275 | } |
| 276 | __DEVICE__ double __fma_rz(double __a, double __b, double __c) { |
| 277 | return __nv_fma_rz(__a, __b, __c); |
| 278 | } |
| 279 | __DEVICE__ float __fmaf_ieee_rd(float __a, float __b, float __c) { |
| 280 | return __nv_fmaf_ieee_rd(__a, __b, __c); |
| 281 | } |
| 282 | __DEVICE__ float __fmaf_ieee_rn(float __a, float __b, float __c) { |
| 283 | return __nv_fmaf_ieee_rn(__a, __b, __c); |
| 284 | } |
| 285 | __DEVICE__ float __fmaf_ieee_ru(float __a, float __b, float __c) { |
| 286 | return __nv_fmaf_ieee_ru(__a, __b, __c); |
| 287 | } |
| 288 | __DEVICE__ float __fmaf_ieee_rz(float __a, float __b, float __c) { |
| 289 | return __nv_fmaf_ieee_rz(__a, __b, __c); |
| 290 | } |
| 291 | __DEVICE__ float __fmaf_rd(float __a, float __b, float __c) { |
| 292 | return __nv_fmaf_rd(__a, __b, __c); |
| 293 | } |
| 294 | __DEVICE__ float __fmaf_rn(float __a, float __b, float __c) { |
| 295 | return __nv_fmaf_rn(__a, __b, __c); |
| 296 | } |
| 297 | __DEVICE__ float __fmaf_ru(float __a, float __b, float __c) { |
| 298 | return __nv_fmaf_ru(__a, __b, __c); |
| 299 | } |
| 300 | __DEVICE__ float __fmaf_rz(float __a, float __b, float __c) { |
| 301 | return __nv_fmaf_rz(__a, __b, __c); |
| 302 | } |
| 303 | __DEVICE__ float __fmul_rd(float __a, float __b) { |
| 304 | return __nv_fmul_rd(__a, __b); |
| 305 | } |
| 306 | __DEVICE__ float __fmul_rn(float __a, float __b) { |
| 307 | return __nv_fmul_rn(__a, __b); |
| 308 | } |
| 309 | __DEVICE__ float __fmul_ru(float __a, float __b) { |
| 310 | return __nv_fmul_ru(__a, __b); |
| 311 | } |
| 312 | __DEVICE__ float __fmul_rz(float __a, float __b) { |
| 313 | return __nv_fmul_rz(__a, __b); |
| 314 | } |
| 315 | __DEVICE__ float __frcp_rd(float __a) { return __nv_frcp_rd(__a); } |
| 316 | __DEVICE__ float __frcp_rn(float __a) { return __nv_frcp_rn(__a); } |
| 317 | __DEVICE__ float __frcp_ru(float __a) { return __nv_frcp_ru(__a); } |
| 318 | __DEVICE__ float __frcp_rz(float __a) { return __nv_frcp_rz(__a); } |
| 319 | __DEVICE__ float __frsqrt_rn(float __a) { return __nv_frsqrt_rn(__a); } |
| 320 | __DEVICE__ float __fsqrt_rd(float __a) { return __nv_fsqrt_rd(__a); } |
| 321 | __DEVICE__ float __fsqrt_rn(float __a) { return __nv_fsqrt_rn(__a); } |
| 322 | __DEVICE__ float __fsqrt_ru(float __a) { return __nv_fsqrt_ru(__a); } |
| 323 | __DEVICE__ float __fsqrt_rz(float __a) { return __nv_fsqrt_rz(__a); } |
| 324 | __DEVICE__ float __fsub_rd(float __a, float __b) { |
| 325 | return __nv_fsub_rd(__a, __b); |
| 326 | } |
| 327 | __DEVICE__ float __fsub_rn(float __a, float __b) { |
| 328 | return __nv_fsub_rn(__a, __b); |
| 329 | } |
| 330 | __DEVICE__ float __fsub_ru(float __a, float __b) { |
| 331 | return __nv_fsub_ru(__a, __b); |
| 332 | } |
| 333 | __DEVICE__ float __fsub_rz(float __a, float __b) { |
| 334 | return __nv_fsub_rz(__a, __b); |
| 335 | } |
| 336 | __DEVICE__ int __hadd(int __a, int __b) { return __nv_hadd(__a, __b); } |
| 337 | __DEVICE__ double __hiloint2double(int __a, int __b) { |
| 338 | return __nv_hiloint2double(__a, __b); |
| 339 | } |
| 340 | __DEVICE__ int __iAtomicAdd(int *__p, int __v) { |
| 341 | return __nvvm_atom_add_gen_i(__p, __v); |
| 342 | } |
| 343 | __DEVICE__ int __iAtomicAdd_block(int *__p, int __v) { |
Sasha Smundak | 0fc590b | 2020-10-07 08:11:59 -0700 | [diff] [blame] | 344 | return __nvvm_atom_cta_add_gen_i(__p, __v); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 345 | } |
| 346 | __DEVICE__ int __iAtomicAdd_system(int *__p, int __v) { |
Sasha Smundak | 0fc590b | 2020-10-07 08:11:59 -0700 | [diff] [blame] | 347 | return __nvvm_atom_sys_add_gen_i(__p, __v); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 348 | } |
| 349 | __DEVICE__ int __iAtomicAnd(int *__p, int __v) { |
| 350 | return __nvvm_atom_and_gen_i(__p, __v); |
| 351 | } |
| 352 | __DEVICE__ int __iAtomicAnd_block(int *__p, int __v) { |
| 353 | return __nvvm_atom_cta_and_gen_i(__p, __v); |
| 354 | } |
| 355 | __DEVICE__ int __iAtomicAnd_system(int *__p, int __v) { |
| 356 | return __nvvm_atom_sys_and_gen_i(__p, __v); |
| 357 | } |
| 358 | __DEVICE__ int __iAtomicCAS(int *__p, int __cmp, int __v) { |
| 359 | return __nvvm_atom_cas_gen_i(__p, __cmp, __v); |
| 360 | } |
| 361 | __DEVICE__ int __iAtomicCAS_block(int *__p, int __cmp, int __v) { |
| 362 | return __nvvm_atom_cta_cas_gen_i(__p, __cmp, __v); |
| 363 | } |
| 364 | __DEVICE__ int __iAtomicCAS_system(int *__p, int __cmp, int __v) { |
| 365 | return __nvvm_atom_sys_cas_gen_i(__p, __cmp, __v); |
| 366 | } |
| 367 | __DEVICE__ int __iAtomicExch(int *__p, int __v) { |
| 368 | return __nvvm_atom_xchg_gen_i(__p, __v); |
| 369 | } |
| 370 | __DEVICE__ int __iAtomicExch_block(int *__p, int __v) { |
| 371 | return __nvvm_atom_cta_xchg_gen_i(__p, __v); |
| 372 | } |
| 373 | __DEVICE__ int __iAtomicExch_system(int *__p, int __v) { |
| 374 | return __nvvm_atom_sys_xchg_gen_i(__p, __v); |
| 375 | } |
| 376 | __DEVICE__ int __iAtomicMax(int *__p, int __v) { |
| 377 | return __nvvm_atom_max_gen_i(__p, __v); |
| 378 | } |
| 379 | __DEVICE__ int __iAtomicMax_block(int *__p, int __v) { |
| 380 | return __nvvm_atom_cta_max_gen_i(__p, __v); |
| 381 | } |
| 382 | __DEVICE__ int __iAtomicMax_system(int *__p, int __v) { |
| 383 | return __nvvm_atom_sys_max_gen_i(__p, __v); |
| 384 | } |
| 385 | __DEVICE__ int __iAtomicMin(int *__p, int __v) { |
| 386 | return __nvvm_atom_min_gen_i(__p, __v); |
| 387 | } |
| 388 | __DEVICE__ int __iAtomicMin_block(int *__p, int __v) { |
| 389 | return __nvvm_atom_cta_min_gen_i(__p, __v); |
| 390 | } |
| 391 | __DEVICE__ int __iAtomicMin_system(int *__p, int __v) { |
| 392 | return __nvvm_atom_sys_min_gen_i(__p, __v); |
| 393 | } |
| 394 | __DEVICE__ int __iAtomicOr(int *__p, int __v) { |
| 395 | return __nvvm_atom_or_gen_i(__p, __v); |
| 396 | } |
| 397 | __DEVICE__ int __iAtomicOr_block(int *__p, int __v) { |
| 398 | return __nvvm_atom_cta_or_gen_i(__p, __v); |
| 399 | } |
| 400 | __DEVICE__ int __iAtomicOr_system(int *__p, int __v) { |
| 401 | return __nvvm_atom_sys_or_gen_i(__p, __v); |
| 402 | } |
| 403 | __DEVICE__ int __iAtomicXor(int *__p, int __v) { |
| 404 | return __nvvm_atom_xor_gen_i(__p, __v); |
| 405 | } |
| 406 | __DEVICE__ int __iAtomicXor_block(int *__p, int __v) { |
| 407 | return __nvvm_atom_cta_xor_gen_i(__p, __v); |
| 408 | } |
| 409 | __DEVICE__ int __iAtomicXor_system(int *__p, int __v) { |
| 410 | return __nvvm_atom_sys_xor_gen_i(__p, __v); |
| 411 | } |
| 412 | __DEVICE__ long long __illAtomicMax(long long *__p, long long __v) { |
| 413 | return __nvvm_atom_max_gen_ll(__p, __v); |
| 414 | } |
| 415 | __DEVICE__ long long __illAtomicMax_block(long long *__p, long long __v) { |
| 416 | return __nvvm_atom_cta_max_gen_ll(__p, __v); |
| 417 | } |
| 418 | __DEVICE__ long long __illAtomicMax_system(long long *__p, long long __v) { |
| 419 | return __nvvm_atom_sys_max_gen_ll(__p, __v); |
| 420 | } |
| 421 | __DEVICE__ long long __illAtomicMin(long long *__p, long long __v) { |
| 422 | return __nvvm_atom_min_gen_ll(__p, __v); |
| 423 | } |
| 424 | __DEVICE__ long long __illAtomicMin_block(long long *__p, long long __v) { |
| 425 | return __nvvm_atom_cta_min_gen_ll(__p, __v); |
| 426 | } |
| 427 | __DEVICE__ long long __illAtomicMin_system(long long *__p, long long __v) { |
| 428 | return __nvvm_atom_sys_min_gen_ll(__p, __v); |
| 429 | } |
| 430 | __DEVICE__ double __int2double_rn(int __a) { return __nv_int2double_rn(__a); } |
| 431 | __DEVICE__ float __int2float_rd(int __a) { return __nv_int2float_rd(__a); } |
| 432 | __DEVICE__ float __int2float_rn(int __a) { return __nv_int2float_rn(__a); } |
| 433 | __DEVICE__ float __int2float_ru(int __a) { return __nv_int2float_ru(__a); } |
| 434 | __DEVICE__ float __int2float_rz(int __a) { return __nv_int2float_rz(__a); } |
| 435 | __DEVICE__ float __int_as_float(int __a) { return __nv_int_as_float(__a); } |
| 436 | __DEVICE__ int __isfinited(double __a) { return __nv_isfinited(__a); } |
| 437 | __DEVICE__ int __isinf(double __a) { return __nv_isinfd(__a); } |
| 438 | __DEVICE__ int __isinff(float __a) { return __nv_isinff(__a); } |
Logan Chien | df4f766 | 2019-09-04 16:45:23 -0700 | [diff] [blame] | 439 | #ifdef _MSC_VER |
| 440 | __DEVICE__ int __isinfl(long double __a); |
| 441 | #endif |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 442 | __DEVICE__ int __isnan(double __a) { return __nv_isnand(__a); } |
| 443 | __DEVICE__ int __isnanf(float __a) { return __nv_isnanf(__a); } |
Logan Chien | df4f766 | 2019-09-04 16:45:23 -0700 | [diff] [blame] | 444 | #ifdef _MSC_VER |
| 445 | __DEVICE__ int __isnanl(long double __a); |
| 446 | #endif |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 447 | __DEVICE__ double __ll2double_rd(long long __a) { |
| 448 | return __nv_ll2double_rd(__a); |
| 449 | } |
| 450 | __DEVICE__ double __ll2double_rn(long long __a) { |
| 451 | return __nv_ll2double_rn(__a); |
| 452 | } |
| 453 | __DEVICE__ double __ll2double_ru(long long __a) { |
| 454 | return __nv_ll2double_ru(__a); |
| 455 | } |
| 456 | __DEVICE__ double __ll2double_rz(long long __a) { |
| 457 | return __nv_ll2double_rz(__a); |
| 458 | } |
| 459 | __DEVICE__ float __ll2float_rd(long long __a) { return __nv_ll2float_rd(__a); } |
| 460 | __DEVICE__ float __ll2float_rn(long long __a) { return __nv_ll2float_rn(__a); } |
| 461 | __DEVICE__ float __ll2float_ru(long long __a) { return __nv_ll2float_ru(__a); } |
| 462 | __DEVICE__ float __ll2float_rz(long long __a) { return __nv_ll2float_rz(__a); } |
| 463 | __DEVICE__ long long __llAtomicAnd(long long *__p, long long __v) { |
| 464 | return __nvvm_atom_and_gen_ll(__p, __v); |
| 465 | } |
| 466 | __DEVICE__ long long __llAtomicAnd_block(long long *__p, long long __v) { |
| 467 | return __nvvm_atom_cta_and_gen_ll(__p, __v); |
| 468 | } |
| 469 | __DEVICE__ long long __llAtomicAnd_system(long long *__p, long long __v) { |
| 470 | return __nvvm_atom_sys_and_gen_ll(__p, __v); |
| 471 | } |
| 472 | __DEVICE__ long long __llAtomicOr(long long *__p, long long __v) { |
| 473 | return __nvvm_atom_or_gen_ll(__p, __v); |
| 474 | } |
| 475 | __DEVICE__ long long __llAtomicOr_block(long long *__p, long long __v) { |
| 476 | return __nvvm_atom_cta_or_gen_ll(__p, __v); |
| 477 | } |
| 478 | __DEVICE__ long long __llAtomicOr_system(long long *__p, long long __v) { |
| 479 | return __nvvm_atom_sys_or_gen_ll(__p, __v); |
| 480 | } |
| 481 | __DEVICE__ long long __llAtomicXor(long long *__p, long long __v) { |
| 482 | return __nvvm_atom_xor_gen_ll(__p, __v); |
| 483 | } |
| 484 | __DEVICE__ long long __llAtomicXor_block(long long *__p, long long __v) { |
| 485 | return __nvvm_atom_cta_xor_gen_ll(__p, __v); |
| 486 | } |
| 487 | __DEVICE__ long long __llAtomicXor_system(long long *__p, long long __v) { |
| 488 | return __nvvm_atom_sys_xor_gen_ll(__p, __v); |
| 489 | } |
| 490 | __DEVICE__ float __log10f(float __a) { return __nv_fast_log10f(__a); } |
| 491 | __DEVICE__ float __log2f(float __a) { return __nv_fast_log2f(__a); } |
| 492 | __DEVICE__ float __logf(float __a) { return __nv_fast_logf(__a); } |
| 493 | __DEVICE__ double __longlong_as_double(long long __a) { |
| 494 | return __nv_longlong_as_double(__a); |
| 495 | } |
| 496 | __DEVICE__ int __mul24(int __a, int __b) { return __nv_mul24(__a, __b); } |
| 497 | __DEVICE__ long long __mul64hi(long long __a, long long __b) { |
| 498 | return __nv_mul64hi(__a, __b); |
| 499 | } |
| 500 | __DEVICE__ int __mulhi(int __a, int __b) { return __nv_mulhi(__a, __b); } |
| 501 | __DEVICE__ unsigned int __pm0(void) { return __nvvm_read_ptx_sreg_pm0(); } |
| 502 | __DEVICE__ unsigned int __pm1(void) { return __nvvm_read_ptx_sreg_pm1(); } |
| 503 | __DEVICE__ unsigned int __pm2(void) { return __nvvm_read_ptx_sreg_pm2(); } |
| 504 | __DEVICE__ unsigned int __pm3(void) { return __nvvm_read_ptx_sreg_pm3(); } |
| 505 | __DEVICE__ int __popc(int __a) { return __nv_popc(__a); } |
| 506 | __DEVICE__ int __popcll(long long __a) { return __nv_popcll(__a); } |
| 507 | __DEVICE__ float __powf(float __a, float __b) { |
| 508 | return __nv_fast_powf(__a, __b); |
| 509 | } |
| 510 | |
| 511 | // Parameter must have a known integer value. |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 512 | #define __prof_trigger(__a) __asm__ __volatile__("pmevent \t%0;" ::"i"(__a)) |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 513 | __DEVICE__ int __rhadd(int __a, int __b) { return __nv_rhadd(__a, __b); } |
| 514 | __DEVICE__ unsigned int __sad(int __a, int __b, unsigned int __c) { |
| 515 | return __nv_sad(__a, __b, __c); |
| 516 | } |
| 517 | __DEVICE__ float __saturatef(float __a) { return __nv_saturatef(__a); } |
| 518 | __DEVICE__ int __signbitd(double __a) { return __nv_signbitd(__a); } |
| 519 | __DEVICE__ int __signbitf(float __a) { return __nv_signbitf(__a); } |
Logan Chien | df4f766 | 2019-09-04 16:45:23 -0700 | [diff] [blame] | 520 | __DEVICE__ void __sincosf(float __a, float *__s, float *__c) { |
| 521 | return __nv_fast_sincosf(__a, __s, __c); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 522 | } |
| 523 | __DEVICE__ float __sinf(float __a) { return __nv_fast_sinf(__a); } |
| 524 | __DEVICE__ int __syncthreads_and(int __a) { return __nvvm_bar0_and(__a); } |
| 525 | __DEVICE__ int __syncthreads_count(int __a) { return __nvvm_bar0_popc(__a); } |
| 526 | __DEVICE__ int __syncthreads_or(int __a) { return __nvvm_bar0_or(__a); } |
| 527 | __DEVICE__ float __tanf(float __a) { return __nv_fast_tanf(__a); } |
| 528 | __DEVICE__ void __threadfence(void) { __nvvm_membar_gl(); } |
| 529 | __DEVICE__ void __threadfence_block(void) { __nvvm_membar_cta(); }; |
| 530 | __DEVICE__ void __threadfence_system(void) { __nvvm_membar_sys(); }; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 531 | __DEVICE__ void __trap(void) { __asm__ __volatile__("trap;"); } |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 532 | __DEVICE__ unsigned int __uAtomicAdd(unsigned int *__p, unsigned int __v) { |
| 533 | return __nvvm_atom_add_gen_i((int *)__p, __v); |
| 534 | } |
| 535 | __DEVICE__ unsigned int __uAtomicAdd_block(unsigned int *__p, |
| 536 | unsigned int __v) { |
| 537 | return __nvvm_atom_cta_add_gen_i((int *)__p, __v); |
| 538 | } |
| 539 | __DEVICE__ unsigned int __uAtomicAdd_system(unsigned int *__p, |
| 540 | unsigned int __v) { |
| 541 | return __nvvm_atom_sys_add_gen_i((int *)__p, __v); |
| 542 | } |
| 543 | __DEVICE__ unsigned int __uAtomicAnd(unsigned int *__p, unsigned int __v) { |
| 544 | return __nvvm_atom_and_gen_i((int *)__p, __v); |
| 545 | } |
| 546 | __DEVICE__ unsigned int __uAtomicAnd_block(unsigned int *__p, |
| 547 | unsigned int __v) { |
| 548 | return __nvvm_atom_cta_and_gen_i((int *)__p, __v); |
| 549 | } |
| 550 | __DEVICE__ unsigned int __uAtomicAnd_system(unsigned int *__p, |
| 551 | unsigned int __v) { |
| 552 | return __nvvm_atom_sys_and_gen_i((int *)__p, __v); |
| 553 | } |
| 554 | __DEVICE__ unsigned int __uAtomicCAS(unsigned int *__p, unsigned int __cmp, |
| 555 | unsigned int __v) { |
| 556 | return __nvvm_atom_cas_gen_i((int *)__p, __cmp, __v); |
| 557 | } |
| 558 | __DEVICE__ unsigned int |
| 559 | __uAtomicCAS_block(unsigned int *__p, unsigned int __cmp, unsigned int __v) { |
| 560 | return __nvvm_atom_cta_cas_gen_i((int *)__p, __cmp, __v); |
| 561 | } |
| 562 | __DEVICE__ unsigned int |
| 563 | __uAtomicCAS_system(unsigned int *__p, unsigned int __cmp, unsigned int __v) { |
| 564 | return __nvvm_atom_sys_cas_gen_i((int *)__p, __cmp, __v); |
| 565 | } |
| 566 | __DEVICE__ unsigned int __uAtomicDec(unsigned int *__p, unsigned int __v) { |
| 567 | return __nvvm_atom_dec_gen_ui(__p, __v); |
| 568 | } |
| 569 | __DEVICE__ unsigned int __uAtomicDec_block(unsigned int *__p, |
| 570 | unsigned int __v) { |
| 571 | return __nvvm_atom_cta_dec_gen_ui(__p, __v); |
| 572 | } |
| 573 | __DEVICE__ unsigned int __uAtomicDec_system(unsigned int *__p, |
| 574 | unsigned int __v) { |
| 575 | return __nvvm_atom_sys_dec_gen_ui(__p, __v); |
| 576 | } |
| 577 | __DEVICE__ unsigned int __uAtomicExch(unsigned int *__p, unsigned int __v) { |
| 578 | return __nvvm_atom_xchg_gen_i((int *)__p, __v); |
| 579 | } |
| 580 | __DEVICE__ unsigned int __uAtomicExch_block(unsigned int *__p, |
| 581 | unsigned int __v) { |
| 582 | return __nvvm_atom_cta_xchg_gen_i((int *)__p, __v); |
| 583 | } |
| 584 | __DEVICE__ unsigned int __uAtomicExch_system(unsigned int *__p, |
| 585 | unsigned int __v) { |
| 586 | return __nvvm_atom_sys_xchg_gen_i((int *)__p, __v); |
| 587 | } |
| 588 | __DEVICE__ unsigned int __uAtomicInc(unsigned int *__p, unsigned int __v) { |
| 589 | return __nvvm_atom_inc_gen_ui(__p, __v); |
| 590 | } |
| 591 | __DEVICE__ unsigned int __uAtomicInc_block(unsigned int *__p, |
| 592 | unsigned int __v) { |
| 593 | return __nvvm_atom_cta_inc_gen_ui(__p, __v); |
| 594 | } |
| 595 | __DEVICE__ unsigned int __uAtomicInc_system(unsigned int *__p, |
| 596 | unsigned int __v) { |
| 597 | return __nvvm_atom_sys_inc_gen_ui(__p, __v); |
| 598 | } |
| 599 | __DEVICE__ unsigned int __uAtomicMax(unsigned int *__p, unsigned int __v) { |
| 600 | return __nvvm_atom_max_gen_ui(__p, __v); |
| 601 | } |
| 602 | __DEVICE__ unsigned int __uAtomicMax_block(unsigned int *__p, |
| 603 | unsigned int __v) { |
| 604 | return __nvvm_atom_cta_max_gen_ui(__p, __v); |
| 605 | } |
| 606 | __DEVICE__ unsigned int __uAtomicMax_system(unsigned int *__p, |
| 607 | unsigned int __v) { |
| 608 | return __nvvm_atom_sys_max_gen_ui(__p, __v); |
| 609 | } |
| 610 | __DEVICE__ unsigned int __uAtomicMin(unsigned int *__p, unsigned int __v) { |
| 611 | return __nvvm_atom_min_gen_ui(__p, __v); |
| 612 | } |
| 613 | __DEVICE__ unsigned int __uAtomicMin_block(unsigned int *__p, |
| 614 | unsigned int __v) { |
| 615 | return __nvvm_atom_cta_min_gen_ui(__p, __v); |
| 616 | } |
| 617 | __DEVICE__ unsigned int __uAtomicMin_system(unsigned int *__p, |
| 618 | unsigned int __v) { |
| 619 | return __nvvm_atom_sys_min_gen_ui(__p, __v); |
| 620 | } |
| 621 | __DEVICE__ unsigned int __uAtomicOr(unsigned int *__p, unsigned int __v) { |
| 622 | return __nvvm_atom_or_gen_i((int *)__p, __v); |
| 623 | } |
| 624 | __DEVICE__ unsigned int __uAtomicOr_block(unsigned int *__p, unsigned int __v) { |
| 625 | return __nvvm_atom_cta_or_gen_i((int *)__p, __v); |
| 626 | } |
| 627 | __DEVICE__ unsigned int __uAtomicOr_system(unsigned int *__p, |
| 628 | unsigned int __v) { |
| 629 | return __nvvm_atom_sys_or_gen_i((int *)__p, __v); |
| 630 | } |
| 631 | __DEVICE__ unsigned int __uAtomicXor(unsigned int *__p, unsigned int __v) { |
| 632 | return __nvvm_atom_xor_gen_i((int *)__p, __v); |
| 633 | } |
| 634 | __DEVICE__ unsigned int __uAtomicXor_block(unsigned int *__p, |
| 635 | unsigned int __v) { |
| 636 | return __nvvm_atom_cta_xor_gen_i((int *)__p, __v); |
| 637 | } |
| 638 | __DEVICE__ unsigned int __uAtomicXor_system(unsigned int *__p, |
| 639 | unsigned int __v) { |
| 640 | return __nvvm_atom_sys_xor_gen_i((int *)__p, __v); |
| 641 | } |
| 642 | __DEVICE__ unsigned int __uhadd(unsigned int __a, unsigned int __b) { |
| 643 | return __nv_uhadd(__a, __b); |
| 644 | } |
| 645 | __DEVICE__ double __uint2double_rn(unsigned int __a) { |
| 646 | return __nv_uint2double_rn(__a); |
| 647 | } |
| 648 | __DEVICE__ float __uint2float_rd(unsigned int __a) { |
| 649 | return __nv_uint2float_rd(__a); |
| 650 | } |
| 651 | __DEVICE__ float __uint2float_rn(unsigned int __a) { |
| 652 | return __nv_uint2float_rn(__a); |
| 653 | } |
| 654 | __DEVICE__ float __uint2float_ru(unsigned int __a) { |
| 655 | return __nv_uint2float_ru(__a); |
| 656 | } |
| 657 | __DEVICE__ float __uint2float_rz(unsigned int __a) { |
| 658 | return __nv_uint2float_rz(__a); |
| 659 | } |
| 660 | __DEVICE__ float __uint_as_float(unsigned int __a) { |
| 661 | return __nv_uint_as_float(__a); |
| 662 | } // |
| 663 | __DEVICE__ double __ull2double_rd(unsigned long long __a) { |
| 664 | return __nv_ull2double_rd(__a); |
| 665 | } |
| 666 | __DEVICE__ double __ull2double_rn(unsigned long long __a) { |
| 667 | return __nv_ull2double_rn(__a); |
| 668 | } |
| 669 | __DEVICE__ double __ull2double_ru(unsigned long long __a) { |
| 670 | return __nv_ull2double_ru(__a); |
| 671 | } |
| 672 | __DEVICE__ double __ull2double_rz(unsigned long long __a) { |
| 673 | return __nv_ull2double_rz(__a); |
| 674 | } |
| 675 | __DEVICE__ float __ull2float_rd(unsigned long long __a) { |
| 676 | return __nv_ull2float_rd(__a); |
| 677 | } |
| 678 | __DEVICE__ float __ull2float_rn(unsigned long long __a) { |
| 679 | return __nv_ull2float_rn(__a); |
| 680 | } |
| 681 | __DEVICE__ float __ull2float_ru(unsigned long long __a) { |
| 682 | return __nv_ull2float_ru(__a); |
| 683 | } |
| 684 | __DEVICE__ float __ull2float_rz(unsigned long long __a) { |
| 685 | return __nv_ull2float_rz(__a); |
| 686 | } |
| 687 | __DEVICE__ unsigned long long __ullAtomicAdd(unsigned long long *__p, |
| 688 | unsigned long long __v) { |
| 689 | return __nvvm_atom_add_gen_ll((long long *)__p, __v); |
| 690 | } |
| 691 | __DEVICE__ unsigned long long __ullAtomicAdd_block(unsigned long long *__p, |
| 692 | unsigned long long __v) { |
| 693 | return __nvvm_atom_cta_add_gen_ll((long long *)__p, __v); |
| 694 | } |
| 695 | __DEVICE__ unsigned long long __ullAtomicAdd_system(unsigned long long *__p, |
| 696 | unsigned long long __v) { |
| 697 | return __nvvm_atom_sys_add_gen_ll((long long *)__p, __v); |
| 698 | } |
| 699 | __DEVICE__ unsigned long long __ullAtomicAnd(unsigned long long *__p, |
| 700 | unsigned long long __v) { |
| 701 | return __nvvm_atom_and_gen_ll((long long *)__p, __v); |
| 702 | } |
| 703 | __DEVICE__ unsigned long long __ullAtomicAnd_block(unsigned long long *__p, |
| 704 | unsigned long long __v) { |
| 705 | return __nvvm_atom_cta_and_gen_ll((long long *)__p, __v); |
| 706 | } |
| 707 | __DEVICE__ unsigned long long __ullAtomicAnd_system(unsigned long long *__p, |
| 708 | unsigned long long __v) { |
| 709 | return __nvvm_atom_sys_and_gen_ll((long long *)__p, __v); |
| 710 | } |
| 711 | __DEVICE__ unsigned long long __ullAtomicCAS(unsigned long long *__p, |
| 712 | unsigned long long __cmp, |
| 713 | unsigned long long __v) { |
| 714 | return __nvvm_atom_cas_gen_ll((long long *)__p, __cmp, __v); |
| 715 | } |
| 716 | __DEVICE__ unsigned long long __ullAtomicCAS_block(unsigned long long *__p, |
| 717 | unsigned long long __cmp, |
| 718 | unsigned long long __v) { |
| 719 | return __nvvm_atom_cta_cas_gen_ll((long long *)__p, __cmp, __v); |
| 720 | } |
| 721 | __DEVICE__ unsigned long long __ullAtomicCAS_system(unsigned long long *__p, |
| 722 | unsigned long long __cmp, |
| 723 | unsigned long long __v) { |
| 724 | return __nvvm_atom_sys_cas_gen_ll((long long *)__p, __cmp, __v); |
| 725 | } |
| 726 | __DEVICE__ unsigned long long __ullAtomicExch(unsigned long long *__p, |
| 727 | unsigned long long __v) { |
| 728 | return __nvvm_atom_xchg_gen_ll((long long *)__p, __v); |
| 729 | } |
| 730 | __DEVICE__ unsigned long long __ullAtomicExch_block(unsigned long long *__p, |
| 731 | unsigned long long __v) { |
| 732 | return __nvvm_atom_cta_xchg_gen_ll((long long *)__p, __v); |
| 733 | } |
| 734 | __DEVICE__ unsigned long long __ullAtomicExch_system(unsigned long long *__p, |
| 735 | unsigned long long __v) { |
| 736 | return __nvvm_atom_sys_xchg_gen_ll((long long *)__p, __v); |
| 737 | } |
| 738 | __DEVICE__ unsigned long long __ullAtomicMax(unsigned long long *__p, |
| 739 | unsigned long long __v) { |
| 740 | return __nvvm_atom_max_gen_ull(__p, __v); |
| 741 | } |
| 742 | __DEVICE__ unsigned long long __ullAtomicMax_block(unsigned long long *__p, |
| 743 | unsigned long long __v) { |
| 744 | return __nvvm_atom_cta_max_gen_ull(__p, __v); |
| 745 | } |
| 746 | __DEVICE__ unsigned long long __ullAtomicMax_system(unsigned long long *__p, |
| 747 | unsigned long long __v) { |
| 748 | return __nvvm_atom_sys_max_gen_ull(__p, __v); |
| 749 | } |
| 750 | __DEVICE__ unsigned long long __ullAtomicMin(unsigned long long *__p, |
| 751 | unsigned long long __v) { |
| 752 | return __nvvm_atom_min_gen_ull(__p, __v); |
| 753 | } |
| 754 | __DEVICE__ unsigned long long __ullAtomicMin_block(unsigned long long *__p, |
| 755 | unsigned long long __v) { |
| 756 | return __nvvm_atom_cta_min_gen_ull(__p, __v); |
| 757 | } |
| 758 | __DEVICE__ unsigned long long __ullAtomicMin_system(unsigned long long *__p, |
| 759 | unsigned long long __v) { |
| 760 | return __nvvm_atom_sys_min_gen_ull(__p, __v); |
| 761 | } |
| 762 | __DEVICE__ unsigned long long __ullAtomicOr(unsigned long long *__p, |
| 763 | unsigned long long __v) { |
| 764 | return __nvvm_atom_or_gen_ll((long long *)__p, __v); |
| 765 | } |
| 766 | __DEVICE__ unsigned long long __ullAtomicOr_block(unsigned long long *__p, |
| 767 | unsigned long long __v) { |
| 768 | return __nvvm_atom_cta_or_gen_ll((long long *)__p, __v); |
| 769 | } |
| 770 | __DEVICE__ unsigned long long __ullAtomicOr_system(unsigned long long *__p, |
| 771 | unsigned long long __v) { |
| 772 | return __nvvm_atom_sys_or_gen_ll((long long *)__p, __v); |
| 773 | } |
| 774 | __DEVICE__ unsigned long long __ullAtomicXor(unsigned long long *__p, |
| 775 | unsigned long long __v) { |
| 776 | return __nvvm_atom_xor_gen_ll((long long *)__p, __v); |
| 777 | } |
| 778 | __DEVICE__ unsigned long long __ullAtomicXor_block(unsigned long long *__p, |
| 779 | unsigned long long __v) { |
| 780 | return __nvvm_atom_cta_xor_gen_ll((long long *)__p, __v); |
| 781 | } |
| 782 | __DEVICE__ unsigned long long __ullAtomicXor_system(unsigned long long *__p, |
| 783 | unsigned long long __v) { |
| 784 | return __nvvm_atom_sys_xor_gen_ll((long long *)__p, __v); |
| 785 | } |
| 786 | __DEVICE__ unsigned int __umul24(unsigned int __a, unsigned int __b) { |
| 787 | return __nv_umul24(__a, __b); |
| 788 | } |
| 789 | __DEVICE__ unsigned long long __umul64hi(unsigned long long __a, |
| 790 | unsigned long long __b) { |
| 791 | return __nv_umul64hi(__a, __b); |
| 792 | } |
| 793 | __DEVICE__ unsigned int __umulhi(unsigned int __a, unsigned int __b) { |
| 794 | return __nv_umulhi(__a, __b); |
| 795 | } |
| 796 | __DEVICE__ unsigned int __urhadd(unsigned int __a, unsigned int __b) { |
| 797 | return __nv_urhadd(__a, __b); |
| 798 | } |
| 799 | __DEVICE__ unsigned int __usad(unsigned int __a, unsigned int __b, |
| 800 | unsigned int __c) { |
| 801 | return __nv_usad(__a, __b, __c); |
| 802 | } |
| 803 | |
| 804 | #if CUDA_VERSION >= 9000 && CUDA_VERSION < 9020 |
| 805 | __DEVICE__ unsigned int __vabs2(unsigned int __a) { return __nv_vabs2(__a); } |
| 806 | __DEVICE__ unsigned int __vabs4(unsigned int __a) { return __nv_vabs4(__a); } |
| 807 | __DEVICE__ unsigned int __vabsdiffs2(unsigned int __a, unsigned int __b) { |
| 808 | return __nv_vabsdiffs2(__a, __b); |
| 809 | } |
| 810 | __DEVICE__ unsigned int __vabsdiffs4(unsigned int __a, unsigned int __b) { |
| 811 | return __nv_vabsdiffs4(__a, __b); |
| 812 | } |
| 813 | __DEVICE__ unsigned int __vabsdiffu2(unsigned int __a, unsigned int __b) { |
| 814 | return __nv_vabsdiffu2(__a, __b); |
| 815 | } |
| 816 | __DEVICE__ unsigned int __vabsdiffu4(unsigned int __a, unsigned int __b) { |
| 817 | return __nv_vabsdiffu4(__a, __b); |
| 818 | } |
| 819 | __DEVICE__ unsigned int __vabsss2(unsigned int __a) { |
| 820 | return __nv_vabsss2(__a); |
| 821 | } |
| 822 | __DEVICE__ unsigned int __vabsss4(unsigned int __a) { |
| 823 | return __nv_vabsss4(__a); |
| 824 | } |
| 825 | __DEVICE__ unsigned int __vadd2(unsigned int __a, unsigned int __b) { |
| 826 | return __nv_vadd2(__a, __b); |
| 827 | } |
| 828 | __DEVICE__ unsigned int __vadd4(unsigned int __a, unsigned int __b) { |
| 829 | return __nv_vadd4(__a, __b); |
| 830 | } |
| 831 | __DEVICE__ unsigned int __vaddss2(unsigned int __a, unsigned int __b) { |
| 832 | return __nv_vaddss2(__a, __b); |
| 833 | } |
| 834 | __DEVICE__ unsigned int __vaddss4(unsigned int __a, unsigned int __b) { |
| 835 | return __nv_vaddss4(__a, __b); |
| 836 | } |
| 837 | __DEVICE__ unsigned int __vaddus2(unsigned int __a, unsigned int __b) { |
| 838 | return __nv_vaddus2(__a, __b); |
| 839 | } |
| 840 | __DEVICE__ unsigned int __vaddus4(unsigned int __a, unsigned int __b) { |
| 841 | return __nv_vaddus4(__a, __b); |
| 842 | } |
| 843 | __DEVICE__ unsigned int __vavgs2(unsigned int __a, unsigned int __b) { |
| 844 | return __nv_vavgs2(__a, __b); |
| 845 | } |
| 846 | __DEVICE__ unsigned int __vavgs4(unsigned int __a, unsigned int __b) { |
| 847 | return __nv_vavgs4(__a, __b); |
| 848 | } |
| 849 | __DEVICE__ unsigned int __vavgu2(unsigned int __a, unsigned int __b) { |
| 850 | return __nv_vavgu2(__a, __b); |
| 851 | } |
| 852 | __DEVICE__ unsigned int __vavgu4(unsigned int __a, unsigned int __b) { |
| 853 | return __nv_vavgu4(__a, __b); |
| 854 | } |
| 855 | __DEVICE__ unsigned int __vcmpeq2(unsigned int __a, unsigned int __b) { |
| 856 | return __nv_vcmpeq2(__a, __b); |
| 857 | } |
| 858 | __DEVICE__ unsigned int __vcmpeq4(unsigned int __a, unsigned int __b) { |
| 859 | return __nv_vcmpeq4(__a, __b); |
| 860 | } |
| 861 | __DEVICE__ unsigned int __vcmpges2(unsigned int __a, unsigned int __b) { |
| 862 | return __nv_vcmpges2(__a, __b); |
| 863 | } |
| 864 | __DEVICE__ unsigned int __vcmpges4(unsigned int __a, unsigned int __b) { |
| 865 | return __nv_vcmpges4(__a, __b); |
| 866 | } |
| 867 | __DEVICE__ unsigned int __vcmpgeu2(unsigned int __a, unsigned int __b) { |
| 868 | return __nv_vcmpgeu2(__a, __b); |
| 869 | } |
| 870 | __DEVICE__ unsigned int __vcmpgeu4(unsigned int __a, unsigned int __b) { |
| 871 | return __nv_vcmpgeu4(__a, __b); |
| 872 | } |
| 873 | __DEVICE__ unsigned int __vcmpgts2(unsigned int __a, unsigned int __b) { |
| 874 | return __nv_vcmpgts2(__a, __b); |
| 875 | } |
| 876 | __DEVICE__ unsigned int __vcmpgts4(unsigned int __a, unsigned int __b) { |
| 877 | return __nv_vcmpgts4(__a, __b); |
| 878 | } |
| 879 | __DEVICE__ unsigned int __vcmpgtu2(unsigned int __a, unsigned int __b) { |
| 880 | return __nv_vcmpgtu2(__a, __b); |
| 881 | } |
| 882 | __DEVICE__ unsigned int __vcmpgtu4(unsigned int __a, unsigned int __b) { |
| 883 | return __nv_vcmpgtu4(__a, __b); |
| 884 | } |
| 885 | __DEVICE__ unsigned int __vcmples2(unsigned int __a, unsigned int __b) { |
| 886 | return __nv_vcmples2(__a, __b); |
| 887 | } |
| 888 | __DEVICE__ unsigned int __vcmples4(unsigned int __a, unsigned int __b) { |
| 889 | return __nv_vcmples4(__a, __b); |
| 890 | } |
| 891 | __DEVICE__ unsigned int __vcmpleu2(unsigned int __a, unsigned int __b) { |
| 892 | return __nv_vcmpleu2(__a, __b); |
| 893 | } |
| 894 | __DEVICE__ unsigned int __vcmpleu4(unsigned int __a, unsigned int __b) { |
| 895 | return __nv_vcmpleu4(__a, __b); |
| 896 | } |
| 897 | __DEVICE__ unsigned int __vcmplts2(unsigned int __a, unsigned int __b) { |
| 898 | return __nv_vcmplts2(__a, __b); |
| 899 | } |
| 900 | __DEVICE__ unsigned int __vcmplts4(unsigned int __a, unsigned int __b) { |
| 901 | return __nv_vcmplts4(__a, __b); |
| 902 | } |
| 903 | __DEVICE__ unsigned int __vcmpltu2(unsigned int __a, unsigned int __b) { |
| 904 | return __nv_vcmpltu2(__a, __b); |
| 905 | } |
| 906 | __DEVICE__ unsigned int __vcmpltu4(unsigned int __a, unsigned int __b) { |
| 907 | return __nv_vcmpltu4(__a, __b); |
| 908 | } |
| 909 | __DEVICE__ unsigned int __vcmpne2(unsigned int __a, unsigned int __b) { |
| 910 | return __nv_vcmpne2(__a, __b); |
| 911 | } |
| 912 | __DEVICE__ unsigned int __vcmpne4(unsigned int __a, unsigned int __b) { |
| 913 | return __nv_vcmpne4(__a, __b); |
| 914 | } |
| 915 | __DEVICE__ unsigned int __vhaddu2(unsigned int __a, unsigned int __b) { |
| 916 | return __nv_vhaddu2(__a, __b); |
| 917 | } |
| 918 | __DEVICE__ unsigned int __vhaddu4(unsigned int __a, unsigned int __b) { |
| 919 | return __nv_vhaddu4(__a, __b); |
| 920 | } |
| 921 | __DEVICE__ unsigned int __vmaxs2(unsigned int __a, unsigned int __b) { |
| 922 | return __nv_vmaxs2(__a, __b); |
| 923 | } |
| 924 | __DEVICE__ unsigned int __vmaxs4(unsigned int __a, unsigned int __b) { |
| 925 | return __nv_vmaxs4(__a, __b); |
| 926 | } |
| 927 | __DEVICE__ unsigned int __vmaxu2(unsigned int __a, unsigned int __b) { |
| 928 | return __nv_vmaxu2(__a, __b); |
| 929 | } |
| 930 | __DEVICE__ unsigned int __vmaxu4(unsigned int __a, unsigned int __b) { |
| 931 | return __nv_vmaxu4(__a, __b); |
| 932 | } |
| 933 | __DEVICE__ unsigned int __vmins2(unsigned int __a, unsigned int __b) { |
| 934 | return __nv_vmins2(__a, __b); |
| 935 | } |
| 936 | __DEVICE__ unsigned int __vmins4(unsigned int __a, unsigned int __b) { |
| 937 | return __nv_vmins4(__a, __b); |
| 938 | } |
| 939 | __DEVICE__ unsigned int __vminu2(unsigned int __a, unsigned int __b) { |
| 940 | return __nv_vminu2(__a, __b); |
| 941 | } |
| 942 | __DEVICE__ unsigned int __vminu4(unsigned int __a, unsigned int __b) { |
| 943 | return __nv_vminu4(__a, __b); |
| 944 | } |
| 945 | __DEVICE__ unsigned int __vneg2(unsigned int __a) { return __nv_vneg2(__a); } |
| 946 | __DEVICE__ unsigned int __vneg4(unsigned int __a) { return __nv_vneg4(__a); } |
| 947 | __DEVICE__ unsigned int __vnegss2(unsigned int __a) { |
| 948 | return __nv_vnegss2(__a); |
| 949 | } |
| 950 | __DEVICE__ unsigned int __vnegss4(unsigned int __a) { |
| 951 | return __nv_vnegss4(__a); |
| 952 | } |
| 953 | __DEVICE__ unsigned int __vsads2(unsigned int __a, unsigned int __b) { |
| 954 | return __nv_vsads2(__a, __b); |
| 955 | } |
| 956 | __DEVICE__ unsigned int __vsads4(unsigned int __a, unsigned int __b) { |
| 957 | return __nv_vsads4(__a, __b); |
| 958 | } |
| 959 | __DEVICE__ unsigned int __vsadu2(unsigned int __a, unsigned int __b) { |
| 960 | return __nv_vsadu2(__a, __b); |
| 961 | } |
| 962 | __DEVICE__ unsigned int __vsadu4(unsigned int __a, unsigned int __b) { |
| 963 | return __nv_vsadu4(__a, __b); |
| 964 | } |
| 965 | __DEVICE__ unsigned int __vseteq2(unsigned int __a, unsigned int __b) { |
| 966 | return __nv_vseteq2(__a, __b); |
| 967 | } |
| 968 | __DEVICE__ unsigned int __vseteq4(unsigned int __a, unsigned int __b) { |
| 969 | return __nv_vseteq4(__a, __b); |
| 970 | } |
| 971 | __DEVICE__ unsigned int __vsetges2(unsigned int __a, unsigned int __b) { |
| 972 | return __nv_vsetges2(__a, __b); |
| 973 | } |
| 974 | __DEVICE__ unsigned int __vsetges4(unsigned int __a, unsigned int __b) { |
| 975 | return __nv_vsetges4(__a, __b); |
| 976 | } |
| 977 | __DEVICE__ unsigned int __vsetgeu2(unsigned int __a, unsigned int __b) { |
| 978 | return __nv_vsetgeu2(__a, __b); |
| 979 | } |
| 980 | __DEVICE__ unsigned int __vsetgeu4(unsigned int __a, unsigned int __b) { |
| 981 | return __nv_vsetgeu4(__a, __b); |
| 982 | } |
| 983 | __DEVICE__ unsigned int __vsetgts2(unsigned int __a, unsigned int __b) { |
| 984 | return __nv_vsetgts2(__a, __b); |
| 985 | } |
| 986 | __DEVICE__ unsigned int __vsetgts4(unsigned int __a, unsigned int __b) { |
| 987 | return __nv_vsetgts4(__a, __b); |
| 988 | } |
| 989 | __DEVICE__ unsigned int __vsetgtu2(unsigned int __a, unsigned int __b) { |
| 990 | return __nv_vsetgtu2(__a, __b); |
| 991 | } |
| 992 | __DEVICE__ unsigned int __vsetgtu4(unsigned int __a, unsigned int __b) { |
| 993 | return __nv_vsetgtu4(__a, __b); |
| 994 | } |
| 995 | __DEVICE__ unsigned int __vsetles2(unsigned int __a, unsigned int __b) { |
| 996 | return __nv_vsetles2(__a, __b); |
| 997 | } |
| 998 | __DEVICE__ unsigned int __vsetles4(unsigned int __a, unsigned int __b) { |
| 999 | return __nv_vsetles4(__a, __b); |
| 1000 | } |
| 1001 | __DEVICE__ unsigned int __vsetleu2(unsigned int __a, unsigned int __b) { |
| 1002 | return __nv_vsetleu2(__a, __b); |
| 1003 | } |
| 1004 | __DEVICE__ unsigned int __vsetleu4(unsigned int __a, unsigned int __b) { |
| 1005 | return __nv_vsetleu4(__a, __b); |
| 1006 | } |
| 1007 | __DEVICE__ unsigned int __vsetlts2(unsigned int __a, unsigned int __b) { |
| 1008 | return __nv_vsetlts2(__a, __b); |
| 1009 | } |
| 1010 | __DEVICE__ unsigned int __vsetlts4(unsigned int __a, unsigned int __b) { |
| 1011 | return __nv_vsetlts4(__a, __b); |
| 1012 | } |
| 1013 | __DEVICE__ unsigned int __vsetltu2(unsigned int __a, unsigned int __b) { |
| 1014 | return __nv_vsetltu2(__a, __b); |
| 1015 | } |
| 1016 | __DEVICE__ unsigned int __vsetltu4(unsigned int __a, unsigned int __b) { |
| 1017 | return __nv_vsetltu4(__a, __b); |
| 1018 | } |
| 1019 | __DEVICE__ unsigned int __vsetne2(unsigned int __a, unsigned int __b) { |
| 1020 | return __nv_vsetne2(__a, __b); |
| 1021 | } |
| 1022 | __DEVICE__ unsigned int __vsetne4(unsigned int __a, unsigned int __b) { |
| 1023 | return __nv_vsetne4(__a, __b); |
| 1024 | } |
| 1025 | __DEVICE__ unsigned int __vsub2(unsigned int __a, unsigned int __b) { |
| 1026 | return __nv_vsub2(__a, __b); |
| 1027 | } |
| 1028 | __DEVICE__ unsigned int __vsub4(unsigned int __a, unsigned int __b) { |
| 1029 | return __nv_vsub4(__a, __b); |
| 1030 | } |
| 1031 | __DEVICE__ unsigned int __vsubss2(unsigned int __a, unsigned int __b) { |
| 1032 | return __nv_vsubss2(__a, __b); |
| 1033 | } |
| 1034 | __DEVICE__ unsigned int __vsubss4(unsigned int __a, unsigned int __b) { |
| 1035 | return __nv_vsubss4(__a, __b); |
| 1036 | } |
| 1037 | __DEVICE__ unsigned int __vsubus2(unsigned int __a, unsigned int __b) { |
| 1038 | return __nv_vsubus2(__a, __b); |
| 1039 | } |
| 1040 | __DEVICE__ unsigned int __vsubus4(unsigned int __a, unsigned int __b) { |
| 1041 | return __nv_vsubus4(__a, __b); |
| 1042 | } |
| 1043 | #else // CUDA_VERSION >= 9020 |
| 1044 | // CUDA no longer provides inline assembly (or bitcode) implementation of these |
| 1045 | // functions, so we have to reimplment them. The implementation is naive and is |
| 1046 | // not optimized for performance. |
| 1047 | |
| 1048 | // Helper function to convert N-bit boolean subfields into all-0 or all-1. |
| 1049 | // E.g. __bool2mask(0x01000100,8) -> 0xff00ff00 |
| 1050 | // __bool2mask(0x00010000,16) -> 0xffff0000 |
| 1051 | __DEVICE__ unsigned int __bool2mask(unsigned int __a, int shift) { |
| 1052 | return (__a << shift) - __a; |
| 1053 | } |
| 1054 | __DEVICE__ unsigned int __vabs2(unsigned int __a) { |
| 1055 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1056 | __asm__("vabsdiff2.s32.s32.s32 %0,%1,%2,%3;" |
| 1057 | : "=r"(r) |
| 1058 | : "r"(__a), "r"(0), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1059 | return r; |
| 1060 | } |
| 1061 | __DEVICE__ unsigned int __vabs4(unsigned int __a) { |
| 1062 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1063 | __asm__("vabsdiff4.s32.s32.s32 %0,%1,%2,%3;" |
| 1064 | : "=r"(r) |
| 1065 | : "r"(__a), "r"(0), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1066 | return r; |
| 1067 | } |
| 1068 | __DEVICE__ unsigned int __vabsdiffs2(unsigned int __a, unsigned int __b) { |
| 1069 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1070 | __asm__("vabsdiff2.s32.s32.s32 %0,%1,%2,%3;" |
| 1071 | : "=r"(r) |
| 1072 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1073 | return r; |
| 1074 | } |
| 1075 | |
| 1076 | __DEVICE__ unsigned int __vabsdiffs4(unsigned int __a, unsigned int __b) { |
| 1077 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1078 | __asm__("vabsdiff4.s32.s32.s32 %0,%1,%2,%3;" |
| 1079 | : "=r"(r) |
| 1080 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1081 | return r; |
| 1082 | } |
| 1083 | __DEVICE__ unsigned int __vabsdiffu2(unsigned int __a, unsigned int __b) { |
| 1084 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1085 | __asm__("vabsdiff2.u32.u32.u32 %0,%1,%2,%3;" |
| 1086 | : "=r"(r) |
| 1087 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1088 | return r; |
| 1089 | } |
| 1090 | __DEVICE__ unsigned int __vabsdiffu4(unsigned int __a, unsigned int __b) { |
| 1091 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1092 | __asm__("vabsdiff4.u32.u32.u32 %0,%1,%2,%3;" |
| 1093 | : "=r"(r) |
| 1094 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1095 | return r; |
| 1096 | } |
| 1097 | __DEVICE__ unsigned int __vabsss2(unsigned int __a) { |
| 1098 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1099 | __asm__("vabsdiff2.s32.s32.s32.sat %0,%1,%2,%3;" |
| 1100 | : "=r"(r) |
| 1101 | : "r"(__a), "r"(0), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1102 | return r; |
| 1103 | } |
| 1104 | __DEVICE__ unsigned int __vabsss4(unsigned int __a) { |
| 1105 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1106 | __asm__("vabsdiff4.s32.s32.s32.sat %0,%1,%2,%3;" |
| 1107 | : "=r"(r) |
| 1108 | : "r"(__a), "r"(0), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1109 | return r; |
| 1110 | } |
| 1111 | __DEVICE__ unsigned int __vadd2(unsigned int __a, unsigned int __b) { |
| 1112 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1113 | __asm__("vadd2.u32.u32.u32 %0,%1,%2,%3;" |
| 1114 | : "=r"(r) |
| 1115 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1116 | return r; |
| 1117 | } |
| 1118 | __DEVICE__ unsigned int __vadd4(unsigned int __a, unsigned int __b) { |
| 1119 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1120 | __asm__("vadd4.u32.u32.u32 %0,%1,%2,%3;" |
| 1121 | : "=r"(r) |
| 1122 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1123 | return r; |
| 1124 | } |
| 1125 | __DEVICE__ unsigned int __vaddss2(unsigned int __a, unsigned int __b) { |
| 1126 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1127 | __asm__("vadd2.s32.s32.s32.sat %0,%1,%2,%3;" |
| 1128 | : "=r"(r) |
| 1129 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1130 | return r; |
| 1131 | } |
| 1132 | __DEVICE__ unsigned int __vaddss4(unsigned int __a, unsigned int __b) { |
| 1133 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1134 | __asm__("vadd4.s32.s32.s32.sat %0,%1,%2,%3;" |
| 1135 | : "=r"(r) |
| 1136 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1137 | return r; |
| 1138 | } |
| 1139 | __DEVICE__ unsigned int __vaddus2(unsigned int __a, unsigned int __b) { |
| 1140 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1141 | __asm__("vadd2.u32.u32.u32.sat %0,%1,%2,%3;" |
| 1142 | : "=r"(r) |
| 1143 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1144 | return r; |
| 1145 | } |
| 1146 | __DEVICE__ unsigned int __vaddus4(unsigned int __a, unsigned int __b) { |
| 1147 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1148 | __asm__("vadd4.u32.u32.u32.sat %0,%1,%2,%3;" |
| 1149 | : "=r"(r) |
| 1150 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1151 | return r; |
| 1152 | } |
| 1153 | __DEVICE__ unsigned int __vavgs2(unsigned int __a, unsigned int __b) { |
| 1154 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1155 | __asm__("vavrg2.s32.s32.s32 %0,%1,%2,%3;" |
| 1156 | : "=r"(r) |
| 1157 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1158 | return r; |
| 1159 | } |
| 1160 | __DEVICE__ unsigned int __vavgs4(unsigned int __a, unsigned int __b) { |
| 1161 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1162 | __asm__("vavrg4.s32.s32.s32 %0,%1,%2,%3;" |
| 1163 | : "=r"(r) |
| 1164 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1165 | return r; |
| 1166 | } |
| 1167 | __DEVICE__ unsigned int __vavgu2(unsigned int __a, unsigned int __b) { |
| 1168 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1169 | __asm__("vavrg2.u32.u32.u32 %0,%1,%2,%3;" |
| 1170 | : "=r"(r) |
| 1171 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1172 | return r; |
| 1173 | } |
| 1174 | __DEVICE__ unsigned int __vavgu4(unsigned int __a, unsigned int __b) { |
| 1175 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1176 | __asm__("vavrg4.u32.u32.u32 %0,%1,%2,%3;" |
| 1177 | : "=r"(r) |
| 1178 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1179 | return r; |
| 1180 | } |
| 1181 | __DEVICE__ unsigned int __vseteq2(unsigned int __a, unsigned int __b) { |
| 1182 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1183 | __asm__("vset2.u32.u32.eq %0,%1,%2,%3;" |
| 1184 | : "=r"(r) |
| 1185 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1186 | return r; |
| 1187 | } |
| 1188 | __DEVICE__ unsigned int __vcmpeq2(unsigned int __a, unsigned int __b) { |
| 1189 | return __bool2mask(__vseteq2(__a, __b), 16); |
| 1190 | } |
| 1191 | __DEVICE__ unsigned int __vseteq4(unsigned int __a, unsigned int __b) { |
| 1192 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1193 | __asm__("vset4.u32.u32.eq %0,%1,%2,%3;" |
| 1194 | : "=r"(r) |
| 1195 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1196 | return r; |
| 1197 | } |
| 1198 | __DEVICE__ unsigned int __vcmpeq4(unsigned int __a, unsigned int __b) { |
| 1199 | return __bool2mask(__vseteq4(__a, __b), 8); |
| 1200 | } |
| 1201 | __DEVICE__ unsigned int __vsetges2(unsigned int __a, unsigned int __b) { |
| 1202 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1203 | __asm__("vset2.s32.s32.ge %0,%1,%2,%3;" |
| 1204 | : "=r"(r) |
| 1205 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1206 | return r; |
| 1207 | } |
| 1208 | __DEVICE__ unsigned int __vcmpges2(unsigned int __a, unsigned int __b) { |
| 1209 | return __bool2mask(__vsetges2(__a, __b), 16); |
| 1210 | } |
| 1211 | __DEVICE__ unsigned int __vsetges4(unsigned int __a, unsigned int __b) { |
| 1212 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1213 | __asm__("vset4.s32.s32.ge %0,%1,%2,%3;" |
| 1214 | : "=r"(r) |
| 1215 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1216 | return r; |
| 1217 | } |
| 1218 | __DEVICE__ unsigned int __vcmpges4(unsigned int __a, unsigned int __b) { |
| 1219 | return __bool2mask(__vsetges4(__a, __b), 8); |
| 1220 | } |
| 1221 | __DEVICE__ unsigned int __vsetgeu2(unsigned int __a, unsigned int __b) { |
| 1222 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1223 | __asm__("vset2.u32.u32.ge %0,%1,%2,%3;" |
| 1224 | : "=r"(r) |
| 1225 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1226 | return r; |
| 1227 | } |
| 1228 | __DEVICE__ unsigned int __vcmpgeu2(unsigned int __a, unsigned int __b) { |
| 1229 | return __bool2mask(__vsetgeu2(__a, __b), 16); |
| 1230 | } |
| 1231 | __DEVICE__ unsigned int __vsetgeu4(unsigned int __a, unsigned int __b) { |
| 1232 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1233 | __asm__("vset4.u32.u32.ge %0,%1,%2,%3;" |
| 1234 | : "=r"(r) |
| 1235 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1236 | return r; |
| 1237 | } |
| 1238 | __DEVICE__ unsigned int __vcmpgeu4(unsigned int __a, unsigned int __b) { |
| 1239 | return __bool2mask(__vsetgeu4(__a, __b), 8); |
| 1240 | } |
| 1241 | __DEVICE__ unsigned int __vsetgts2(unsigned int __a, unsigned int __b) { |
| 1242 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1243 | __asm__("vset2.s32.s32.gt %0,%1,%2,%3;" |
| 1244 | : "=r"(r) |
| 1245 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1246 | return r; |
| 1247 | } |
| 1248 | __DEVICE__ unsigned int __vcmpgts2(unsigned int __a, unsigned int __b) { |
| 1249 | return __bool2mask(__vsetgts2(__a, __b), 16); |
| 1250 | } |
| 1251 | __DEVICE__ unsigned int __vsetgts4(unsigned int __a, unsigned int __b) { |
| 1252 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1253 | __asm__("vset4.s32.s32.gt %0,%1,%2,%3;" |
| 1254 | : "=r"(r) |
| 1255 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1256 | return r; |
| 1257 | } |
| 1258 | __DEVICE__ unsigned int __vcmpgts4(unsigned int __a, unsigned int __b) { |
| 1259 | return __bool2mask(__vsetgts4(__a, __b), 8); |
| 1260 | } |
| 1261 | __DEVICE__ unsigned int __vsetgtu2(unsigned int __a, unsigned int __b) { |
| 1262 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1263 | __asm__("vset2.u32.u32.gt %0,%1,%2,%3;" |
| 1264 | : "=r"(r) |
| 1265 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1266 | return r; |
| 1267 | } |
| 1268 | __DEVICE__ unsigned int __vcmpgtu2(unsigned int __a, unsigned int __b) { |
| 1269 | return __bool2mask(__vsetgtu2(__a, __b), 16); |
| 1270 | } |
| 1271 | __DEVICE__ unsigned int __vsetgtu4(unsigned int __a, unsigned int __b) { |
| 1272 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1273 | __asm__("vset4.u32.u32.gt %0,%1,%2,%3;" |
| 1274 | : "=r"(r) |
| 1275 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1276 | return r; |
| 1277 | } |
| 1278 | __DEVICE__ unsigned int __vcmpgtu4(unsigned int __a, unsigned int __b) { |
| 1279 | return __bool2mask(__vsetgtu4(__a, __b), 8); |
| 1280 | } |
| 1281 | __DEVICE__ unsigned int __vsetles2(unsigned int __a, unsigned int __b) { |
| 1282 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1283 | __asm__("vset2.s32.s32.le %0,%1,%2,%3;" |
| 1284 | : "=r"(r) |
| 1285 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1286 | return r; |
| 1287 | } |
| 1288 | __DEVICE__ unsigned int __vcmples2(unsigned int __a, unsigned int __b) { |
| 1289 | return __bool2mask(__vsetles2(__a, __b), 16); |
| 1290 | } |
| 1291 | __DEVICE__ unsigned int __vsetles4(unsigned int __a, unsigned int __b) { |
| 1292 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1293 | __asm__("vset4.s32.s32.le %0,%1,%2,%3;" |
| 1294 | : "=r"(r) |
| 1295 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1296 | return r; |
| 1297 | } |
| 1298 | __DEVICE__ unsigned int __vcmples4(unsigned int __a, unsigned int __b) { |
| 1299 | return __bool2mask(__vsetles4(__a, __b), 8); |
| 1300 | } |
| 1301 | __DEVICE__ unsigned int __vsetleu2(unsigned int __a, unsigned int __b) { |
| 1302 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1303 | __asm__("vset2.u32.u32.le %0,%1,%2,%3;" |
| 1304 | : "=r"(r) |
| 1305 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1306 | return r; |
| 1307 | } |
| 1308 | __DEVICE__ unsigned int __vcmpleu2(unsigned int __a, unsigned int __b) { |
| 1309 | return __bool2mask(__vsetleu2(__a, __b), 16); |
| 1310 | } |
| 1311 | __DEVICE__ unsigned int __vsetleu4(unsigned int __a, unsigned int __b) { |
| 1312 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1313 | __asm__("vset4.u32.u32.le %0,%1,%2,%3;" |
| 1314 | : "=r"(r) |
| 1315 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1316 | return r; |
| 1317 | } |
| 1318 | __DEVICE__ unsigned int __vcmpleu4(unsigned int __a, unsigned int __b) { |
| 1319 | return __bool2mask(__vsetleu4(__a, __b), 8); |
| 1320 | } |
| 1321 | __DEVICE__ unsigned int __vsetlts2(unsigned int __a, unsigned int __b) { |
| 1322 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1323 | __asm__("vset2.s32.s32.lt %0,%1,%2,%3;" |
| 1324 | : "=r"(r) |
| 1325 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1326 | return r; |
| 1327 | } |
| 1328 | __DEVICE__ unsigned int __vcmplts2(unsigned int __a, unsigned int __b) { |
| 1329 | return __bool2mask(__vsetlts2(__a, __b), 16); |
| 1330 | } |
| 1331 | __DEVICE__ unsigned int __vsetlts4(unsigned int __a, unsigned int __b) { |
| 1332 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1333 | __asm__("vset4.s32.s32.lt %0,%1,%2,%3;" |
| 1334 | : "=r"(r) |
| 1335 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1336 | return r; |
| 1337 | } |
| 1338 | __DEVICE__ unsigned int __vcmplts4(unsigned int __a, unsigned int __b) { |
| 1339 | return __bool2mask(__vsetlts4(__a, __b), 8); |
| 1340 | } |
| 1341 | __DEVICE__ unsigned int __vsetltu2(unsigned int __a, unsigned int __b) { |
| 1342 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1343 | __asm__("vset2.u32.u32.lt %0,%1,%2,%3;" |
| 1344 | : "=r"(r) |
| 1345 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1346 | return r; |
| 1347 | } |
| 1348 | __DEVICE__ unsigned int __vcmpltu2(unsigned int __a, unsigned int __b) { |
| 1349 | return __bool2mask(__vsetltu2(__a, __b), 16); |
| 1350 | } |
| 1351 | __DEVICE__ unsigned int __vsetltu4(unsigned int __a, unsigned int __b) { |
| 1352 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1353 | __asm__("vset4.u32.u32.lt %0,%1,%2,%3;" |
| 1354 | : "=r"(r) |
| 1355 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1356 | return r; |
| 1357 | } |
| 1358 | __DEVICE__ unsigned int __vcmpltu4(unsigned int __a, unsigned int __b) { |
| 1359 | return __bool2mask(__vsetltu4(__a, __b), 8); |
| 1360 | } |
| 1361 | __DEVICE__ unsigned int __vsetne2(unsigned int __a, unsigned int __b) { |
| 1362 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1363 | __asm__("vset2.u32.u32.ne %0,%1,%2,%3;" |
| 1364 | : "=r"(r) |
| 1365 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1366 | return r; |
| 1367 | } |
| 1368 | __DEVICE__ unsigned int __vcmpne2(unsigned int __a, unsigned int __b) { |
| 1369 | return __bool2mask(__vsetne2(__a, __b), 16); |
| 1370 | } |
| 1371 | __DEVICE__ unsigned int __vsetne4(unsigned int __a, unsigned int __b) { |
| 1372 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1373 | __asm__("vset4.u32.u32.ne %0,%1,%2,%3;" |
| 1374 | : "=r"(r) |
| 1375 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1376 | return r; |
| 1377 | } |
| 1378 | __DEVICE__ unsigned int __vcmpne4(unsigned int __a, unsigned int __b) { |
| 1379 | return __bool2mask(__vsetne4(__a, __b), 8); |
| 1380 | } |
| 1381 | |
| 1382 | // Based on ITEM 23 in AIM-239: http://dspace.mit.edu/handle/1721.1/6086 |
| 1383 | // (a & b) + (a | b) = a + b = (a ^ b) + 2 * (a & b) => |
| 1384 | // (a + b) / 2 = ((a ^ b) >> 1) + (a & b) |
| 1385 | // To operate on multiple sub-elements we need to make sure to mask out bits |
| 1386 | // that crossed over into adjacent elements during the shift. |
| 1387 | __DEVICE__ unsigned int __vhaddu2(unsigned int __a, unsigned int __b) { |
| 1388 | return (((__a ^ __b) >> 1) & ~0x80008000u) + (__a & __b); |
| 1389 | } |
| 1390 | __DEVICE__ unsigned int __vhaddu4(unsigned int __a, unsigned int __b) { |
| 1391 | return (((__a ^ __b) >> 1) & ~0x80808080u) + (__a & __b); |
| 1392 | } |
| 1393 | |
| 1394 | __DEVICE__ unsigned int __vmaxs2(unsigned int __a, unsigned int __b) { |
| 1395 | unsigned int r; |
| 1396 | if ((__a & 0x8000) && (__b & 0x8000)) { |
| 1397 | // Work around a bug in ptxas which produces invalid result if low element |
| 1398 | // is negative. |
| 1399 | unsigned mask = __vcmpgts2(__a, __b); |
| 1400 | r = (__a & mask) | (__b & ~mask); |
| 1401 | } else { |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1402 | __asm__("vmax2.s32.s32.s32 %0,%1,%2,%3;" |
| 1403 | : "=r"(r) |
| 1404 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1405 | } |
| 1406 | return r; |
| 1407 | } |
| 1408 | __DEVICE__ unsigned int __vmaxs4(unsigned int __a, unsigned int __b) { |
| 1409 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1410 | __asm__("vmax4.s32.s32.s32 %0,%1,%2,%3;" |
| 1411 | : "=r"(r) |
| 1412 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1413 | return r; |
| 1414 | } |
| 1415 | __DEVICE__ unsigned int __vmaxu2(unsigned int __a, unsigned int __b) { |
| 1416 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1417 | __asm__("vmax2.u32.u32.u32 %0,%1,%2,%3;" |
| 1418 | : "=r"(r) |
| 1419 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1420 | return r; |
| 1421 | } |
| 1422 | __DEVICE__ unsigned int __vmaxu4(unsigned int __a, unsigned int __b) { |
| 1423 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1424 | __asm__("vmax4.u32.u32.u32 %0,%1,%2,%3;" |
| 1425 | : "=r"(r) |
| 1426 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1427 | return r; |
| 1428 | } |
| 1429 | __DEVICE__ unsigned int __vmins2(unsigned int __a, unsigned int __b) { |
| 1430 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1431 | __asm__("vmin2.s32.s32.s32 %0,%1,%2,%3;" |
| 1432 | : "=r"(r) |
| 1433 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1434 | return r; |
| 1435 | } |
| 1436 | __DEVICE__ unsigned int __vmins4(unsigned int __a, unsigned int __b) { |
| 1437 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1438 | __asm__("vmin4.s32.s32.s32 %0,%1,%2,%3;" |
| 1439 | : "=r"(r) |
| 1440 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1441 | return r; |
| 1442 | } |
| 1443 | __DEVICE__ unsigned int __vminu2(unsigned int __a, unsigned int __b) { |
| 1444 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1445 | __asm__("vmin2.u32.u32.u32 %0,%1,%2,%3;" |
| 1446 | : "=r"(r) |
| 1447 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1448 | return r; |
| 1449 | } |
| 1450 | __DEVICE__ unsigned int __vminu4(unsigned int __a, unsigned int __b) { |
| 1451 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1452 | __asm__("vmin4.u32.u32.u32 %0,%1,%2,%3;" |
| 1453 | : "=r"(r) |
| 1454 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1455 | return r; |
| 1456 | } |
| 1457 | __DEVICE__ unsigned int __vsads2(unsigned int __a, unsigned int __b) { |
| 1458 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1459 | __asm__("vabsdiff2.s32.s32.s32.add %0,%1,%2,%3;" |
| 1460 | : "=r"(r) |
| 1461 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1462 | return r; |
| 1463 | } |
| 1464 | __DEVICE__ unsigned int __vsads4(unsigned int __a, unsigned int __b) { |
| 1465 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1466 | __asm__("vabsdiff4.s32.s32.s32.add %0,%1,%2,%3;" |
| 1467 | : "=r"(r) |
| 1468 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1469 | return r; |
| 1470 | } |
| 1471 | __DEVICE__ unsigned int __vsadu2(unsigned int __a, unsigned int __b) { |
| 1472 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1473 | __asm__("vabsdiff2.u32.u32.u32.add %0,%1,%2,%3;" |
| 1474 | : "=r"(r) |
| 1475 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1476 | return r; |
| 1477 | } |
| 1478 | __DEVICE__ unsigned int __vsadu4(unsigned int __a, unsigned int __b) { |
| 1479 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1480 | __asm__("vabsdiff4.u32.u32.u32.add %0,%1,%2,%3;" |
| 1481 | : "=r"(r) |
| 1482 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1483 | return r; |
| 1484 | } |
| 1485 | |
| 1486 | __DEVICE__ unsigned int __vsub2(unsigned int __a, unsigned int __b) { |
| 1487 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1488 | __asm__("vsub2.u32.u32.u32 %0,%1,%2,%3;" |
| 1489 | : "=r"(r) |
| 1490 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1491 | return r; |
| 1492 | } |
| 1493 | __DEVICE__ unsigned int __vneg2(unsigned int __a) { return __vsub2(0, __a); } |
| 1494 | |
| 1495 | __DEVICE__ unsigned int __vsub4(unsigned int __a, unsigned int __b) { |
| 1496 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1497 | __asm__("vsub4.u32.u32.u32 %0,%1,%2,%3;" |
| 1498 | : "=r"(r) |
| 1499 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1500 | return r; |
| 1501 | } |
| 1502 | __DEVICE__ unsigned int __vneg4(unsigned int __a) { return __vsub4(0, __a); } |
| 1503 | __DEVICE__ unsigned int __vsubss2(unsigned int __a, unsigned int __b) { |
| 1504 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1505 | __asm__("vsub2.s32.s32.s32.sat %0,%1,%2,%3;" |
| 1506 | : "=r"(r) |
| 1507 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1508 | return r; |
| 1509 | } |
| 1510 | __DEVICE__ unsigned int __vnegss2(unsigned int __a) { |
| 1511 | return __vsubss2(0, __a); |
| 1512 | } |
| 1513 | __DEVICE__ unsigned int __vsubss4(unsigned int __a, unsigned int __b) { |
| 1514 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1515 | __asm__("vsub4.s32.s32.s32.sat %0,%1,%2,%3;" |
| 1516 | : "=r"(r) |
| 1517 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1518 | return r; |
| 1519 | } |
| 1520 | __DEVICE__ unsigned int __vnegss4(unsigned int __a) { |
| 1521 | return __vsubss4(0, __a); |
| 1522 | } |
| 1523 | __DEVICE__ unsigned int __vsubus2(unsigned int __a, unsigned int __b) { |
| 1524 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1525 | __asm__("vsub2.u32.u32.u32.sat %0,%1,%2,%3;" |
| 1526 | : "=r"(r) |
| 1527 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1528 | return r; |
| 1529 | } |
| 1530 | __DEVICE__ unsigned int __vsubus4(unsigned int __a, unsigned int __b) { |
| 1531 | unsigned int r; |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 1532 | __asm__("vsub4.u32.u32.u32.sat %0,%1,%2,%3;" |
| 1533 | : "=r"(r) |
| 1534 | : "r"(__a), "r"(__b), "r"(0)); |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1535 | return r; |
| 1536 | } |
| 1537 | #endif // CUDA_VERSION >= 9020 |
Sasha Smundak | 0fc590b | 2020-10-07 08:11:59 -0700 | [diff] [blame] | 1538 | |
| 1539 | // For OpenMP we require the user to include <time.h> as we need to know what |
| 1540 | // clock_t is on the system. |
| 1541 | #ifndef __OPENMP_NVPTX__ |
| 1542 | __DEVICE__ /* clock_t= */ int clock() { return __nvvm_read_ptx_sreg_clock(); } |
| 1543 | #endif |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1544 | __DEVICE__ long long clock64() { return __nvvm_read_ptx_sreg_clock64(); } |
Sasha Smundak | 0fc590b | 2020-10-07 08:11:59 -0700 | [diff] [blame] | 1545 | |
Logan Chien | df4f766 | 2019-09-04 16:45:23 -0700 | [diff] [blame] | 1546 | // These functions shouldn't be declared when including this header |
| 1547 | // for math function resolution purposes. |
Sasha Smundak | 0fc590b | 2020-10-07 08:11:59 -0700 | [diff] [blame] | 1548 | #ifndef __OPENMP_NVPTX__ |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1549 | __DEVICE__ void *memcpy(void *__a, const void *__b, size_t __c) { |
| 1550 | return __builtin_memcpy(__a, __b, __c); |
| 1551 | } |
| 1552 | __DEVICE__ void *memset(void *__a, int __b, size_t __c) { |
| 1553 | return __builtin_memset(__a, __b, __c); |
| 1554 | } |
Logan Chien | df4f766 | 2019-09-04 16:45:23 -0700 | [diff] [blame] | 1555 | #endif |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1556 | |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1557 | #pragma pop_macro("__DEVICE__") |
Logan Chien | 55afb0a | 2018-10-15 10:42:14 +0800 | [diff] [blame] | 1558 | #endif // __CLANG_CUDA_DEVICE_FUNCTIONS_H__ |