AMDGPULibCalls.cpp File Reference

This file does AMD library function optimizations. More...

#include "AMDGPU.h"
#include "AMDGPULibFunc.h"
#include "AMDGPUSubtarget.h"
#include "llvm/Analysis/AssumptionCache.h"
#include "llvm/Analysis/TargetLibraryInfo.h"
#include "llvm/Analysis/ValueTracking.h"
#include "llvm/IR/AttributeMask.h"
#include "llvm/IR/Dominators.h"
#include "llvm/IR/IRBuilder.h"
#include "llvm/IR/MDBuilder.h"
#include "llvm/IR/PatternMatch.h"
#include <cmath>

Go to the source code of this file.

Classes
class	llvm::AMDGPULibCalls
struct	TableEntry

Namespaces
namespace	llvm
	This is an optimization pass for GlobalISel generic memory operations.

Macros
#define	DEBUG_TYPE   "amdgpu-simplifylib"
#define	MATH_PI   numbers::pi
#define	MATH_E   numbers::e
#define	MATH_SQRT2   numbers::sqrt2
#define	MATH_SQRT1_2   numbers::inv_sqrt2

Typedefs
using	TableRef = ArrayRef<TableEntry>

Enumerations
enum class	PowKind { Pow , PowR , PowN , RootN }

Functions
template<typename IRB>
static CallInst *	CreateCallEx (IRB &B, FunctionCallee Callee, Value *Arg, const Twine &Name="")
template<typename IRB>
static CallInst *	CreateCallEx2 (IRB &B, FunctionCallee Callee, Value *Arg1, Value *Arg2, const Twine &Name="")
static FunctionType *	getPownType (FunctionType *FT)
static bool	HasNative (AMDGPULibFunc::EFuncId id)
static TableRef	getOptTable (AMDGPULibFunc::EFuncId id)
static int	getVecSize (const AMDGPULibFunc &FInfo)
static AMDGPULibFunc::EType	getArgType (const AMDGPULibFunc &FInfo)
static Constant *	getConstantFloat (const ArrayRef< APFloat > Values, const Type *Ty)
static double	llvm::log2 (double V)
static Value *	emitIsInteger (IRBuilder<> &B, Value *Y)
static Value *	emitIsEvenInteger (IRBuilder<> &B, Value *Y)
static Value *	emitIsOddInteger (IRBuilder<> &B, Value *Y)
static Value *	emitIsInf (IRBuilder<> &B, Value *val)
static Value *	emitFastExpYLnx (IRBuilder<> &B, Value *X, Value *Y)
static Value *	emitPowFixup (IRBuilder<> &B, Value *X, Value *Y, Value *ExpYLnX, PowKind Kind)
	Emit special case management epilog code for fast pow, powr, pown, and rootn expansions.

Variables
static cl::list< std::string >	UseNative ("amdgpu-use-native", cl::desc("Comma separated list of functions to replace with native, or all"), cl::CommaSeparated, cl::ValueOptional, cl::Hidden)
static const TableEntry	tbl_acos []
static const TableEntry	tbl_acosh []
static const TableEntry	tbl_acospi []
static const TableEntry	tbl_asin []
static const TableEntry	tbl_asinh []
static const TableEntry	tbl_asinpi []
static const TableEntry	tbl_atan []
static const TableEntry	tbl_atanh []
static const TableEntry	tbl_atanpi []
static const TableEntry	tbl_cbrt []
static const TableEntry	tbl_cos []
static const TableEntry	tbl_cosh []
static const TableEntry	tbl_cospi []
static const TableEntry	tbl_erfc []
static const TableEntry	tbl_erf []
static const TableEntry	tbl_exp []
static const TableEntry	tbl_exp2 []
static const TableEntry	tbl_exp10 []
static const TableEntry	tbl_expm1 []
static const TableEntry	tbl_log []
static const TableEntry	tbl_log2 []
static const TableEntry	tbl_log10 []
static const TableEntry	tbl_rsqrt []
static const TableEntry	tbl_sin []
static const TableEntry	tbl_sinh []
static const TableEntry	tbl_sinpi []
static const TableEntry	tbl_sqrt []
static const TableEntry	tbl_tan []
static const TableEntry	tbl_tanh []
static const TableEntry	tbl_tanpi []
static const TableEntry	tbl_tgamma []

Detailed Description

This file does AMD library function optimizations.

Definition in file AMDGPULibCalls.cpp.

Macro Definition Documentation

DEBUG_TYPE

#define DEBUG_TYPE   "amdgpu-simplifylib"

Definition at line 27 of file AMDGPULibCalls.cpp.

