32 lines
1.6 KiB
Text
32 lines
1.6 KiB
Text
|
|
#define DIVIDE(x, size) (((x) + (size) - 1) / (size))
|
|
|
|
#define DBGS(__x) printf("%s\n", __x)
|
|
#define DBGI(__x) printf("%s: %i\n", #__x, __x)
|
|
#define DBGI2(__x, __y) printf("%s, %s: %i, %i\n", #__x, #__y, __x, __y)
|
|
#define DBGI3(__x, __y, __z) printf("%s, %s, %s: %i, %i, %i\n", #__x, #__y, #__z, __x, __y, __z)
|
|
#define DBGX(__x) printf("%s: %x\n", #__x, __x)
|
|
#define DBGX2(__x, __y) printf("%s, %s: %x, %x\n", #__x, #__y, __x, __y)
|
|
#define DBGX3(__x, __y, __z) printf("%s, %s, %s: %x, %x, %x\n", #__x, #__y, #__z, __x, __y, __z)
|
|
#define DBGF(__x) printf("%s: %f\n", #__x, __x)
|
|
#define DBGF2(__x, __y) printf("%s, %s: %f, %f\n", #__x, #__y, __x, __y)
|
|
#define DBGF3(__x, __y, __z) printf("%s, %s, %s: %f, %f, %f\n", #__x, #__y, #__z, __x, __y, __z)
|
|
#define DBGH(__x) printf("%s: %f\n", #__x, __half2float(__x))
|
|
#define DBGH2(__x, __y) printf("%s, %s: %f, %f\n", #__x, #__y, __half2float(__x), __half2float(__y))
|
|
#define DBGH3(__x, __y, __z) printf("%s, %s, %s: %f, %f, %f\n", #__x, #__y, #__z, __half2float(__x), __half2float(__y), __half2float(__z))
|
|
|
|
#define DBGIH(__x, __y) printf("%s, %s: %i, %f\n", #__x, #__y, __x, __half2float(__y))
|
|
#define DBGIH2(__x, __y, __z) printf("%s, %s, %s: %i, %f, %f\n", #__x, #__y, #__z, __x, __half2float(__y), __half2float(__z))
|
|
|
|
__forceinline__ __device__ half dq_scale_(const int qs, const half max_scale)
|
|
{
|
|
half qs_h = __hmul(__int2half_rn(qs + 1), __float2half_rn(1.0f / 16.0f));
|
|
qs_h = __hmul(qs_h, qs_h);
|
|
qs_h = __hmul(qs_h, max_scale);
|
|
return qs_h;
|
|
}
|
|
|
|
__forceinline__ __device__ float clamp(float x, float a, float b)
|
|
{
|
|
return fmaxf(a, fminf(b, x));
|
|
}
|