"vscode:/vscode.git/clone" did not exist on "3d0ac19403b3ae9b10e2c8ad9809e752eb1dee06"
math_v2.hpp 363 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
#ifndef CK_MATH_V2_HPP
#define CK_MATH_V2_HPP

#include "data_type.hpp"

namespace ck {
namespace math {

static inline __device__ half_t abs(half_t x) { return __habs(x); };
static inline __device__ half_t sqrtf(half_t x) { return hsqrt(x); };
static inline __device__ bool isnan(half_t x) { return __hisnan(x); };

} // namespace math
} // namespace ck

#endif