MATH_E

#define MATH_E   numbers::e

Definition at line 38 of file AMDGPULibCalls.cpp.

MATH_PI

#define MATH_PI   numbers::pi

Definition at line 37 of file AMDGPULibCalls.cpp.

MATH_SQRT1_2

#define MATH_SQRT1_2   numbers::inv_sqrt2

Definition at line 40 of file AMDGPULibCalls.cpp.

MATH_SQRT2

#define MATH_SQRT2   numbers::sqrt2

Definition at line 39 of file AMDGPULibCalls.cpp.

Typedef Documentation

TableRef

using TableRef = ArrayRef<TableEntry>

Definition at line 360 of file AMDGPULibCalls.cpp.

Enumeration Type Documentation

PowKind

enum class PowKind

strong

Enumerator
Pow
PowR
PowN
RootN

Definition at line 42 of file AMDGPULibCalls.cpp.

Function Documentation

CreateCallEx()

template<typename IRB>

CallInst * CreateCallEx ( IRB & B, FunctionCallee Callee, Value * Arg, const Twine & Name = "" )

static

Definition at line 158 of file AMDGPULibCalls.cpp.

References B(), llvm::dyn_cast(), and F.

CreateCallEx2()

template<typename IRB>

CallInst * CreateCallEx2 ( IRB & B, FunctionCallee Callee, Value * Arg1, Value * Arg2, const Twine & Name = "" )

static

Definition at line 167 of file AMDGPULibCalls.cpp.

References B(), llvm::dyn_cast(), and F.

emitFastExpYLnx()

Value * emitFastExpYLnx ( IRBuilder<> & B, Value * X, Value * Y )

static

Definition at line 1306 of file AMDGPULibCalls.cpp.

References B(), X, and Y.

emitIsEvenInteger()

Value * emitIsEvenInteger ( IRBuilder<> & B, Value * Y )

static

Definition at line 1285 of file AMDGPULibCalls.cpp.

References B(), emitIsInteger(), and Y.

Referenced by emitIsOddInteger().

emitIsInf()

Value * emitIsInf ( IRBuilder<> & B, Value * val )

static

Definition at line 1300 of file AMDGPULibCalls.cpp.

References B(), llvm::ConstantFP::getInfinity(), and llvm::Value::getType().

Referenced by emitPowFixup().

emitIsInteger()

Value * emitIsInteger ( IRBuilder<> & B, Value * Y )

static

Definition at line 1280 of file AMDGPULibCalls.cpp.

References B(), and Y.

Referenced by emitIsEvenInteger(), emitIsOddInteger(), and emitPowFixup().

emitIsOddInteger()

Value * emitIsOddInteger ( IRBuilder<> & B, Value * Y )

static

Definition at line 1292 of file AMDGPULibCalls.cpp.

References B(), emitIsEvenInteger(), emitIsInteger(), and Y.

Referenced by emitPowFixup().

emitPowFixup()

Value * emitPowFixup ( IRBuilder<> & B, Value * X, Value * Y, Value * ExpYLnX, PowKind Kind )

static

Emit special case management epilog code for fast pow, powr, pown, and rootn expansions.

x and y should be the arguments to the library call (possibly with some values clamped). expylnx should be the result to use in normal circumstances.

Definition at line 1317 of file AMDGPULibCalls.cpp.

References B(), emitIsInf(), emitIsInteger(), emitIsOddInteger(), llvm::ConstantFP::getInfinity(), llvm::ConstantFP::getQNaN(), llvm::ConstantFP::getZero(), llvm_unreachable, Pow, PowN, PowR, RootN, X, Y, and Zero.

getArgType()

AMDGPULibFunc::EType getArgType ( const AMDGPULibFunc & FInfo )

inlinestatic

Definition at line 410 of file AMDGPULibCalls.cpp.

References llvm::AMDGPULibFuncBase::Param::ArgType, and llvm::AMDGPULibFunc::getLeads().

Referenced by llvm::AMDGPULibCalls::fold(), and llvm::AMDGPULibCalls::useNative().

getConstantFloat()

Constant * getConstantFloat ( const ArrayRef< APFloat > Values, const Type * Ty )

static

Definition at line 798 of file AMDGPULibCalls.cpp.

References assert(), llvm::cast(), llvm::ConstantVector::get(), llvm::SmallVectorTemplateBase< T, bool >::push_back(), llvm::SmallVectorImpl< T >::reserve(), llvm::APFloatBase::rmNearestTiesToEven, and llvm::Values.

getOptTable()

TableRef getOptTable ( AMDGPULibFunc::EFuncId id )

static

