Aditya Atluri
182 linhas
4.8 KiB
C
182 linhas
4.8 KiB
C
/*
|
|
Copyright (c) 2015-2016 Advanced Micro Devices, Inc. All rights reserved.
|
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
of this software and associated documentation files (the "Software"), to deal
|
|
in the Software without restriction, including without limitation the rights
|
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
copies of the Software, and to permit persons to whom the Software is
|
|
furnished to do so, subject to the following conditions:
|
|
The above copyright notice and this permission notice shall be included in
|
|
all copies or substantial portions of the Software.
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANNTY OF ANY KIND, EXPRESS OR
|
|
IMPLIED, INNCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
FITNNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANNY CLAIM, DAMAGES OR OTHER
|
|
LIABILITY, WHETHER INN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
OUT OF OR INN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
THE SOFTWARE.
|
|
*/
|
|
|
|
#ifndef HIP_FP16_H
|
|
#define HIP_FP16_H
|
|
|
|
#include "hip_runtime.h"
|
|
|
|
typedef struct{
|
|
unsigned x: 16;
|
|
} __half;
|
|
|
|
|
|
typedef struct __attribute__((aligned(4))){
|
|
__half p,q;
|
|
} __half2;
|
|
|
|
typedef __half half;
|
|
typedef __half2 half2;
|
|
|
|
typedef struct{
|
|
union{
|
|
float f;
|
|
unsigned u;
|
|
};
|
|
} struct_float;
|
|
|
|
/*
|
|
Arithmetic functions
|
|
*/
|
|
|
|
__device__ __half __hadd(const __half a, const __half b);
|
|
|
|
__device__ __half __hadd_sat(const __half a, const __half b);
|
|
|
|
__device__ __half __hfma(const __half a, const __half b, const __half c);
|
|
|
|
__device__ __half __hfma_sat(const __half a, const __half b, const __half c);
|
|
|
|
__device__ __half __hmul(const __half a, const __half b);
|
|
|
|
__device__ __half __hmul_sat(const __half a, const __half b);
|
|
|
|
__device__ __half __hneq(const __half a);
|
|
|
|
__device__ __half __hsub(const __half a, const __half b);
|
|
|
|
__device__ __half __hsub_sat(const __half a, const __half b);
|
|
|
|
|
|
|
|
/*
|
|
Half2 Arithmetic Instructions
|
|
*/
|
|
|
|
__device__ __half2 __hadd2(const __half2 a, const __half2 b);
|
|
|
|
__device__ __half2 __hadd2_sat(const __half2 a, const __half2 b);
|
|
|
|
__device__ __half2 __hfma2(const __half2 a, const __half2 b, const __half2 c);
|
|
|
|
__device__ __half2 __hfma2_sat(const __half2 a, const __half2 b, const __half2 c);
|
|
|
|
__device__ __half2 __hmul2(const __half2 a, const __half2 b);
|
|
|
|
__device__ __half2 __hmul2_sat(const __half2 a, const __half2 b);
|
|
|
|
__device__ __half2 __hneq2(const __half2 a);
|
|
|
|
__device__ __half2 __hsub2(const __half2 a, const __half2 b);
|
|
|
|
__device__ __half2 __hsub2_sat(const __half2 a, const __half2 b);
|
|
|
|
/*
|
|
Half Cmps
|
|
*/
|
|
|
|
__device__ bool __heq(const __half a, const __half b);
|
|
|
|
__device__ bool __hge(const __half a, const __half b);
|
|
|
|
__device__ bool __hgt(const __half a, const __half b);
|
|
|
|
__device__ bool __hisinf(const __half a);
|
|
|
|
__device__ bool __hisnan(const __half a);
|
|
|
|
__device__ bool __hle(const __half a, const __half b);
|
|
|
|
__device__ bool __hlt(const __half a, const __half b);
|
|
|
|
__device__ bool __hne(const __half a, const __half b);
|
|
|
|
/*
|
|
Half2 Cmps
|
|
*/
|
|
|
|
__device__ bool __hbeq2(const __half2 a, const __half2 b);
|
|
|
|
__device__ bool __hbge2(const __half2 a, const __half2 b);
|
|
|
|
__device__ bool __hbgt2(const __half2 a, const __half2 b);
|
|
|
|
__device__ bool __hble2(const __half2 a, const __half2 b);
|
|
|
|
__device__ bool __hblt2(const __half2 a, const __half2 b);
|
|
|
|
__device__ bool __hbne2(const __half2 a, const __half2 b);
|
|
|
|
__device__ __half2 __heq2(const __half2 a, const __half2 b);
|
|
|
|
__device__ __half2 __hge2(const __half2 a, const __half2 b);
|
|
|
|
__device__ __half2 __hgt2(const __half2 a, const __half2 b);
|
|
|
|
__device__ __half2 __hisnan2(const __half2 a);
|
|
|
|
__device__ __half2 __hle2(const __half2 a, const __half2 b);
|
|
|
|
__device__ __half2 __hlt2(const __half2 a, const __half2 b);
|
|
|
|
__device__ __half2 __hne2(const __half2 a, const __half2 b);
|
|
|
|
|
|
/*
|
|
Half Cnvs and Data Mvmnt
|
|
*/
|
|
|
|
__device__ __half2 __float22half2_rn(const float2 a);
|
|
|
|
__device__ __half __float2half(const float a);
|
|
|
|
__device__ __half2 __float2half2_rn(const float a);
|
|
|
|
__device__ __half2 __floats2half2_rn(const float a, const float b);
|
|
|
|
__device__ float2 __half22float2(const __half2 a);
|
|
|
|
__device__ float __half2float(const __half a);
|
|
|
|
__device__ __half2 __half2half2(const __half a);
|
|
|
|
__device__ __half2 __halves2half2(const __half a, const __half b);
|
|
|
|
__device__ float __high2float(const __half2 a);
|
|
|
|
__device__ __half __high2half(const __half2 a);
|
|
|
|
__device__ __half2 __high2half2(const __half2 a);
|
|
|
|
__device__ __half2 __highs2half2(const __half2 a, const __half2 b);
|
|
|
|
__device__ float __low2float(const __half2 a);
|
|
|
|
__device__ __half __low2half(const __half2 a);
|
|
|
|
__device__ __half2 __low2half2(const __half2 a);
|
|
|
|
__device__ __half2 __lows2half2(const __half2 a, const __half2 b);
|
|
|
|
__device__ __half2 __lowhigh2highlow(const __half2 a);
|
|
|
|
__device__ __half2 __low2half2(const __half2 a, const __half2 b);
|
|
|
|
#endif
|