Definition at line 362 of file AMDGPULibCalls.cpp.

References llvm::AMDGPULibFuncBase::EI_ACOS, llvm::AMDGPULibFuncBase::EI_ACOSH, llvm::AMDGPULibFuncBase::EI_ACOSPI, llvm::AMDGPULibFuncBase::EI_ASIN, llvm::AMDGPULibFuncBase::EI_ASINH, llvm::AMDGPULibFuncBase::EI_ASINPI, llvm::AMDGPULibFuncBase::EI_ATAN, llvm::AMDGPULibFuncBase::EI_ATANH, llvm::AMDGPULibFuncBase::EI_ATANPI, llvm::AMDGPULibFuncBase::EI_CBRT, llvm::AMDGPULibFuncBase::EI_COS, llvm::AMDGPULibFuncBase::EI_COSH, llvm::AMDGPULibFuncBase::EI_COSPI, llvm::AMDGPULibFuncBase::EI_ERF, llvm::AMDGPULibFuncBase::EI_ERFC, llvm::AMDGPULibFuncBase::EI_EXP, llvm::AMDGPULibFuncBase::EI_EXP10, llvm::AMDGPULibFuncBase::EI_EXP2, llvm::AMDGPULibFuncBase::EI_EXPM1, llvm::AMDGPULibFuncBase::EI_LOG, llvm::AMDGPULibFuncBase::EI_LOG10, llvm::AMDGPULibFuncBase::EI_LOG2, llvm::AMDGPULibFuncBase::EI_NCOS, llvm::AMDGPULibFuncBase::EI_NEXP2, llvm::AMDGPULibFuncBase::EI_NLOG2, llvm::AMDGPULibFuncBase::EI_NRSQRT, llvm::AMDGPULibFuncBase::EI_NSIN, llvm::AMDGPULibFuncBase::EI_NSQRT, llvm::AMDGPULibFuncBase::EI_RSQRT, llvm::AMDGPULibFuncBase::EI_SIN, llvm::AMDGPULibFuncBase::EI_SINH, llvm::AMDGPULibFuncBase::EI_SINPI, llvm::AMDGPULibFuncBase::EI_SQRT, llvm::AMDGPULibFuncBase::EI_TAN, llvm::AMDGPULibFuncBase::EI_TANH, llvm::AMDGPULibFuncBase::EI_TANPI, llvm::AMDGPULibFuncBase::EI_TGAMMA, tbl_acos, tbl_acosh, tbl_acospi, tbl_asin, tbl_asinh, tbl_asinpi, tbl_atan, tbl_atanh, tbl_atanpi, tbl_cbrt, tbl_cos, tbl_cosh, tbl_cospi, tbl_erf, tbl_erfc, tbl_exp, tbl_exp10, tbl_exp2, tbl_expm1, tbl_log, tbl_log10, tbl_log2, tbl_rsqrt, tbl_sin, tbl_sinh, tbl_sinpi, tbl_sqrt, tbl_tan, tbl_tanh, tbl_tanpi, and tbl_tgamma.

getPownType()

FunctionType * getPownType ( FunctionType * FT )

static

Definition at line 175 of file AMDGPULibCalls.cpp.

References llvm::dyn_cast(), llvm::FunctionType::get(), llvm::VectorType::get(), and llvm::Type::getInt32Ty().

getVecSize()

int getVecSize ( const AMDGPULibFunc & FInfo )

inlinestatic

Definition at line 406 of file AMDGPULibCalls.cpp.

References llvm::AMDGPULibFunc::getLeads(), and llvm::AMDGPULibFuncBase::Param::VectorSize.

HasNative()

bool HasNative ( AMDGPULibFunc::EFuncId id )

static

Definition at line 337 of file AMDGPULibCalls.cpp.

References llvm::AMDGPULibFuncBase::EI_COS, llvm::AMDGPULibFuncBase::EI_DIVIDE, llvm::AMDGPULibFuncBase::EI_EXP, llvm::AMDGPULibFuncBase::EI_EXP10, llvm::AMDGPULibFuncBase::EI_EXP2, llvm::AMDGPULibFuncBase::EI_LOG, llvm::AMDGPULibFuncBase::EI_LOG10, llvm::AMDGPULibFuncBase::EI_LOG2, llvm::AMDGPULibFuncBase::EI_POWR, llvm::AMDGPULibFuncBase::EI_RECIP, llvm::AMDGPULibFuncBase::EI_RSQRT, llvm::AMDGPULibFuncBase::EI_SIN, llvm::AMDGPULibFuncBase::EI_SINCOS, llvm::AMDGPULibFuncBase::EI_SQRT, and llvm::AMDGPULibFuncBase::EI_TAN.

Referenced by llvm::AMDGPULibCalls::useNative().

Variable Documentation

tbl_acos

const TableEntry tbl_acos[]

static

Initial value:

= {
  {MATH_PI / 2.0, 0.0},
  {MATH_PI / 2.0, -0.0},
  {0.0, 1.0},
  {MATH_PI, -1.0}
}

Definition at line 193 of file AMDGPULibCalls.cpp.

Referenced by getOptTable().

tbl_acosh

const TableEntry tbl_acosh[]

static

Initial value:

= {
  {0.0, 1.0}
}

Definition at line 199 of file AMDGPULibCalls.cpp.

Referenced by getOptTable().

tbl_acospi

const TableEntry tbl_acospi[]

static

Initial value:

= {
  {0.5, 0.0},
  {0.5, -0.0},
  {0.0, 1.0},
  {1.0, -1.0}
}

Definition at line 202 of file AMDGPULibCalls.cpp.

Referenced by getOptTable().

tbl_asin

const TableEntry tbl_asin[]

static

Initial value:

= {
  {0.0, 0.0},
  {-0.0, -0.0},
  {MATH_PI / 2.0, 1.0},
  {-MATH_PI / 2.0, -1.0}
}

Definition at line 208 of file AMDGPULibCalls.cpp.

Referenced by getOptTable().

tbl_asinh

const TableEntry tbl_asinh[]

static

Initial value:

= {
  {0.0, 0.0},
  {-0.0, -0.0}
}

Definition at line 214 of file AMDGPULibCalls.cpp.

Referenced by getOptTable().

tbl_asinpi

const TableEntry tbl_asinpi[]

static

Initial value:

= {
  {0.0, 0.0},
  {-0.0, -0.0},
  {0.5, 1.0},
  {-0.5, -1.0}
}

Definition at line 218 of file AMDGPULibCalls.cpp.

Referenced by getOptTable().

tbl_atan

const TableEntry tbl_atan[]

static

Initial value:

= {
  {0.0, 0.0},
  {-0.0, -0.0},
  {MATH_PI / 4.0, 1.0},
  {-MATH_PI / 4.0, -1.0}
}

Definition at line 224 of file AMDGPULibCalls.cpp.

Referenced by getOptTable().

tbl_atanh

const TableEntry tbl_atanh[]

static

Initial value:

= {
  {0.0, 0.0},
  {-0.0, -0.0}
}

Definition at line 230 of file AMDGPULibCalls.cpp.

Referenced by getOptTable().

tbl_atanpi

const TableEntry tbl_atanpi[]

static

Initial value:

= {
  {0.0, 0.0},
  {-0.0, -0.0},
  {0.25, 1.0},
  {-0.25, -1.0}
}

Definition at line 234 of file AMDGPULibCalls.cpp.

Referenced by getOptTable().

tbl_cbrt

const TableEntry tbl_cbrt[]

static

Initial value:

= {
  {0.0, 0.0},
  {-0.0, -0.0},
  {1.0, 1.0},
  {-1.0, -1.0},
}

Definition at line 240 of file AMDGPULibCalls.cpp.

Referenced by getOptTable().

tbl_cos

const TableEntry tbl_cos[]

static

Initial value:

= {
  {1.0, 0.0},
  {1.0, -0.0}
}

Definition at line 246 of file AMDGPULibCalls.cpp.

Referenced by getOptTable().

tbl_cosh

const TableEntry tbl_cosh[]

static

Initial value:

= {
  {1.0, 0.0},
  {1.0, -0.0}
}

Definition at line 250 of file AMDGPULibCalls.cpp.

Referenced by getOptTable().

tbl_cospi

const TableEntry tbl_cospi[]

static

Initial value:

= {
  {1.0, 0.0},
  {1.0, -0.0}
}

Definition at line 254 of file AMDGPULibCalls.cpp.

Referenced by getOptTable().

tbl_erf

const TableEntry tbl_erf[]

static

Initial value:

= {
  {0.0, 0.0},
  {-0.0, -0.0}
}

Definition at line 262 of file AMDGPULibCalls.cpp.

Referenced by getOptTable().

tbl_erfc

const TableEntry tbl_erfc[]

static

Initial value:

= {
  {1.0, 0.0},
  {1.0, -0.0}
}

Definition at line 258 of file AMDGPULibCalls.cpp.

Referenced by getOptTable().

tbl_exp

const TableEntry tbl_exp[]

static

Initial value:

= {
  {1.0, 0.0},
  {1.0, -0.0},
  {MATH_E, 1.0}
}

Definition at line 266 of file AMDGPULibCalls.cpp.

Referenced by getOptTable().

tbl_exp10

const TableEntry tbl_exp10[]

static

Initial value:

= {
  {1.0, 0.0},
  {1.0, -0.0},
  {10.0, 1.0}
}

Definition at line 276 of file AMDGPULibCalls.cpp.

Referenced by getOptTable().

tbl_exp2

const TableEntry tbl_exp2[]

static

Initial value:

= {
  {1.0, 0.0},
  {1.0, -0.0},
  {2.0, 1.0}
}

Definition at line 271 of file AMDGPULibCalls.cpp.

Referenced by getOptTable().

tbl_expm1

const TableEntry tbl_expm1[]

static

Initial value:

= {
  {0.0, 0.0},
  {-0.0, -0.0}
}

Definition at line 281 of file AMDGPULibCalls.cpp.

Referenced by getOptTable().

tbl_log

const TableEntry tbl_log[]

static

Initial value:

= {
  {0.0, 1.0},
  {1.0, MATH_E}
}

Definition at line 285 of file AMDGPULibCalls.cpp.

Referenced by getOptTable().

tbl_log10

const TableEntry tbl_log10[]

static

Initial value:

= {
  {0.0, 1.0},
  {1.0, 10.0}
}

Definition at line 293 of file AMDGPULibCalls.cpp.

Referenced by getOptTable().

tbl_log2

const TableEntry tbl_log2[]

static

Initial value:

= {
  {0.0, 1.0},
  {1.0, 2.0}
}

Definition at line 289 of file AMDGPULibCalls.cpp.

Referenced by getOptTable().

tbl_rsqrt

const TableEntry tbl_rsqrt[]

static

Initial value:

= {
  {1.0, 1.0},
  {MATH_SQRT1_2, 2.0}
}

Definition at line 297 of file AMDGPULibCalls.cpp.

Referenced by getOptTable().

tbl_sin

const TableEntry tbl_sin[]

static

Initial value:

= {
  {0.0, 0.0},
  {-0.0, -0.0}
}

Definition at line 301 of file AMDGPULibCalls.cpp.

Referenced by getOptTable().

tbl_sinh

const TableEntry tbl_sinh[]

static

Initial value:

= {
  {0.0, 0.0},
  {-0.0, -0.0}
}

Definition at line 305 of file AMDGPULibCalls.cpp.

Referenced by getOptTable().

tbl_sinpi

const TableEntry tbl_sinpi[]

static

Initial value:

= {
  {0.0, 0.0},
  {-0.0, -0.0}
}

Definition at line 309 of file AMDGPULibCalls.cpp.

Referenced by getOptTable().

tbl_sqrt

const TableEntry tbl_sqrt[]

static

Initial value:

= {
  {0.0, 0.0},
  {1.0, 1.0},
  {MATH_SQRT2, 2.0}
}

Definition at line 313 of file AMDGPULibCalls.cpp.

Referenced by getOptTable().

tbl_tan

const TableEntry tbl_tan[]

static

Initial value:

= {
  {0.0, 0.0},
  {-0.0, -0.0}
}

Definition at line 318 of file AMDGPULibCalls.cpp.

Referenced by getOptTable().

tbl_tanh

const TableEntry tbl_tanh[]

static

Initial value:

= {
  {0.0, 0.0},
  {-0.0, -0.0}
}

Definition at line 322 of file AMDGPULibCalls.cpp.

Referenced by getOptTable().

tbl_tanpi

const TableEntry tbl_tanpi[]

static

Initial value:

= {
  {0.0, 0.0},
  {-0.0, -0.0}
}

Definition at line 326 of file AMDGPULibCalls.cpp.

Referenced by getOptTable().

tbl_tgamma

const TableEntry tbl_tgamma[]

static

Initial value:

= {
  {1.0, 1.0},
  {1.0, 2.0},
  {2.0, 3.0},
  {6.0, 4.0}
}

Definition at line 330 of file AMDGPULibCalls.cpp.

Referenced by getOptTable().

UseNative

cl::list< std::string > UseNative("amdgpu-use-native", cl::desc("Comma separated list of functions to replace with native, or all"), cl::CommaSeparated, cl::ValueOptional, cl::Hidden) ( "amdgpu-use-native" , cl::desc("Comma separated list of functions to replace with native, or all") , cl::CommaSeparated , cl::ValueOptional , cl::Hidden  )

static

Referenced by getScalarAtomicRMWExpansion(), llvm::AMDGPULibCalls::initNativeFuncs(), and llvm::AMDGPUUseNativeCallsPass::run().