10b57cec5SDimitry Andric //===- NeonEmitter.cpp - Generate arm_neon.h for use with clang -*- C++ -*-===// 20b57cec5SDimitry Andric // 30b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 40b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information. 50b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 60b57cec5SDimitry Andric // 70b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 80b57cec5SDimitry Andric // 90b57cec5SDimitry Andric // This tablegen backend is responsible for emitting arm_neon.h, which includes 100b57cec5SDimitry Andric // a declaration and definition of each function specified by the ARM NEON 110b57cec5SDimitry Andric // compiler interface. See ARM document DUI0348B. 120b57cec5SDimitry Andric // 130b57cec5SDimitry Andric // Each NEON instruction is implemented in terms of 1 or more functions which 140b57cec5SDimitry Andric // are suffixed with the element type of the input vectors. Functions may be 150b57cec5SDimitry Andric // implemented in terms of generic vector operations such as +, *, -, etc. or 160b57cec5SDimitry Andric // by calling a __builtin_-prefixed function which will be handled by clang's 170b57cec5SDimitry Andric // CodeGen library. 180b57cec5SDimitry Andric // 190b57cec5SDimitry Andric // Additional validation code can be generated by this file when runHeader() is 200b57cec5SDimitry Andric // called, rather than the normal run() entry point. 210b57cec5SDimitry Andric // 220b57cec5SDimitry Andric // See also the documentation in include/clang/Basic/arm_neon.td. 230b57cec5SDimitry Andric // 240b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 250b57cec5SDimitry Andric 26a7dea167SDimitry Andric #include "TableGenBackends.h" 270b57cec5SDimitry Andric #include "llvm/ADT/ArrayRef.h" 280b57cec5SDimitry Andric #include "llvm/ADT/DenseMap.h" 290b57cec5SDimitry Andric #include "llvm/ADT/None.h" 300b57cec5SDimitry Andric #include "llvm/ADT/SmallVector.h" 310b57cec5SDimitry Andric #include "llvm/ADT/STLExtras.h" 320b57cec5SDimitry Andric #include "llvm/ADT/StringExtras.h" 330b57cec5SDimitry Andric #include "llvm/ADT/StringRef.h" 340b57cec5SDimitry Andric #include "llvm/Support/Casting.h" 350b57cec5SDimitry Andric #include "llvm/Support/ErrorHandling.h" 360b57cec5SDimitry Andric #include "llvm/Support/raw_ostream.h" 370b57cec5SDimitry Andric #include "llvm/TableGen/Error.h" 380b57cec5SDimitry Andric #include "llvm/TableGen/Record.h" 390b57cec5SDimitry Andric #include "llvm/TableGen/SetTheory.h" 400b57cec5SDimitry Andric #include <algorithm> 410b57cec5SDimitry Andric #include <cassert> 420b57cec5SDimitry Andric #include <cctype> 430b57cec5SDimitry Andric #include <cstddef> 440b57cec5SDimitry Andric #include <cstdint> 450b57cec5SDimitry Andric #include <deque> 460b57cec5SDimitry Andric #include <map> 470b57cec5SDimitry Andric #include <set> 480b57cec5SDimitry Andric #include <sstream> 490b57cec5SDimitry Andric #include <string> 500b57cec5SDimitry Andric #include <utility> 510b57cec5SDimitry Andric #include <vector> 520b57cec5SDimitry Andric 530b57cec5SDimitry Andric using namespace llvm; 540b57cec5SDimitry Andric 550b57cec5SDimitry Andric namespace { 560b57cec5SDimitry Andric 570b57cec5SDimitry Andric // While globals are generally bad, this one allows us to perform assertions 580b57cec5SDimitry Andric // liberally and somehow still trace them back to the def they indirectly 590b57cec5SDimitry Andric // came from. 600b57cec5SDimitry Andric static Record *CurrentRecord = nullptr; 610b57cec5SDimitry Andric static void assert_with_loc(bool Assertion, const std::string &Str) { 620b57cec5SDimitry Andric if (!Assertion) { 630b57cec5SDimitry Andric if (CurrentRecord) 640b57cec5SDimitry Andric PrintFatalError(CurrentRecord->getLoc(), Str); 650b57cec5SDimitry Andric else 660b57cec5SDimitry Andric PrintFatalError(Str); 670b57cec5SDimitry Andric } 680b57cec5SDimitry Andric } 690b57cec5SDimitry Andric 700b57cec5SDimitry Andric enum ClassKind { 710b57cec5SDimitry Andric ClassNone, 720b57cec5SDimitry Andric ClassI, // generic integer instruction, e.g., "i8" suffix 730b57cec5SDimitry Andric ClassS, // signed/unsigned/poly, e.g., "s8", "u8" or "p8" suffix 740b57cec5SDimitry Andric ClassW, // width-specific instruction, e.g., "8" suffix 750b57cec5SDimitry Andric ClassB, // bitcast arguments with enum argument to specify type 760b57cec5SDimitry Andric ClassL, // Logical instructions which are op instructions 770b57cec5SDimitry Andric // but we need to not emit any suffix for in our 780b57cec5SDimitry Andric // tests. 790b57cec5SDimitry Andric ClassNoTest // Instructions which we do not test since they are 800b57cec5SDimitry Andric // not TRUE instructions. 810b57cec5SDimitry Andric }; 820b57cec5SDimitry Andric 830b57cec5SDimitry Andric /// NeonTypeFlags - Flags to identify the types for overloaded Neon 840b57cec5SDimitry Andric /// builtins. These must be kept in sync with the flags in 850b57cec5SDimitry Andric /// include/clang/Basic/TargetBuiltins.h. 860b57cec5SDimitry Andric namespace NeonTypeFlags { 870b57cec5SDimitry Andric 880b57cec5SDimitry Andric enum { EltTypeMask = 0xf, UnsignedFlag = 0x10, QuadFlag = 0x20 }; 890b57cec5SDimitry Andric 900b57cec5SDimitry Andric enum EltType { 910b57cec5SDimitry Andric Int8, 920b57cec5SDimitry Andric Int16, 930b57cec5SDimitry Andric Int32, 940b57cec5SDimitry Andric Int64, 950b57cec5SDimitry Andric Poly8, 960b57cec5SDimitry Andric Poly16, 970b57cec5SDimitry Andric Poly64, 980b57cec5SDimitry Andric Poly128, 990b57cec5SDimitry Andric Float16, 1000b57cec5SDimitry Andric Float32, 1010b57cec5SDimitry Andric Float64 1020b57cec5SDimitry Andric }; 1030b57cec5SDimitry Andric 1040b57cec5SDimitry Andric } // end namespace NeonTypeFlags 1050b57cec5SDimitry Andric 1060b57cec5SDimitry Andric class NeonEmitter; 1070b57cec5SDimitry Andric 1080b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 1090b57cec5SDimitry Andric // TypeSpec 1100b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 1110b57cec5SDimitry Andric 1120b57cec5SDimitry Andric /// A TypeSpec is just a simple wrapper around a string, but gets its own type 1130b57cec5SDimitry Andric /// for strong typing purposes. 1140b57cec5SDimitry Andric /// 1150b57cec5SDimitry Andric /// A TypeSpec can be used to create a type. 1160b57cec5SDimitry Andric class TypeSpec : public std::string { 1170b57cec5SDimitry Andric public: 1180b57cec5SDimitry Andric static std::vector<TypeSpec> fromTypeSpecs(StringRef Str) { 1190b57cec5SDimitry Andric std::vector<TypeSpec> Ret; 1200b57cec5SDimitry Andric TypeSpec Acc; 1210b57cec5SDimitry Andric for (char I : Str.str()) { 1220b57cec5SDimitry Andric if (islower(I)) { 1230b57cec5SDimitry Andric Acc.push_back(I); 1240b57cec5SDimitry Andric Ret.push_back(TypeSpec(Acc)); 1250b57cec5SDimitry Andric Acc.clear(); 1260b57cec5SDimitry Andric } else { 1270b57cec5SDimitry Andric Acc.push_back(I); 1280b57cec5SDimitry Andric } 1290b57cec5SDimitry Andric } 1300b57cec5SDimitry Andric return Ret; 1310b57cec5SDimitry Andric } 1320b57cec5SDimitry Andric }; 1330b57cec5SDimitry Andric 1340b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 1350b57cec5SDimitry Andric // Type 1360b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 1370b57cec5SDimitry Andric 1380b57cec5SDimitry Andric /// A Type. Not much more to say here. 1390b57cec5SDimitry Andric class Type { 1400b57cec5SDimitry Andric private: 1410b57cec5SDimitry Andric TypeSpec TS; 1420b57cec5SDimitry Andric 143*480093f4SDimitry Andric enum TypeKind { 144*480093f4SDimitry Andric Void, 145*480093f4SDimitry Andric Float, 146*480093f4SDimitry Andric SInt, 147*480093f4SDimitry Andric UInt, 148*480093f4SDimitry Andric Poly, 149*480093f4SDimitry Andric }; 150*480093f4SDimitry Andric TypeKind Kind; 151*480093f4SDimitry Andric bool Immediate, Constant, Pointer; 1520b57cec5SDimitry Andric // ScalarForMangling and NoManglingQ are really not suited to live here as 1530b57cec5SDimitry Andric // they are not related to the type. But they live in the TypeSpec (not the 1540b57cec5SDimitry Andric // prototype), so this is really the only place to store them. 1550b57cec5SDimitry Andric bool ScalarForMangling, NoManglingQ; 1560b57cec5SDimitry Andric unsigned Bitwidth, ElementBitwidth, NumVectors; 1570b57cec5SDimitry Andric 1580b57cec5SDimitry Andric public: 1590b57cec5SDimitry Andric Type() 160*480093f4SDimitry Andric : Kind(Void), Immediate(false), Constant(false), 161*480093f4SDimitry Andric Pointer(false), ScalarForMangling(false), NoManglingQ(false), 162*480093f4SDimitry Andric Bitwidth(0), ElementBitwidth(0), NumVectors(0) {} 1630b57cec5SDimitry Andric 164*480093f4SDimitry Andric Type(TypeSpec TS, StringRef CharMods) 165*480093f4SDimitry Andric : TS(std::move(TS)), Kind(Void), Immediate(false), 166*480093f4SDimitry Andric Constant(false), Pointer(false), ScalarForMangling(false), 167*480093f4SDimitry Andric NoManglingQ(false), Bitwidth(0), ElementBitwidth(0), NumVectors(0) { 168*480093f4SDimitry Andric applyModifiers(CharMods); 1690b57cec5SDimitry Andric } 1700b57cec5SDimitry Andric 1710b57cec5SDimitry Andric /// Returns a type representing "void". 1720b57cec5SDimitry Andric static Type getVoid() { return Type(); } 1730b57cec5SDimitry Andric 1740b57cec5SDimitry Andric bool operator==(const Type &Other) const { return str() == Other.str(); } 1750b57cec5SDimitry Andric bool operator!=(const Type &Other) const { return !operator==(Other); } 1760b57cec5SDimitry Andric 1770b57cec5SDimitry Andric // 1780b57cec5SDimitry Andric // Query functions 1790b57cec5SDimitry Andric // 1800b57cec5SDimitry Andric bool isScalarForMangling() const { return ScalarForMangling; } 1810b57cec5SDimitry Andric bool noManglingQ() const { return NoManglingQ; } 1820b57cec5SDimitry Andric 1830b57cec5SDimitry Andric bool isPointer() const { return Pointer; } 184*480093f4SDimitry Andric bool isValue() const { return !isVoid() && !isPointer(); } 185*480093f4SDimitry Andric bool isScalar() const { return isValue() && NumVectors == 0; } 186*480093f4SDimitry Andric bool isVector() const { return isValue() && NumVectors > 0; } 187*480093f4SDimitry Andric bool isConstPointer() const { return Constant; } 188*480093f4SDimitry Andric bool isFloating() const { return Kind == Float; } 189*480093f4SDimitry Andric bool isInteger() const { return Kind == SInt || Kind == UInt; } 190*480093f4SDimitry Andric bool isPoly() const { return Kind == Poly; } 191*480093f4SDimitry Andric bool isSigned() const { return Kind == SInt; } 1920b57cec5SDimitry Andric bool isImmediate() const { return Immediate; } 193*480093f4SDimitry Andric bool isFloat() const { return isFloating() && ElementBitwidth == 32; } 194*480093f4SDimitry Andric bool isDouble() const { return isFloating() && ElementBitwidth == 64; } 195*480093f4SDimitry Andric bool isHalf() const { return isFloating() && ElementBitwidth == 16; } 1960b57cec5SDimitry Andric bool isChar() const { return ElementBitwidth == 8; } 197*480093f4SDimitry Andric bool isShort() const { return isInteger() && ElementBitwidth == 16; } 198*480093f4SDimitry Andric bool isInt() const { return isInteger() && ElementBitwidth == 32; } 199*480093f4SDimitry Andric bool isLong() const { return isInteger() && ElementBitwidth == 64; } 200*480093f4SDimitry Andric bool isVoid() const { return Kind == Void; } 2010b57cec5SDimitry Andric unsigned getNumElements() const { return Bitwidth / ElementBitwidth; } 2020b57cec5SDimitry Andric unsigned getSizeInBits() const { return Bitwidth; } 2030b57cec5SDimitry Andric unsigned getElementSizeInBits() const { return ElementBitwidth; } 2040b57cec5SDimitry Andric unsigned getNumVectors() const { return NumVectors; } 2050b57cec5SDimitry Andric 2060b57cec5SDimitry Andric // 2070b57cec5SDimitry Andric // Mutator functions 2080b57cec5SDimitry Andric // 209*480093f4SDimitry Andric void makeUnsigned() { 210*480093f4SDimitry Andric assert(!isVoid() && "not a potentially signed type"); 211*480093f4SDimitry Andric Kind = UInt; 212*480093f4SDimitry Andric } 213*480093f4SDimitry Andric void makeSigned() { 214*480093f4SDimitry Andric assert(!isVoid() && "not a potentially signed type"); 215*480093f4SDimitry Andric Kind = SInt; 216*480093f4SDimitry Andric } 2170b57cec5SDimitry Andric 2180b57cec5SDimitry Andric void makeInteger(unsigned ElemWidth, bool Sign) { 219*480093f4SDimitry Andric assert(!isVoid() && "converting void to int probably not useful"); 220*480093f4SDimitry Andric Kind = Sign ? SInt : UInt; 2210b57cec5SDimitry Andric Immediate = false; 2220b57cec5SDimitry Andric ElementBitwidth = ElemWidth; 2230b57cec5SDimitry Andric } 2240b57cec5SDimitry Andric 2250b57cec5SDimitry Andric void makeImmediate(unsigned ElemWidth) { 226*480093f4SDimitry Andric Kind = SInt; 2270b57cec5SDimitry Andric Immediate = true; 2280b57cec5SDimitry Andric ElementBitwidth = ElemWidth; 2290b57cec5SDimitry Andric } 2300b57cec5SDimitry Andric 2310b57cec5SDimitry Andric void makeScalar() { 2320b57cec5SDimitry Andric Bitwidth = ElementBitwidth; 2330b57cec5SDimitry Andric NumVectors = 0; 2340b57cec5SDimitry Andric } 2350b57cec5SDimitry Andric 2360b57cec5SDimitry Andric void makeOneVector() { 2370b57cec5SDimitry Andric assert(isVector()); 2380b57cec5SDimitry Andric NumVectors = 1; 2390b57cec5SDimitry Andric } 2400b57cec5SDimitry Andric 2410b57cec5SDimitry Andric void doubleLanes() { 2420b57cec5SDimitry Andric assert_with_loc(Bitwidth != 128, "Can't get bigger than 128!"); 2430b57cec5SDimitry Andric Bitwidth = 128; 2440b57cec5SDimitry Andric } 2450b57cec5SDimitry Andric 2460b57cec5SDimitry Andric void halveLanes() { 2470b57cec5SDimitry Andric assert_with_loc(Bitwidth != 64, "Can't get smaller than 64!"); 2480b57cec5SDimitry Andric Bitwidth = 64; 2490b57cec5SDimitry Andric } 2500b57cec5SDimitry Andric 2510b57cec5SDimitry Andric /// Return the C string representation of a type, which is the typename 2520b57cec5SDimitry Andric /// defined in stdint.h or arm_neon.h. 2530b57cec5SDimitry Andric std::string str() const; 2540b57cec5SDimitry Andric 2550b57cec5SDimitry Andric /// Return the string representation of a type, which is an encoded 2560b57cec5SDimitry Andric /// string for passing to the BUILTIN() macro in Builtins.def. 2570b57cec5SDimitry Andric std::string builtin_str() const; 2580b57cec5SDimitry Andric 2590b57cec5SDimitry Andric /// Return the value in NeonTypeFlags for this type. 2600b57cec5SDimitry Andric unsigned getNeonEnum() const; 2610b57cec5SDimitry Andric 2620b57cec5SDimitry Andric /// Parse a type from a stdint.h or arm_neon.h typedef name, 2630b57cec5SDimitry Andric /// for example uint32x2_t or int64_t. 2640b57cec5SDimitry Andric static Type fromTypedefName(StringRef Name); 2650b57cec5SDimitry Andric 2660b57cec5SDimitry Andric private: 2670b57cec5SDimitry Andric /// Creates the type based on the typespec string in TS. 2680b57cec5SDimitry Andric /// Sets "Quad" to true if the "Q" or "H" modifiers were 2690b57cec5SDimitry Andric /// seen. This is needed by applyModifier as some modifiers 2700b57cec5SDimitry Andric /// only take effect if the type size was changed by "Q" or "H". 2710b57cec5SDimitry Andric void applyTypespec(bool &Quad); 272*480093f4SDimitry Andric /// Applies prototype modifiers to the type. 273*480093f4SDimitry Andric void applyModifiers(StringRef Mods); 2740b57cec5SDimitry Andric }; 2750b57cec5SDimitry Andric 2760b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 2770b57cec5SDimitry Andric // Variable 2780b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 2790b57cec5SDimitry Andric 2800b57cec5SDimitry Andric /// A variable is a simple class that just has a type and a name. 2810b57cec5SDimitry Andric class Variable { 2820b57cec5SDimitry Andric Type T; 2830b57cec5SDimitry Andric std::string N; 2840b57cec5SDimitry Andric 2850b57cec5SDimitry Andric public: 2860b57cec5SDimitry Andric Variable() : T(Type::getVoid()), N("") {} 2870b57cec5SDimitry Andric Variable(Type T, std::string N) : T(std::move(T)), N(std::move(N)) {} 2880b57cec5SDimitry Andric 2890b57cec5SDimitry Andric Type getType() const { return T; } 2900b57cec5SDimitry Andric std::string getName() const { return "__" + N; } 2910b57cec5SDimitry Andric }; 2920b57cec5SDimitry Andric 2930b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 2940b57cec5SDimitry Andric // Intrinsic 2950b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 2960b57cec5SDimitry Andric 2970b57cec5SDimitry Andric /// The main grunt class. This represents an instantiation of an intrinsic with 2980b57cec5SDimitry Andric /// a particular typespec and prototype. 2990b57cec5SDimitry Andric class Intrinsic { 3000b57cec5SDimitry Andric friend class DagEmitter; 3010b57cec5SDimitry Andric 3020b57cec5SDimitry Andric /// The Record this intrinsic was created from. 3030b57cec5SDimitry Andric Record *R; 304*480093f4SDimitry Andric /// The unmangled name. 305*480093f4SDimitry Andric std::string Name; 3060b57cec5SDimitry Andric /// The input and output typespecs. InTS == OutTS except when 3070b57cec5SDimitry Andric /// CartesianProductOfTypes is 1 - this is the case for vreinterpret. 3080b57cec5SDimitry Andric TypeSpec OutTS, InTS; 3090b57cec5SDimitry Andric /// The base class kind. Most intrinsics use ClassS, which has full type 3100b57cec5SDimitry Andric /// info for integers (s32/u32). Some use ClassI, which doesn't care about 3110b57cec5SDimitry Andric /// signedness (i32), while some (ClassB) have no type at all, only a width 3120b57cec5SDimitry Andric /// (32). 3130b57cec5SDimitry Andric ClassKind CK; 3140b57cec5SDimitry Andric /// The list of DAGs for the body. May be empty, in which case we should 3150b57cec5SDimitry Andric /// emit a builtin call. 3160b57cec5SDimitry Andric ListInit *Body; 3170b57cec5SDimitry Andric /// The architectural #ifdef guard. 3180b57cec5SDimitry Andric std::string Guard; 3190b57cec5SDimitry Andric /// Set if the Unavailable bit is 1. This means we don't generate a body, 3200b57cec5SDimitry Andric /// just an "unavailable" attribute on a declaration. 3210b57cec5SDimitry Andric bool IsUnavailable; 3220b57cec5SDimitry Andric /// Is this intrinsic safe for big-endian? or does it need its arguments 3230b57cec5SDimitry Andric /// reversing? 3240b57cec5SDimitry Andric bool BigEndianSafe; 3250b57cec5SDimitry Andric 3260b57cec5SDimitry Andric /// The types of return value [0] and parameters [1..]. 3270b57cec5SDimitry Andric std::vector<Type> Types; 328*480093f4SDimitry Andric /// The index of the key type passed to CGBuiltin.cpp for polymorphic calls. 329*480093f4SDimitry Andric int PolymorphicKeyType; 3300b57cec5SDimitry Andric /// The local variables defined. 3310b57cec5SDimitry Andric std::map<std::string, Variable> Variables; 3320b57cec5SDimitry Andric /// NeededEarly - set if any other intrinsic depends on this intrinsic. 3330b57cec5SDimitry Andric bool NeededEarly; 3340b57cec5SDimitry Andric /// UseMacro - set if we should implement using a macro or unset for a 3350b57cec5SDimitry Andric /// function. 3360b57cec5SDimitry Andric bool UseMacro; 3370b57cec5SDimitry Andric /// The set of intrinsics that this intrinsic uses/requires. 3380b57cec5SDimitry Andric std::set<Intrinsic *> Dependencies; 3390b57cec5SDimitry Andric /// The "base type", which is Type('d', OutTS). InBaseType is only 3400b57cec5SDimitry Andric /// different if CartesianProductOfTypes = 1 (for vreinterpret). 3410b57cec5SDimitry Andric Type BaseType, InBaseType; 3420b57cec5SDimitry Andric /// The return variable. 3430b57cec5SDimitry Andric Variable RetVar; 3440b57cec5SDimitry Andric /// A postfix to apply to every variable. Defaults to "". 3450b57cec5SDimitry Andric std::string VariablePostfix; 3460b57cec5SDimitry Andric 3470b57cec5SDimitry Andric NeonEmitter &Emitter; 3480b57cec5SDimitry Andric std::stringstream OS; 3490b57cec5SDimitry Andric 350a7dea167SDimitry Andric bool isBigEndianSafe() const { 351a7dea167SDimitry Andric if (BigEndianSafe) 352a7dea167SDimitry Andric return true; 353a7dea167SDimitry Andric 354a7dea167SDimitry Andric for (const auto &T : Types){ 355a7dea167SDimitry Andric if (T.isVector() && T.getNumElements() > 1) 356a7dea167SDimitry Andric return false; 357a7dea167SDimitry Andric } 358a7dea167SDimitry Andric return true; 359a7dea167SDimitry Andric } 360a7dea167SDimitry Andric 3610b57cec5SDimitry Andric public: 3620b57cec5SDimitry Andric Intrinsic(Record *R, StringRef Name, StringRef Proto, TypeSpec OutTS, 3630b57cec5SDimitry Andric TypeSpec InTS, ClassKind CK, ListInit *Body, NeonEmitter &Emitter, 3640b57cec5SDimitry Andric StringRef Guard, bool IsUnavailable, bool BigEndianSafe) 365*480093f4SDimitry Andric : R(R), Name(Name.str()), OutTS(OutTS), InTS(InTS), CK(CK), Body(Body), 366*480093f4SDimitry Andric Guard(Guard.str()), IsUnavailable(IsUnavailable), 367*480093f4SDimitry Andric BigEndianSafe(BigEndianSafe), PolymorphicKeyType(0), NeededEarly(false), 368*480093f4SDimitry Andric UseMacro(false), BaseType(OutTS, "."), InBaseType(InTS, "."), 369*480093f4SDimitry Andric Emitter(Emitter) { 3700b57cec5SDimitry Andric // Modify the TypeSpec per-argument to get a concrete Type, and create 3710b57cec5SDimitry Andric // known variables for each. 3720b57cec5SDimitry Andric // Types[0] is the return value. 373*480093f4SDimitry Andric unsigned Pos = 0; 374*480093f4SDimitry Andric Types.emplace_back(OutTS, getNextModifiers(Proto, Pos)); 375*480093f4SDimitry Andric StringRef Mods = getNextModifiers(Proto, Pos); 376*480093f4SDimitry Andric while (!Mods.empty()) { 377*480093f4SDimitry Andric Types.emplace_back(InTS, Mods); 378*480093f4SDimitry Andric if (Mods.find("!") != StringRef::npos) 379*480093f4SDimitry Andric PolymorphicKeyType = Types.size() - 1; 380*480093f4SDimitry Andric 381*480093f4SDimitry Andric Mods = getNextModifiers(Proto, Pos); 382*480093f4SDimitry Andric } 383*480093f4SDimitry Andric 384*480093f4SDimitry Andric for (auto Type : Types) { 385*480093f4SDimitry Andric // If this builtin takes an immediate argument, we need to #define it rather 386*480093f4SDimitry Andric // than use a standard declaration, so that SemaChecking can range check 387*480093f4SDimitry Andric // the immediate passed by the user. 388*480093f4SDimitry Andric 389*480093f4SDimitry Andric // Pointer arguments need to use macros to avoid hiding aligned attributes 390*480093f4SDimitry Andric // from the pointer type. 391*480093f4SDimitry Andric 392*480093f4SDimitry Andric // It is not permitted to pass or return an __fp16 by value, so intrinsics 393*480093f4SDimitry Andric // taking a scalar float16_t must be implemented as macros. 394*480093f4SDimitry Andric if (Type.isImmediate() || Type.isPointer() || 395*480093f4SDimitry Andric (Type.isScalar() && Type.isHalf())) 396*480093f4SDimitry Andric UseMacro = true; 397*480093f4SDimitry Andric } 3980b57cec5SDimitry Andric } 3990b57cec5SDimitry Andric 4000b57cec5SDimitry Andric /// Get the Record that this intrinsic is based off. 4010b57cec5SDimitry Andric Record *getRecord() const { return R; } 4020b57cec5SDimitry Andric /// Get the set of Intrinsics that this intrinsic calls. 4030b57cec5SDimitry Andric /// this is the set of immediate dependencies, NOT the 4040b57cec5SDimitry Andric /// transitive closure. 4050b57cec5SDimitry Andric const std::set<Intrinsic *> &getDependencies() const { return Dependencies; } 4060b57cec5SDimitry Andric /// Get the architectural guard string (#ifdef). 4070b57cec5SDimitry Andric std::string getGuard() const { return Guard; } 4080b57cec5SDimitry Andric /// Get the non-mangled name. 4090b57cec5SDimitry Andric std::string getName() const { return Name; } 4100b57cec5SDimitry Andric 4110b57cec5SDimitry Andric /// Return true if the intrinsic takes an immediate operand. 4120b57cec5SDimitry Andric bool hasImmediate() const { 413*480093f4SDimitry Andric return std::any_of(Types.begin(), Types.end(), 414*480093f4SDimitry Andric [](const Type &T) { return T.isImmediate(); }); 4150b57cec5SDimitry Andric } 4160b57cec5SDimitry Andric 4170b57cec5SDimitry Andric /// Return the parameter index of the immediate operand. 4180b57cec5SDimitry Andric unsigned getImmediateIdx() const { 419*480093f4SDimitry Andric for (unsigned Idx = 0; Idx < Types.size(); ++Idx) 420*480093f4SDimitry Andric if (Types[Idx].isImmediate()) 4210b57cec5SDimitry Andric return Idx - 1; 422*480093f4SDimitry Andric llvm_unreachable("Intrinsic has no immediate"); 4230b57cec5SDimitry Andric } 4240b57cec5SDimitry Andric 4250b57cec5SDimitry Andric 426*480093f4SDimitry Andric unsigned getNumParams() const { return Types.size() - 1; } 4270b57cec5SDimitry Andric Type getReturnType() const { return Types[0]; } 4280b57cec5SDimitry Andric Type getParamType(unsigned I) const { return Types[I + 1]; } 4290b57cec5SDimitry Andric Type getBaseType() const { return BaseType; } 430*480093f4SDimitry Andric Type getPolymorphicKeyType() const { return Types[PolymorphicKeyType]; } 4310b57cec5SDimitry Andric 4320b57cec5SDimitry Andric /// Return true if the prototype has a scalar argument. 4330b57cec5SDimitry Andric bool protoHasScalar() const; 4340b57cec5SDimitry Andric 4350b57cec5SDimitry Andric /// Return the index that parameter PIndex will sit at 4360b57cec5SDimitry Andric /// in a generated function call. This is often just PIndex, 4370b57cec5SDimitry Andric /// but may not be as things such as multiple-vector operands 4380b57cec5SDimitry Andric /// and sret parameters need to be taken into accont. 4390b57cec5SDimitry Andric unsigned getGeneratedParamIdx(unsigned PIndex) { 4400b57cec5SDimitry Andric unsigned Idx = 0; 4410b57cec5SDimitry Andric if (getReturnType().getNumVectors() > 1) 4420b57cec5SDimitry Andric // Multiple vectors are passed as sret. 4430b57cec5SDimitry Andric ++Idx; 4440b57cec5SDimitry Andric 4450b57cec5SDimitry Andric for (unsigned I = 0; I < PIndex; ++I) 4460b57cec5SDimitry Andric Idx += std::max(1U, getParamType(I).getNumVectors()); 4470b57cec5SDimitry Andric 4480b57cec5SDimitry Andric return Idx; 4490b57cec5SDimitry Andric } 4500b57cec5SDimitry Andric 4510b57cec5SDimitry Andric bool hasBody() const { return Body && !Body->getValues().empty(); } 4520b57cec5SDimitry Andric 4530b57cec5SDimitry Andric void setNeededEarly() { NeededEarly = true; } 4540b57cec5SDimitry Andric 4550b57cec5SDimitry Andric bool operator<(const Intrinsic &Other) const { 4560b57cec5SDimitry Andric // Sort lexicographically on a two-tuple (Guard, Name) 4570b57cec5SDimitry Andric if (Guard != Other.Guard) 4580b57cec5SDimitry Andric return Guard < Other.Guard; 4590b57cec5SDimitry Andric return Name < Other.Name; 4600b57cec5SDimitry Andric } 4610b57cec5SDimitry Andric 4620b57cec5SDimitry Andric ClassKind getClassKind(bool UseClassBIfScalar = false) { 4630b57cec5SDimitry Andric if (UseClassBIfScalar && !protoHasScalar()) 4640b57cec5SDimitry Andric return ClassB; 4650b57cec5SDimitry Andric return CK; 4660b57cec5SDimitry Andric } 4670b57cec5SDimitry Andric 4680b57cec5SDimitry Andric /// Return the name, mangled with type information. 4690b57cec5SDimitry Andric /// If ForceClassS is true, use ClassS (u32/s32) instead 4700b57cec5SDimitry Andric /// of the intrinsic's own type class. 4710b57cec5SDimitry Andric std::string getMangledName(bool ForceClassS = false) const; 4720b57cec5SDimitry Andric /// Return the type code for a builtin function call. 4730b57cec5SDimitry Andric std::string getInstTypeCode(Type T, ClassKind CK) const; 4740b57cec5SDimitry Andric /// Return the type string for a BUILTIN() macro in Builtins.def. 4750b57cec5SDimitry Andric std::string getBuiltinTypeStr(); 4760b57cec5SDimitry Andric 4770b57cec5SDimitry Andric /// Generate the intrinsic, returning code. 4780b57cec5SDimitry Andric std::string generate(); 4790b57cec5SDimitry Andric /// Perform type checking and populate the dependency graph, but 4800b57cec5SDimitry Andric /// don't generate code yet. 4810b57cec5SDimitry Andric void indexBody(); 4820b57cec5SDimitry Andric 4830b57cec5SDimitry Andric private: 484*480093f4SDimitry Andric StringRef getNextModifiers(StringRef Proto, unsigned &Pos) const; 485*480093f4SDimitry Andric 4860b57cec5SDimitry Andric std::string mangleName(std::string Name, ClassKind CK) const; 4870b57cec5SDimitry Andric 4880b57cec5SDimitry Andric void initVariables(); 4890b57cec5SDimitry Andric std::string replaceParamsIn(std::string S); 4900b57cec5SDimitry Andric 4910b57cec5SDimitry Andric void emitBodyAsBuiltinCall(); 4920b57cec5SDimitry Andric 4930b57cec5SDimitry Andric void generateImpl(bool ReverseArguments, 4940b57cec5SDimitry Andric StringRef NamePrefix, StringRef CallPrefix); 4950b57cec5SDimitry Andric void emitReturn(); 4960b57cec5SDimitry Andric void emitBody(StringRef CallPrefix); 4970b57cec5SDimitry Andric void emitShadowedArgs(); 4980b57cec5SDimitry Andric void emitArgumentReversal(); 4990b57cec5SDimitry Andric void emitReturnReversal(); 5000b57cec5SDimitry Andric void emitReverseVariable(Variable &Dest, Variable &Src); 5010b57cec5SDimitry Andric void emitNewLine(); 5020b57cec5SDimitry Andric void emitClosingBrace(); 5030b57cec5SDimitry Andric void emitOpeningBrace(); 5040b57cec5SDimitry Andric void emitPrototype(StringRef NamePrefix); 5050b57cec5SDimitry Andric 5060b57cec5SDimitry Andric class DagEmitter { 5070b57cec5SDimitry Andric Intrinsic &Intr; 5080b57cec5SDimitry Andric StringRef CallPrefix; 5090b57cec5SDimitry Andric 5100b57cec5SDimitry Andric public: 5110b57cec5SDimitry Andric DagEmitter(Intrinsic &Intr, StringRef CallPrefix) : 5120b57cec5SDimitry Andric Intr(Intr), CallPrefix(CallPrefix) { 5130b57cec5SDimitry Andric } 5140b57cec5SDimitry Andric std::pair<Type, std::string> emitDagArg(Init *Arg, std::string ArgName); 5150b57cec5SDimitry Andric std::pair<Type, std::string> emitDagSaveTemp(DagInit *DI); 5160b57cec5SDimitry Andric std::pair<Type, std::string> emitDagSplat(DagInit *DI); 5170b57cec5SDimitry Andric std::pair<Type, std::string> emitDagDup(DagInit *DI); 5180b57cec5SDimitry Andric std::pair<Type, std::string> emitDagDupTyped(DagInit *DI); 5190b57cec5SDimitry Andric std::pair<Type, std::string> emitDagShuffle(DagInit *DI); 5200b57cec5SDimitry Andric std::pair<Type, std::string> emitDagCast(DagInit *DI, bool IsBitCast); 5210b57cec5SDimitry Andric std::pair<Type, std::string> emitDagCall(DagInit *DI); 5220b57cec5SDimitry Andric std::pair<Type, std::string> emitDagNameReplace(DagInit *DI); 5230b57cec5SDimitry Andric std::pair<Type, std::string> emitDagLiteral(DagInit *DI); 5240b57cec5SDimitry Andric std::pair<Type, std::string> emitDagOp(DagInit *DI); 5250b57cec5SDimitry Andric std::pair<Type, std::string> emitDag(DagInit *DI); 5260b57cec5SDimitry Andric }; 5270b57cec5SDimitry Andric }; 5280b57cec5SDimitry Andric 5290b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 5300b57cec5SDimitry Andric // NeonEmitter 5310b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 5320b57cec5SDimitry Andric 5330b57cec5SDimitry Andric class NeonEmitter { 5340b57cec5SDimitry Andric RecordKeeper &Records; 5350b57cec5SDimitry Andric DenseMap<Record *, ClassKind> ClassMap; 5360b57cec5SDimitry Andric std::map<std::string, std::deque<Intrinsic>> IntrinsicMap; 5370b57cec5SDimitry Andric unsigned UniqueNumber; 5380b57cec5SDimitry Andric 5390b57cec5SDimitry Andric void createIntrinsic(Record *R, SmallVectorImpl<Intrinsic *> &Out); 5400b57cec5SDimitry Andric void genBuiltinsDef(raw_ostream &OS, SmallVectorImpl<Intrinsic *> &Defs); 5410b57cec5SDimitry Andric void genOverloadTypeCheckCode(raw_ostream &OS, 5420b57cec5SDimitry Andric SmallVectorImpl<Intrinsic *> &Defs); 5430b57cec5SDimitry Andric void genIntrinsicRangeCheckCode(raw_ostream &OS, 5440b57cec5SDimitry Andric SmallVectorImpl<Intrinsic *> &Defs); 5450b57cec5SDimitry Andric 5460b57cec5SDimitry Andric public: 5470b57cec5SDimitry Andric /// Called by Intrinsic - this attempts to get an intrinsic that takes 5480b57cec5SDimitry Andric /// the given types as arguments. 5490b57cec5SDimitry Andric Intrinsic &getIntrinsic(StringRef Name, ArrayRef<Type> Types); 5500b57cec5SDimitry Andric 5510b57cec5SDimitry Andric /// Called by Intrinsic - returns a globally-unique number. 5520b57cec5SDimitry Andric unsigned getUniqueNumber() { return UniqueNumber++; } 5530b57cec5SDimitry Andric 5540b57cec5SDimitry Andric NeonEmitter(RecordKeeper &R) : Records(R), UniqueNumber(0) { 5550b57cec5SDimitry Andric Record *SI = R.getClass("SInst"); 5560b57cec5SDimitry Andric Record *II = R.getClass("IInst"); 5570b57cec5SDimitry Andric Record *WI = R.getClass("WInst"); 5580b57cec5SDimitry Andric Record *SOpI = R.getClass("SOpInst"); 5590b57cec5SDimitry Andric Record *IOpI = R.getClass("IOpInst"); 5600b57cec5SDimitry Andric Record *WOpI = R.getClass("WOpInst"); 5610b57cec5SDimitry Andric Record *LOpI = R.getClass("LOpInst"); 5620b57cec5SDimitry Andric Record *NoTestOpI = R.getClass("NoTestOpInst"); 5630b57cec5SDimitry Andric 5640b57cec5SDimitry Andric ClassMap[SI] = ClassS; 5650b57cec5SDimitry Andric ClassMap[II] = ClassI; 5660b57cec5SDimitry Andric ClassMap[WI] = ClassW; 5670b57cec5SDimitry Andric ClassMap[SOpI] = ClassS; 5680b57cec5SDimitry Andric ClassMap[IOpI] = ClassI; 5690b57cec5SDimitry Andric ClassMap[WOpI] = ClassW; 5700b57cec5SDimitry Andric ClassMap[LOpI] = ClassL; 5710b57cec5SDimitry Andric ClassMap[NoTestOpI] = ClassNoTest; 5720b57cec5SDimitry Andric } 5730b57cec5SDimitry Andric 5740b57cec5SDimitry Andric // run - Emit arm_neon.h.inc 5750b57cec5SDimitry Andric void run(raw_ostream &o); 5760b57cec5SDimitry Andric 5770b57cec5SDimitry Andric // runFP16 - Emit arm_fp16.h.inc 5780b57cec5SDimitry Andric void runFP16(raw_ostream &o); 5790b57cec5SDimitry Andric 5800b57cec5SDimitry Andric // runHeader - Emit all the __builtin prototypes used in arm_neon.h 5810b57cec5SDimitry Andric // and arm_fp16.h 5820b57cec5SDimitry Andric void runHeader(raw_ostream &o); 5830b57cec5SDimitry Andric 5840b57cec5SDimitry Andric // runTests - Emit tests for all the Neon intrinsics. 5850b57cec5SDimitry Andric void runTests(raw_ostream &o); 5860b57cec5SDimitry Andric }; 5870b57cec5SDimitry Andric 5880b57cec5SDimitry Andric } // end anonymous namespace 5890b57cec5SDimitry Andric 5900b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 5910b57cec5SDimitry Andric // Type implementation 5920b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 5930b57cec5SDimitry Andric 5940b57cec5SDimitry Andric std::string Type::str() const { 595*480093f4SDimitry Andric if (isVoid()) 5960b57cec5SDimitry Andric return "void"; 5970b57cec5SDimitry Andric std::string S; 5980b57cec5SDimitry Andric 599*480093f4SDimitry Andric if (isInteger() && !isSigned()) 6000b57cec5SDimitry Andric S += "u"; 6010b57cec5SDimitry Andric 602*480093f4SDimitry Andric if (isPoly()) 6030b57cec5SDimitry Andric S += "poly"; 604*480093f4SDimitry Andric else if (isFloating()) 6050b57cec5SDimitry Andric S += "float"; 6060b57cec5SDimitry Andric else 6070b57cec5SDimitry Andric S += "int"; 6080b57cec5SDimitry Andric 6090b57cec5SDimitry Andric S += utostr(ElementBitwidth); 6100b57cec5SDimitry Andric if (isVector()) 6110b57cec5SDimitry Andric S += "x" + utostr(getNumElements()); 6120b57cec5SDimitry Andric if (NumVectors > 1) 6130b57cec5SDimitry Andric S += "x" + utostr(NumVectors); 6140b57cec5SDimitry Andric S += "_t"; 6150b57cec5SDimitry Andric 6160b57cec5SDimitry Andric if (Constant) 6170b57cec5SDimitry Andric S += " const"; 6180b57cec5SDimitry Andric if (Pointer) 6190b57cec5SDimitry Andric S += " *"; 6200b57cec5SDimitry Andric 6210b57cec5SDimitry Andric return S; 6220b57cec5SDimitry Andric } 6230b57cec5SDimitry Andric 6240b57cec5SDimitry Andric std::string Type::builtin_str() const { 6250b57cec5SDimitry Andric std::string S; 6260b57cec5SDimitry Andric if (isVoid()) 6270b57cec5SDimitry Andric return "v"; 6280b57cec5SDimitry Andric 629*480093f4SDimitry Andric if (isPointer()) { 6300b57cec5SDimitry Andric // All pointers are void pointers. 631*480093f4SDimitry Andric S = "v"; 632*480093f4SDimitry Andric if (isConstPointer()) 633*480093f4SDimitry Andric S += "C"; 634*480093f4SDimitry Andric S += "*"; 635*480093f4SDimitry Andric return S; 636*480093f4SDimitry Andric } else if (isInteger()) 6370b57cec5SDimitry Andric switch (ElementBitwidth) { 6380b57cec5SDimitry Andric case 8: S += "c"; break; 6390b57cec5SDimitry Andric case 16: S += "s"; break; 6400b57cec5SDimitry Andric case 32: S += "i"; break; 6410b57cec5SDimitry Andric case 64: S += "Wi"; break; 6420b57cec5SDimitry Andric case 128: S += "LLLi"; break; 6430b57cec5SDimitry Andric default: llvm_unreachable("Unhandled case!"); 6440b57cec5SDimitry Andric } 6450b57cec5SDimitry Andric else 6460b57cec5SDimitry Andric switch (ElementBitwidth) { 6470b57cec5SDimitry Andric case 16: S += "h"; break; 6480b57cec5SDimitry Andric case 32: S += "f"; break; 6490b57cec5SDimitry Andric case 64: S += "d"; break; 6500b57cec5SDimitry Andric default: llvm_unreachable("Unhandled case!"); 6510b57cec5SDimitry Andric } 6520b57cec5SDimitry Andric 653*480093f4SDimitry Andric // FIXME: NECESSARY??????????????????????????????????????????????????????????????????????? 654*480093f4SDimitry Andric if (isChar() && !isPointer() && isSigned()) 6550b57cec5SDimitry Andric // Make chars explicitly signed. 6560b57cec5SDimitry Andric S = "S" + S; 657*480093f4SDimitry Andric else if (isInteger() && !isSigned()) 6580b57cec5SDimitry Andric S = "U" + S; 6590b57cec5SDimitry Andric 6600b57cec5SDimitry Andric // Constant indices are "int", but have the "constant expression" modifier. 6610b57cec5SDimitry Andric if (isImmediate()) { 6620b57cec5SDimitry Andric assert(isInteger() && isSigned()); 6630b57cec5SDimitry Andric S = "I" + S; 6640b57cec5SDimitry Andric } 6650b57cec5SDimitry Andric 666*480093f4SDimitry Andric if (isScalar()) 6670b57cec5SDimitry Andric return S; 6680b57cec5SDimitry Andric 6690b57cec5SDimitry Andric std::string Ret; 6700b57cec5SDimitry Andric for (unsigned I = 0; I < NumVectors; ++I) 6710b57cec5SDimitry Andric Ret += "V" + utostr(getNumElements()) + S; 6720b57cec5SDimitry Andric 6730b57cec5SDimitry Andric return Ret; 6740b57cec5SDimitry Andric } 6750b57cec5SDimitry Andric 6760b57cec5SDimitry Andric unsigned Type::getNeonEnum() const { 6770b57cec5SDimitry Andric unsigned Addend; 6780b57cec5SDimitry Andric switch (ElementBitwidth) { 6790b57cec5SDimitry Andric case 8: Addend = 0; break; 6800b57cec5SDimitry Andric case 16: Addend = 1; break; 6810b57cec5SDimitry Andric case 32: Addend = 2; break; 6820b57cec5SDimitry Andric case 64: Addend = 3; break; 6830b57cec5SDimitry Andric case 128: Addend = 4; break; 6840b57cec5SDimitry Andric default: llvm_unreachable("Unhandled element bitwidth!"); 6850b57cec5SDimitry Andric } 6860b57cec5SDimitry Andric 6870b57cec5SDimitry Andric unsigned Base = (unsigned)NeonTypeFlags::Int8 + Addend; 688*480093f4SDimitry Andric if (isPoly()) { 6890b57cec5SDimitry Andric // Adjustment needed because Poly32 doesn't exist. 6900b57cec5SDimitry Andric if (Addend >= 2) 6910b57cec5SDimitry Andric --Addend; 6920b57cec5SDimitry Andric Base = (unsigned)NeonTypeFlags::Poly8 + Addend; 6930b57cec5SDimitry Andric } 694*480093f4SDimitry Andric if (isFloating()) { 6950b57cec5SDimitry Andric assert(Addend != 0 && "Float8 doesn't exist!"); 6960b57cec5SDimitry Andric Base = (unsigned)NeonTypeFlags::Float16 + (Addend - 1); 6970b57cec5SDimitry Andric } 6980b57cec5SDimitry Andric 6990b57cec5SDimitry Andric if (Bitwidth == 128) 7000b57cec5SDimitry Andric Base |= (unsigned)NeonTypeFlags::QuadFlag; 701*480093f4SDimitry Andric if (isInteger() && !isSigned()) 7020b57cec5SDimitry Andric Base |= (unsigned)NeonTypeFlags::UnsignedFlag; 7030b57cec5SDimitry Andric 7040b57cec5SDimitry Andric return Base; 7050b57cec5SDimitry Andric } 7060b57cec5SDimitry Andric 7070b57cec5SDimitry Andric Type Type::fromTypedefName(StringRef Name) { 7080b57cec5SDimitry Andric Type T; 709*480093f4SDimitry Andric T.Kind = SInt; 7100b57cec5SDimitry Andric 7110b57cec5SDimitry Andric if (Name.front() == 'u') { 712*480093f4SDimitry Andric T.Kind = UInt; 7130b57cec5SDimitry Andric Name = Name.drop_front(); 7140b57cec5SDimitry Andric } 7150b57cec5SDimitry Andric 7160b57cec5SDimitry Andric if (Name.startswith("float")) { 717*480093f4SDimitry Andric T.Kind = Float; 7180b57cec5SDimitry Andric Name = Name.drop_front(5); 7190b57cec5SDimitry Andric } else if (Name.startswith("poly")) { 720*480093f4SDimitry Andric T.Kind = Poly; 7210b57cec5SDimitry Andric Name = Name.drop_front(4); 7220b57cec5SDimitry Andric } else { 7230b57cec5SDimitry Andric assert(Name.startswith("int")); 7240b57cec5SDimitry Andric Name = Name.drop_front(3); 7250b57cec5SDimitry Andric } 7260b57cec5SDimitry Andric 7270b57cec5SDimitry Andric unsigned I = 0; 7280b57cec5SDimitry Andric for (I = 0; I < Name.size(); ++I) { 7290b57cec5SDimitry Andric if (!isdigit(Name[I])) 7300b57cec5SDimitry Andric break; 7310b57cec5SDimitry Andric } 7320b57cec5SDimitry Andric Name.substr(0, I).getAsInteger(10, T.ElementBitwidth); 7330b57cec5SDimitry Andric Name = Name.drop_front(I); 7340b57cec5SDimitry Andric 7350b57cec5SDimitry Andric T.Bitwidth = T.ElementBitwidth; 7360b57cec5SDimitry Andric T.NumVectors = 1; 7370b57cec5SDimitry Andric 7380b57cec5SDimitry Andric if (Name.front() == 'x') { 7390b57cec5SDimitry Andric Name = Name.drop_front(); 7400b57cec5SDimitry Andric unsigned I = 0; 7410b57cec5SDimitry Andric for (I = 0; I < Name.size(); ++I) { 7420b57cec5SDimitry Andric if (!isdigit(Name[I])) 7430b57cec5SDimitry Andric break; 7440b57cec5SDimitry Andric } 7450b57cec5SDimitry Andric unsigned NumLanes; 7460b57cec5SDimitry Andric Name.substr(0, I).getAsInteger(10, NumLanes); 7470b57cec5SDimitry Andric Name = Name.drop_front(I); 7480b57cec5SDimitry Andric T.Bitwidth = T.ElementBitwidth * NumLanes; 7490b57cec5SDimitry Andric } else { 7500b57cec5SDimitry Andric // Was scalar. 7510b57cec5SDimitry Andric T.NumVectors = 0; 7520b57cec5SDimitry Andric } 7530b57cec5SDimitry Andric if (Name.front() == 'x') { 7540b57cec5SDimitry Andric Name = Name.drop_front(); 7550b57cec5SDimitry Andric unsigned I = 0; 7560b57cec5SDimitry Andric for (I = 0; I < Name.size(); ++I) { 7570b57cec5SDimitry Andric if (!isdigit(Name[I])) 7580b57cec5SDimitry Andric break; 7590b57cec5SDimitry Andric } 7600b57cec5SDimitry Andric Name.substr(0, I).getAsInteger(10, T.NumVectors); 7610b57cec5SDimitry Andric Name = Name.drop_front(I); 7620b57cec5SDimitry Andric } 7630b57cec5SDimitry Andric 7640b57cec5SDimitry Andric assert(Name.startswith("_t") && "Malformed typedef!"); 7650b57cec5SDimitry Andric return T; 7660b57cec5SDimitry Andric } 7670b57cec5SDimitry Andric 7680b57cec5SDimitry Andric void Type::applyTypespec(bool &Quad) { 7690b57cec5SDimitry Andric std::string S = TS; 7700b57cec5SDimitry Andric ScalarForMangling = false; 771*480093f4SDimitry Andric Kind = SInt; 7720b57cec5SDimitry Andric ElementBitwidth = ~0U; 7730b57cec5SDimitry Andric NumVectors = 1; 7740b57cec5SDimitry Andric 7750b57cec5SDimitry Andric for (char I : S) { 7760b57cec5SDimitry Andric switch (I) { 7770b57cec5SDimitry Andric case 'S': 7780b57cec5SDimitry Andric ScalarForMangling = true; 7790b57cec5SDimitry Andric break; 7800b57cec5SDimitry Andric case 'H': 7810b57cec5SDimitry Andric NoManglingQ = true; 7820b57cec5SDimitry Andric Quad = true; 7830b57cec5SDimitry Andric break; 7840b57cec5SDimitry Andric case 'Q': 7850b57cec5SDimitry Andric Quad = true; 7860b57cec5SDimitry Andric break; 7870b57cec5SDimitry Andric case 'P': 788*480093f4SDimitry Andric Kind = Poly; 7890b57cec5SDimitry Andric break; 7900b57cec5SDimitry Andric case 'U': 791*480093f4SDimitry Andric Kind = UInt; 7920b57cec5SDimitry Andric break; 7930b57cec5SDimitry Andric case 'c': 7940b57cec5SDimitry Andric ElementBitwidth = 8; 7950b57cec5SDimitry Andric break; 7960b57cec5SDimitry Andric case 'h': 797*480093f4SDimitry Andric Kind = Float; 7980b57cec5SDimitry Andric LLVM_FALLTHROUGH; 7990b57cec5SDimitry Andric case 's': 8000b57cec5SDimitry Andric ElementBitwidth = 16; 8010b57cec5SDimitry Andric break; 8020b57cec5SDimitry Andric case 'f': 803*480093f4SDimitry Andric Kind = Float; 8040b57cec5SDimitry Andric LLVM_FALLTHROUGH; 8050b57cec5SDimitry Andric case 'i': 8060b57cec5SDimitry Andric ElementBitwidth = 32; 8070b57cec5SDimitry Andric break; 8080b57cec5SDimitry Andric case 'd': 809*480093f4SDimitry Andric Kind = Float; 8100b57cec5SDimitry Andric LLVM_FALLTHROUGH; 8110b57cec5SDimitry Andric case 'l': 8120b57cec5SDimitry Andric ElementBitwidth = 64; 8130b57cec5SDimitry Andric break; 8140b57cec5SDimitry Andric case 'k': 8150b57cec5SDimitry Andric ElementBitwidth = 128; 8160b57cec5SDimitry Andric // Poly doesn't have a 128x1 type. 817*480093f4SDimitry Andric if (isPoly()) 8180b57cec5SDimitry Andric NumVectors = 0; 8190b57cec5SDimitry Andric break; 8200b57cec5SDimitry Andric default: 8210b57cec5SDimitry Andric llvm_unreachable("Unhandled type code!"); 8220b57cec5SDimitry Andric } 8230b57cec5SDimitry Andric } 8240b57cec5SDimitry Andric assert(ElementBitwidth != ~0U && "Bad element bitwidth!"); 8250b57cec5SDimitry Andric 8260b57cec5SDimitry Andric Bitwidth = Quad ? 128 : 64; 8270b57cec5SDimitry Andric } 8280b57cec5SDimitry Andric 829*480093f4SDimitry Andric void Type::applyModifiers(StringRef Mods) { 8300b57cec5SDimitry Andric bool AppliedQuad = false; 8310b57cec5SDimitry Andric applyTypespec(AppliedQuad); 8320b57cec5SDimitry Andric 833*480093f4SDimitry Andric for (char Mod : Mods) { 8340b57cec5SDimitry Andric switch (Mod) { 835*480093f4SDimitry Andric case '.': 836*480093f4SDimitry Andric break; 8370b57cec5SDimitry Andric case 'v': 838*480093f4SDimitry Andric Kind = Void; 8390b57cec5SDimitry Andric break; 840*480093f4SDimitry Andric case 'S': 841*480093f4SDimitry Andric Kind = SInt; 8420b57cec5SDimitry Andric break; 8430b57cec5SDimitry Andric case 'U': 844*480093f4SDimitry Andric Kind = UInt; 8450b57cec5SDimitry Andric break; 8460b57cec5SDimitry Andric case 'F': 847*480093f4SDimitry Andric Kind = Float; 8480b57cec5SDimitry Andric break; 849*480093f4SDimitry Andric case 'P': 850*480093f4SDimitry Andric Kind = Poly; 8510b57cec5SDimitry Andric break; 852*480093f4SDimitry Andric case '>': 853*480093f4SDimitry Andric assert(ElementBitwidth < 128); 854*480093f4SDimitry Andric ElementBitwidth *= 2; 855*480093f4SDimitry Andric break; 856*480093f4SDimitry Andric case '<': 857*480093f4SDimitry Andric assert(ElementBitwidth > 8); 858*480093f4SDimitry Andric ElementBitwidth /= 2; 8590b57cec5SDimitry Andric break; 8600b57cec5SDimitry Andric case '1': 8610b57cec5SDimitry Andric NumVectors = 0; 8620b57cec5SDimitry Andric break; 8630b57cec5SDimitry Andric case '2': 8640b57cec5SDimitry Andric NumVectors = 2; 8650b57cec5SDimitry Andric break; 8660b57cec5SDimitry Andric case '3': 8670b57cec5SDimitry Andric NumVectors = 3; 8680b57cec5SDimitry Andric break; 8690b57cec5SDimitry Andric case '4': 8700b57cec5SDimitry Andric NumVectors = 4; 8710b57cec5SDimitry Andric break; 872*480093f4SDimitry Andric case '*': 873*480093f4SDimitry Andric Pointer = true; 8740b57cec5SDimitry Andric break; 875*480093f4SDimitry Andric case 'c': 876*480093f4SDimitry Andric Constant = true; 8770b57cec5SDimitry Andric break; 878*480093f4SDimitry Andric case 'Q': 879*480093f4SDimitry Andric Bitwidth = 128; 8800b57cec5SDimitry Andric break; 881*480093f4SDimitry Andric case 'q': 882*480093f4SDimitry Andric Bitwidth = 64; 8830b57cec5SDimitry Andric break; 884*480093f4SDimitry Andric case 'I': 885*480093f4SDimitry Andric Kind = SInt; 886*480093f4SDimitry Andric ElementBitwidth = Bitwidth = 32; 887*480093f4SDimitry Andric NumVectors = 0; 888*480093f4SDimitry Andric Immediate = true; 8890b57cec5SDimitry Andric break; 890*480093f4SDimitry Andric case 'p': 891*480093f4SDimitry Andric if (isPoly()) 892*480093f4SDimitry Andric Kind = UInt; 893*480093f4SDimitry Andric break; 894*480093f4SDimitry Andric case '!': 895*480093f4SDimitry Andric // Key type, handled elsewhere. 8960b57cec5SDimitry Andric break; 8970b57cec5SDimitry Andric default: 8980b57cec5SDimitry Andric llvm_unreachable("Unhandled character!"); 8990b57cec5SDimitry Andric } 9000b57cec5SDimitry Andric } 901*480093f4SDimitry Andric } 9020b57cec5SDimitry Andric 9030b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 9040b57cec5SDimitry Andric // Intrinsic implementation 9050b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 9060b57cec5SDimitry Andric 907*480093f4SDimitry Andric StringRef Intrinsic::getNextModifiers(StringRef Proto, unsigned &Pos) const { 908*480093f4SDimitry Andric if (Proto.size() == Pos) 909*480093f4SDimitry Andric return StringRef(); 910*480093f4SDimitry Andric else if (Proto[Pos] != '(') 911*480093f4SDimitry Andric return Proto.substr(Pos++, 1); 912*480093f4SDimitry Andric 913*480093f4SDimitry Andric size_t Start = Pos + 1; 914*480093f4SDimitry Andric size_t End = Proto.find(')', Start); 915*480093f4SDimitry Andric assert_with_loc(End != StringRef::npos, "unmatched modifier group paren"); 916*480093f4SDimitry Andric Pos = End + 1; 917*480093f4SDimitry Andric return Proto.slice(Start, End); 918*480093f4SDimitry Andric } 919*480093f4SDimitry Andric 9200b57cec5SDimitry Andric std::string Intrinsic::getInstTypeCode(Type T, ClassKind CK) const { 9210b57cec5SDimitry Andric char typeCode = '\0'; 9220b57cec5SDimitry Andric bool printNumber = true; 9230b57cec5SDimitry Andric 9240b57cec5SDimitry Andric if (CK == ClassB) 9250b57cec5SDimitry Andric return ""; 9260b57cec5SDimitry Andric 9270b57cec5SDimitry Andric if (T.isPoly()) 9280b57cec5SDimitry Andric typeCode = 'p'; 9290b57cec5SDimitry Andric else if (T.isInteger()) 9300b57cec5SDimitry Andric typeCode = T.isSigned() ? 's' : 'u'; 9310b57cec5SDimitry Andric else 9320b57cec5SDimitry Andric typeCode = 'f'; 9330b57cec5SDimitry Andric 9340b57cec5SDimitry Andric if (CK == ClassI) { 9350b57cec5SDimitry Andric switch (typeCode) { 9360b57cec5SDimitry Andric default: 9370b57cec5SDimitry Andric break; 9380b57cec5SDimitry Andric case 's': 9390b57cec5SDimitry Andric case 'u': 9400b57cec5SDimitry Andric case 'p': 9410b57cec5SDimitry Andric typeCode = 'i'; 9420b57cec5SDimitry Andric break; 9430b57cec5SDimitry Andric } 9440b57cec5SDimitry Andric } 9450b57cec5SDimitry Andric if (CK == ClassB) { 9460b57cec5SDimitry Andric typeCode = '\0'; 9470b57cec5SDimitry Andric } 9480b57cec5SDimitry Andric 9490b57cec5SDimitry Andric std::string S; 9500b57cec5SDimitry Andric if (typeCode != '\0') 9510b57cec5SDimitry Andric S.push_back(typeCode); 9520b57cec5SDimitry Andric if (printNumber) 9530b57cec5SDimitry Andric S += utostr(T.getElementSizeInBits()); 9540b57cec5SDimitry Andric 9550b57cec5SDimitry Andric return S; 9560b57cec5SDimitry Andric } 9570b57cec5SDimitry Andric 9580b57cec5SDimitry Andric std::string Intrinsic::getBuiltinTypeStr() { 9590b57cec5SDimitry Andric ClassKind LocalCK = getClassKind(true); 9600b57cec5SDimitry Andric std::string S; 9610b57cec5SDimitry Andric 9620b57cec5SDimitry Andric Type RetT = getReturnType(); 9630b57cec5SDimitry Andric if ((LocalCK == ClassI || LocalCK == ClassW) && RetT.isScalar() && 9640b57cec5SDimitry Andric !RetT.isFloating()) 9650b57cec5SDimitry Andric RetT.makeInteger(RetT.getElementSizeInBits(), false); 9660b57cec5SDimitry Andric 9670b57cec5SDimitry Andric // Since the return value must be one type, return a vector type of the 9680b57cec5SDimitry Andric // appropriate width which we will bitcast. An exception is made for 9690b57cec5SDimitry Andric // returning structs of 2, 3, or 4 vectors which are returned in a sret-like 9700b57cec5SDimitry Andric // fashion, storing them to a pointer arg. 9710b57cec5SDimitry Andric if (RetT.getNumVectors() > 1) { 9720b57cec5SDimitry Andric S += "vv*"; // void result with void* first argument 9730b57cec5SDimitry Andric } else { 9740b57cec5SDimitry Andric if (RetT.isPoly()) 9750b57cec5SDimitry Andric RetT.makeInteger(RetT.getElementSizeInBits(), false); 976*480093f4SDimitry Andric if (!RetT.isScalar() && RetT.isInteger() && !RetT.isSigned()) 9770b57cec5SDimitry Andric RetT.makeSigned(); 9780b57cec5SDimitry Andric 979*480093f4SDimitry Andric if (LocalCK == ClassB && RetT.isValue() && !RetT.isScalar()) 9800b57cec5SDimitry Andric // Cast to vector of 8-bit elements. 9810b57cec5SDimitry Andric RetT.makeInteger(8, true); 9820b57cec5SDimitry Andric 9830b57cec5SDimitry Andric S += RetT.builtin_str(); 9840b57cec5SDimitry Andric } 9850b57cec5SDimitry Andric 9860b57cec5SDimitry Andric for (unsigned I = 0; I < getNumParams(); ++I) { 9870b57cec5SDimitry Andric Type T = getParamType(I); 9880b57cec5SDimitry Andric if (T.isPoly()) 9890b57cec5SDimitry Andric T.makeInteger(T.getElementSizeInBits(), false); 9900b57cec5SDimitry Andric 991*480093f4SDimitry Andric if (LocalCK == ClassB && !T.isScalar()) 9920b57cec5SDimitry Andric T.makeInteger(8, true); 9930b57cec5SDimitry Andric // Halves always get converted to 8-bit elements. 9940b57cec5SDimitry Andric if (T.isHalf() && T.isVector() && !T.isScalarForMangling()) 9950b57cec5SDimitry Andric T.makeInteger(8, true); 9960b57cec5SDimitry Andric 997*480093f4SDimitry Andric if (LocalCK == ClassI && T.isInteger()) 9980b57cec5SDimitry Andric T.makeSigned(); 9990b57cec5SDimitry Andric 10000b57cec5SDimitry Andric if (hasImmediate() && getImmediateIdx() == I) 10010b57cec5SDimitry Andric T.makeImmediate(32); 10020b57cec5SDimitry Andric 10030b57cec5SDimitry Andric S += T.builtin_str(); 10040b57cec5SDimitry Andric } 10050b57cec5SDimitry Andric 10060b57cec5SDimitry Andric // Extra constant integer to hold type class enum for this function, e.g. s8 10070b57cec5SDimitry Andric if (LocalCK == ClassB) 10080b57cec5SDimitry Andric S += "i"; 10090b57cec5SDimitry Andric 10100b57cec5SDimitry Andric return S; 10110b57cec5SDimitry Andric } 10120b57cec5SDimitry Andric 10130b57cec5SDimitry Andric std::string Intrinsic::getMangledName(bool ForceClassS) const { 10140b57cec5SDimitry Andric // Check if the prototype has a scalar operand with the type of the vector 10150b57cec5SDimitry Andric // elements. If not, bitcasting the args will take care of arg checking. 10160b57cec5SDimitry Andric // The actual signedness etc. will be taken care of with special enums. 10170b57cec5SDimitry Andric ClassKind LocalCK = CK; 10180b57cec5SDimitry Andric if (!protoHasScalar()) 10190b57cec5SDimitry Andric LocalCK = ClassB; 10200b57cec5SDimitry Andric 10210b57cec5SDimitry Andric return mangleName(Name, ForceClassS ? ClassS : LocalCK); 10220b57cec5SDimitry Andric } 10230b57cec5SDimitry Andric 10240b57cec5SDimitry Andric std::string Intrinsic::mangleName(std::string Name, ClassKind LocalCK) const { 10250b57cec5SDimitry Andric std::string typeCode = getInstTypeCode(BaseType, LocalCK); 10260b57cec5SDimitry Andric std::string S = Name; 10270b57cec5SDimitry Andric 10280b57cec5SDimitry Andric if (Name == "vcvt_f16_f32" || Name == "vcvt_f32_f16" || 10290b57cec5SDimitry Andric Name == "vcvt_f32_f64" || Name == "vcvt_f64_f32") 10300b57cec5SDimitry Andric return Name; 10310b57cec5SDimitry Andric 10320b57cec5SDimitry Andric if (!typeCode.empty()) { 10330b57cec5SDimitry Andric // If the name ends with _xN (N = 2,3,4), insert the typeCode before _xN. 10340b57cec5SDimitry Andric if (Name.size() >= 3 && isdigit(Name.back()) && 10350b57cec5SDimitry Andric Name[Name.length() - 2] == 'x' && Name[Name.length() - 3] == '_') 10360b57cec5SDimitry Andric S.insert(S.length() - 3, "_" + typeCode); 10370b57cec5SDimitry Andric else 10380b57cec5SDimitry Andric S += "_" + typeCode; 10390b57cec5SDimitry Andric } 10400b57cec5SDimitry Andric 10410b57cec5SDimitry Andric if (BaseType != InBaseType) { 10420b57cec5SDimitry Andric // A reinterpret - out the input base type at the end. 10430b57cec5SDimitry Andric S += "_" + getInstTypeCode(InBaseType, LocalCK); 10440b57cec5SDimitry Andric } 10450b57cec5SDimitry Andric 10460b57cec5SDimitry Andric if (LocalCK == ClassB) 10470b57cec5SDimitry Andric S += "_v"; 10480b57cec5SDimitry Andric 10490b57cec5SDimitry Andric // Insert a 'q' before the first '_' character so that it ends up before 10500b57cec5SDimitry Andric // _lane or _n on vector-scalar operations. 10510b57cec5SDimitry Andric if (BaseType.getSizeInBits() == 128 && !BaseType.noManglingQ()) { 10520b57cec5SDimitry Andric size_t Pos = S.find('_'); 10530b57cec5SDimitry Andric S.insert(Pos, "q"); 10540b57cec5SDimitry Andric } 10550b57cec5SDimitry Andric 10560b57cec5SDimitry Andric char Suffix = '\0'; 10570b57cec5SDimitry Andric if (BaseType.isScalarForMangling()) { 10580b57cec5SDimitry Andric switch (BaseType.getElementSizeInBits()) { 10590b57cec5SDimitry Andric case 8: Suffix = 'b'; break; 10600b57cec5SDimitry Andric case 16: Suffix = 'h'; break; 10610b57cec5SDimitry Andric case 32: Suffix = 's'; break; 10620b57cec5SDimitry Andric case 64: Suffix = 'd'; break; 10630b57cec5SDimitry Andric default: llvm_unreachable("Bad suffix!"); 10640b57cec5SDimitry Andric } 10650b57cec5SDimitry Andric } 10660b57cec5SDimitry Andric if (Suffix != '\0') { 10670b57cec5SDimitry Andric size_t Pos = S.find('_'); 10680b57cec5SDimitry Andric S.insert(Pos, &Suffix, 1); 10690b57cec5SDimitry Andric } 10700b57cec5SDimitry Andric 10710b57cec5SDimitry Andric return S; 10720b57cec5SDimitry Andric } 10730b57cec5SDimitry Andric 10740b57cec5SDimitry Andric std::string Intrinsic::replaceParamsIn(std::string S) { 10750b57cec5SDimitry Andric while (S.find('$') != std::string::npos) { 10760b57cec5SDimitry Andric size_t Pos = S.find('$'); 10770b57cec5SDimitry Andric size_t End = Pos + 1; 10780b57cec5SDimitry Andric while (isalpha(S[End])) 10790b57cec5SDimitry Andric ++End; 10800b57cec5SDimitry Andric 10810b57cec5SDimitry Andric std::string VarName = S.substr(Pos + 1, End - Pos - 1); 10820b57cec5SDimitry Andric assert_with_loc(Variables.find(VarName) != Variables.end(), 10830b57cec5SDimitry Andric "Variable not defined!"); 10840b57cec5SDimitry Andric S.replace(Pos, End - Pos, Variables.find(VarName)->second.getName()); 10850b57cec5SDimitry Andric } 10860b57cec5SDimitry Andric 10870b57cec5SDimitry Andric return S; 10880b57cec5SDimitry Andric } 10890b57cec5SDimitry Andric 10900b57cec5SDimitry Andric void Intrinsic::initVariables() { 10910b57cec5SDimitry Andric Variables.clear(); 10920b57cec5SDimitry Andric 10930b57cec5SDimitry Andric // Modify the TypeSpec per-argument to get a concrete Type, and create 10940b57cec5SDimitry Andric // known variables for each. 1095*480093f4SDimitry Andric for (unsigned I = 1; I < Types.size(); ++I) { 10960b57cec5SDimitry Andric char NameC = '0' + (I - 1); 10970b57cec5SDimitry Andric std::string Name = "p"; 10980b57cec5SDimitry Andric Name.push_back(NameC); 10990b57cec5SDimitry Andric 11000b57cec5SDimitry Andric Variables[Name] = Variable(Types[I], Name + VariablePostfix); 11010b57cec5SDimitry Andric } 11020b57cec5SDimitry Andric RetVar = Variable(Types[0], "ret" + VariablePostfix); 11030b57cec5SDimitry Andric } 11040b57cec5SDimitry Andric 11050b57cec5SDimitry Andric void Intrinsic::emitPrototype(StringRef NamePrefix) { 11060b57cec5SDimitry Andric if (UseMacro) 11070b57cec5SDimitry Andric OS << "#define "; 11080b57cec5SDimitry Andric else 11090b57cec5SDimitry Andric OS << "__ai " << Types[0].str() << " "; 11100b57cec5SDimitry Andric 11110b57cec5SDimitry Andric OS << NamePrefix.str() << mangleName(Name, ClassS) << "("; 11120b57cec5SDimitry Andric 11130b57cec5SDimitry Andric for (unsigned I = 0; I < getNumParams(); ++I) { 11140b57cec5SDimitry Andric if (I != 0) 11150b57cec5SDimitry Andric OS << ", "; 11160b57cec5SDimitry Andric 11170b57cec5SDimitry Andric char NameC = '0' + I; 11180b57cec5SDimitry Andric std::string Name = "p"; 11190b57cec5SDimitry Andric Name.push_back(NameC); 11200b57cec5SDimitry Andric assert(Variables.find(Name) != Variables.end()); 11210b57cec5SDimitry Andric Variable &V = Variables[Name]; 11220b57cec5SDimitry Andric 11230b57cec5SDimitry Andric if (!UseMacro) 11240b57cec5SDimitry Andric OS << V.getType().str() << " "; 11250b57cec5SDimitry Andric OS << V.getName(); 11260b57cec5SDimitry Andric } 11270b57cec5SDimitry Andric 11280b57cec5SDimitry Andric OS << ")"; 11290b57cec5SDimitry Andric } 11300b57cec5SDimitry Andric 11310b57cec5SDimitry Andric void Intrinsic::emitOpeningBrace() { 11320b57cec5SDimitry Andric if (UseMacro) 11330b57cec5SDimitry Andric OS << " __extension__ ({"; 11340b57cec5SDimitry Andric else 11350b57cec5SDimitry Andric OS << " {"; 11360b57cec5SDimitry Andric emitNewLine(); 11370b57cec5SDimitry Andric } 11380b57cec5SDimitry Andric 11390b57cec5SDimitry Andric void Intrinsic::emitClosingBrace() { 11400b57cec5SDimitry Andric if (UseMacro) 11410b57cec5SDimitry Andric OS << "})"; 11420b57cec5SDimitry Andric else 11430b57cec5SDimitry Andric OS << "}"; 11440b57cec5SDimitry Andric } 11450b57cec5SDimitry Andric 11460b57cec5SDimitry Andric void Intrinsic::emitNewLine() { 11470b57cec5SDimitry Andric if (UseMacro) 11480b57cec5SDimitry Andric OS << " \\\n"; 11490b57cec5SDimitry Andric else 11500b57cec5SDimitry Andric OS << "\n"; 11510b57cec5SDimitry Andric } 11520b57cec5SDimitry Andric 11530b57cec5SDimitry Andric void Intrinsic::emitReverseVariable(Variable &Dest, Variable &Src) { 11540b57cec5SDimitry Andric if (Dest.getType().getNumVectors() > 1) { 11550b57cec5SDimitry Andric emitNewLine(); 11560b57cec5SDimitry Andric 11570b57cec5SDimitry Andric for (unsigned K = 0; K < Dest.getType().getNumVectors(); ++K) { 11580b57cec5SDimitry Andric OS << " " << Dest.getName() << ".val[" << K << "] = " 11590b57cec5SDimitry Andric << "__builtin_shufflevector(" 11600b57cec5SDimitry Andric << Src.getName() << ".val[" << K << "], " 11610b57cec5SDimitry Andric << Src.getName() << ".val[" << K << "]"; 11620b57cec5SDimitry Andric for (int J = Dest.getType().getNumElements() - 1; J >= 0; --J) 11630b57cec5SDimitry Andric OS << ", " << J; 11640b57cec5SDimitry Andric OS << ");"; 11650b57cec5SDimitry Andric emitNewLine(); 11660b57cec5SDimitry Andric } 11670b57cec5SDimitry Andric } else { 11680b57cec5SDimitry Andric OS << " " << Dest.getName() 11690b57cec5SDimitry Andric << " = __builtin_shufflevector(" << Src.getName() << ", " << Src.getName(); 11700b57cec5SDimitry Andric for (int J = Dest.getType().getNumElements() - 1; J >= 0; --J) 11710b57cec5SDimitry Andric OS << ", " << J; 11720b57cec5SDimitry Andric OS << ");"; 11730b57cec5SDimitry Andric emitNewLine(); 11740b57cec5SDimitry Andric } 11750b57cec5SDimitry Andric } 11760b57cec5SDimitry Andric 11770b57cec5SDimitry Andric void Intrinsic::emitArgumentReversal() { 1178a7dea167SDimitry Andric if (isBigEndianSafe()) 11790b57cec5SDimitry Andric return; 11800b57cec5SDimitry Andric 11810b57cec5SDimitry Andric // Reverse all vector arguments. 11820b57cec5SDimitry Andric for (unsigned I = 0; I < getNumParams(); ++I) { 11830b57cec5SDimitry Andric std::string Name = "p" + utostr(I); 11840b57cec5SDimitry Andric std::string NewName = "rev" + utostr(I); 11850b57cec5SDimitry Andric 11860b57cec5SDimitry Andric Variable &V = Variables[Name]; 11870b57cec5SDimitry Andric Variable NewV(V.getType(), NewName + VariablePostfix); 11880b57cec5SDimitry Andric 11890b57cec5SDimitry Andric if (!NewV.getType().isVector() || NewV.getType().getNumElements() == 1) 11900b57cec5SDimitry Andric continue; 11910b57cec5SDimitry Andric 11920b57cec5SDimitry Andric OS << " " << NewV.getType().str() << " " << NewV.getName() << ";"; 11930b57cec5SDimitry Andric emitReverseVariable(NewV, V); 11940b57cec5SDimitry Andric V = NewV; 11950b57cec5SDimitry Andric } 11960b57cec5SDimitry Andric } 11970b57cec5SDimitry Andric 11980b57cec5SDimitry Andric void Intrinsic::emitReturnReversal() { 1199a7dea167SDimitry Andric if (isBigEndianSafe()) 12000b57cec5SDimitry Andric return; 12010b57cec5SDimitry Andric if (!getReturnType().isVector() || getReturnType().isVoid() || 12020b57cec5SDimitry Andric getReturnType().getNumElements() == 1) 12030b57cec5SDimitry Andric return; 12040b57cec5SDimitry Andric emitReverseVariable(RetVar, RetVar); 12050b57cec5SDimitry Andric } 12060b57cec5SDimitry Andric 12070b57cec5SDimitry Andric void Intrinsic::emitShadowedArgs() { 12080b57cec5SDimitry Andric // Macro arguments are not type-checked like inline function arguments, 12090b57cec5SDimitry Andric // so assign them to local temporaries to get the right type checking. 12100b57cec5SDimitry Andric if (!UseMacro) 12110b57cec5SDimitry Andric return; 12120b57cec5SDimitry Andric 12130b57cec5SDimitry Andric for (unsigned I = 0; I < getNumParams(); ++I) { 12140b57cec5SDimitry Andric // Do not create a temporary for an immediate argument. 12150b57cec5SDimitry Andric // That would defeat the whole point of using a macro! 1216*480093f4SDimitry Andric if (getParamType(I).isImmediate()) 12170b57cec5SDimitry Andric continue; 12180b57cec5SDimitry Andric // Do not create a temporary for pointer arguments. The input 12190b57cec5SDimitry Andric // pointer may have an alignment hint. 12200b57cec5SDimitry Andric if (getParamType(I).isPointer()) 12210b57cec5SDimitry Andric continue; 12220b57cec5SDimitry Andric 12230b57cec5SDimitry Andric std::string Name = "p" + utostr(I); 12240b57cec5SDimitry Andric 12250b57cec5SDimitry Andric assert(Variables.find(Name) != Variables.end()); 12260b57cec5SDimitry Andric Variable &V = Variables[Name]; 12270b57cec5SDimitry Andric 12280b57cec5SDimitry Andric std::string NewName = "s" + utostr(I); 12290b57cec5SDimitry Andric Variable V2(V.getType(), NewName + VariablePostfix); 12300b57cec5SDimitry Andric 12310b57cec5SDimitry Andric OS << " " << V2.getType().str() << " " << V2.getName() << " = " 12320b57cec5SDimitry Andric << V.getName() << ";"; 12330b57cec5SDimitry Andric emitNewLine(); 12340b57cec5SDimitry Andric 12350b57cec5SDimitry Andric V = V2; 12360b57cec5SDimitry Andric } 12370b57cec5SDimitry Andric } 12380b57cec5SDimitry Andric 12390b57cec5SDimitry Andric bool Intrinsic::protoHasScalar() const { 1240*480093f4SDimitry Andric return std::any_of(Types.begin(), Types.end(), [](const Type &T) { 1241*480093f4SDimitry Andric return T.isScalar() && !T.isImmediate(); 1242*480093f4SDimitry Andric }); 12430b57cec5SDimitry Andric } 12440b57cec5SDimitry Andric 12450b57cec5SDimitry Andric void Intrinsic::emitBodyAsBuiltinCall() { 12460b57cec5SDimitry Andric std::string S; 12470b57cec5SDimitry Andric 12480b57cec5SDimitry Andric // If this builtin returns a struct 2, 3, or 4 vectors, pass it as an implicit 12490b57cec5SDimitry Andric // sret-like argument. 12500b57cec5SDimitry Andric bool SRet = getReturnType().getNumVectors() >= 2; 12510b57cec5SDimitry Andric 12520b57cec5SDimitry Andric StringRef N = Name; 12530b57cec5SDimitry Andric ClassKind LocalCK = CK; 12540b57cec5SDimitry Andric if (!protoHasScalar()) 12550b57cec5SDimitry Andric LocalCK = ClassB; 12560b57cec5SDimitry Andric 12570b57cec5SDimitry Andric if (!getReturnType().isVoid() && !SRet) 12580b57cec5SDimitry Andric S += "(" + RetVar.getType().str() + ") "; 12590b57cec5SDimitry Andric 12600b57cec5SDimitry Andric S += "__builtin_neon_" + mangleName(N, LocalCK) + "("; 12610b57cec5SDimitry Andric 12620b57cec5SDimitry Andric if (SRet) 12630b57cec5SDimitry Andric S += "&" + RetVar.getName() + ", "; 12640b57cec5SDimitry Andric 12650b57cec5SDimitry Andric for (unsigned I = 0; I < getNumParams(); ++I) { 12660b57cec5SDimitry Andric Variable &V = Variables["p" + utostr(I)]; 12670b57cec5SDimitry Andric Type T = V.getType(); 12680b57cec5SDimitry Andric 12690b57cec5SDimitry Andric // Handle multiple-vector values specially, emitting each subvector as an 12700b57cec5SDimitry Andric // argument to the builtin. 12710b57cec5SDimitry Andric if (T.getNumVectors() > 1) { 12720b57cec5SDimitry Andric // Check if an explicit cast is needed. 12730b57cec5SDimitry Andric std::string Cast; 1274a7dea167SDimitry Andric if (LocalCK == ClassB) { 12750b57cec5SDimitry Andric Type T2 = T; 12760b57cec5SDimitry Andric T2.makeOneVector(); 12770b57cec5SDimitry Andric T2.makeInteger(8, /*Signed=*/true); 12780b57cec5SDimitry Andric Cast = "(" + T2.str() + ")"; 12790b57cec5SDimitry Andric } 12800b57cec5SDimitry Andric 12810b57cec5SDimitry Andric for (unsigned J = 0; J < T.getNumVectors(); ++J) 12820b57cec5SDimitry Andric S += Cast + V.getName() + ".val[" + utostr(J) + "], "; 12830b57cec5SDimitry Andric continue; 12840b57cec5SDimitry Andric } 12850b57cec5SDimitry Andric 1286*480093f4SDimitry Andric std::string Arg = V.getName(); 12870b57cec5SDimitry Andric Type CastToType = T; 12880b57cec5SDimitry Andric 12890b57cec5SDimitry Andric // Check if an explicit cast is needed. 1290a7dea167SDimitry Andric if (CastToType.isVector() && 1291a7dea167SDimitry Andric (LocalCK == ClassB || (T.isHalf() && !T.isScalarForMangling()))) { 12920b57cec5SDimitry Andric CastToType.makeInteger(8, true); 12930b57cec5SDimitry Andric Arg = "(" + CastToType.str() + ")" + Arg; 1294a7dea167SDimitry Andric } else if (CastToType.isVector() && LocalCK == ClassI) { 1295*480093f4SDimitry Andric if (CastToType.isInteger()) 1296a7dea167SDimitry Andric CastToType.makeSigned(); 1297a7dea167SDimitry Andric Arg = "(" + CastToType.str() + ")" + Arg; 12980b57cec5SDimitry Andric } 12990b57cec5SDimitry Andric 13000b57cec5SDimitry Andric S += Arg + ", "; 13010b57cec5SDimitry Andric } 13020b57cec5SDimitry Andric 13030b57cec5SDimitry Andric // Extra constant integer to hold type class enum for this function, e.g. s8 13040b57cec5SDimitry Andric if (getClassKind(true) == ClassB) { 1305*480093f4SDimitry Andric S += utostr(getPolymorphicKeyType().getNeonEnum()); 13060b57cec5SDimitry Andric } else { 13070b57cec5SDimitry Andric // Remove extraneous ", ". 13080b57cec5SDimitry Andric S.pop_back(); 13090b57cec5SDimitry Andric S.pop_back(); 13100b57cec5SDimitry Andric } 13110b57cec5SDimitry Andric S += ");"; 13120b57cec5SDimitry Andric 13130b57cec5SDimitry Andric std::string RetExpr; 13140b57cec5SDimitry Andric if (!SRet && !RetVar.getType().isVoid()) 13150b57cec5SDimitry Andric RetExpr = RetVar.getName() + " = "; 13160b57cec5SDimitry Andric 13170b57cec5SDimitry Andric OS << " " << RetExpr << S; 13180b57cec5SDimitry Andric emitNewLine(); 13190b57cec5SDimitry Andric } 13200b57cec5SDimitry Andric 13210b57cec5SDimitry Andric void Intrinsic::emitBody(StringRef CallPrefix) { 13220b57cec5SDimitry Andric std::vector<std::string> Lines; 13230b57cec5SDimitry Andric 13240b57cec5SDimitry Andric assert(RetVar.getType() == Types[0]); 13250b57cec5SDimitry Andric // Create a return variable, if we're not void. 13260b57cec5SDimitry Andric if (!RetVar.getType().isVoid()) { 13270b57cec5SDimitry Andric OS << " " << RetVar.getType().str() << " " << RetVar.getName() << ";"; 13280b57cec5SDimitry Andric emitNewLine(); 13290b57cec5SDimitry Andric } 13300b57cec5SDimitry Andric 13310b57cec5SDimitry Andric if (!Body || Body->getValues().empty()) { 13320b57cec5SDimitry Andric // Nothing specific to output - must output a builtin. 13330b57cec5SDimitry Andric emitBodyAsBuiltinCall(); 13340b57cec5SDimitry Andric return; 13350b57cec5SDimitry Andric } 13360b57cec5SDimitry Andric 13370b57cec5SDimitry Andric // We have a list of "things to output". The last should be returned. 13380b57cec5SDimitry Andric for (auto *I : Body->getValues()) { 13390b57cec5SDimitry Andric if (StringInit *SI = dyn_cast<StringInit>(I)) { 13400b57cec5SDimitry Andric Lines.push_back(replaceParamsIn(SI->getAsString())); 13410b57cec5SDimitry Andric } else if (DagInit *DI = dyn_cast<DagInit>(I)) { 13420b57cec5SDimitry Andric DagEmitter DE(*this, CallPrefix); 13430b57cec5SDimitry Andric Lines.push_back(DE.emitDag(DI).second + ";"); 13440b57cec5SDimitry Andric } 13450b57cec5SDimitry Andric } 13460b57cec5SDimitry Andric 13470b57cec5SDimitry Andric assert(!Lines.empty() && "Empty def?"); 13480b57cec5SDimitry Andric if (!RetVar.getType().isVoid()) 13490b57cec5SDimitry Andric Lines.back().insert(0, RetVar.getName() + " = "); 13500b57cec5SDimitry Andric 13510b57cec5SDimitry Andric for (auto &L : Lines) { 13520b57cec5SDimitry Andric OS << " " << L; 13530b57cec5SDimitry Andric emitNewLine(); 13540b57cec5SDimitry Andric } 13550b57cec5SDimitry Andric } 13560b57cec5SDimitry Andric 13570b57cec5SDimitry Andric void Intrinsic::emitReturn() { 13580b57cec5SDimitry Andric if (RetVar.getType().isVoid()) 13590b57cec5SDimitry Andric return; 13600b57cec5SDimitry Andric if (UseMacro) 13610b57cec5SDimitry Andric OS << " " << RetVar.getName() << ";"; 13620b57cec5SDimitry Andric else 13630b57cec5SDimitry Andric OS << " return " << RetVar.getName() << ";"; 13640b57cec5SDimitry Andric emitNewLine(); 13650b57cec5SDimitry Andric } 13660b57cec5SDimitry Andric 13670b57cec5SDimitry Andric std::pair<Type, std::string> Intrinsic::DagEmitter::emitDag(DagInit *DI) { 13680b57cec5SDimitry Andric // At this point we should only be seeing a def. 13690b57cec5SDimitry Andric DefInit *DefI = cast<DefInit>(DI->getOperator()); 13700b57cec5SDimitry Andric std::string Op = DefI->getAsString(); 13710b57cec5SDimitry Andric 13720b57cec5SDimitry Andric if (Op == "cast" || Op == "bitcast") 13730b57cec5SDimitry Andric return emitDagCast(DI, Op == "bitcast"); 13740b57cec5SDimitry Andric if (Op == "shuffle") 13750b57cec5SDimitry Andric return emitDagShuffle(DI); 13760b57cec5SDimitry Andric if (Op == "dup") 13770b57cec5SDimitry Andric return emitDagDup(DI); 13780b57cec5SDimitry Andric if (Op == "dup_typed") 13790b57cec5SDimitry Andric return emitDagDupTyped(DI); 13800b57cec5SDimitry Andric if (Op == "splat") 13810b57cec5SDimitry Andric return emitDagSplat(DI); 13820b57cec5SDimitry Andric if (Op == "save_temp") 13830b57cec5SDimitry Andric return emitDagSaveTemp(DI); 13840b57cec5SDimitry Andric if (Op == "op") 13850b57cec5SDimitry Andric return emitDagOp(DI); 13860b57cec5SDimitry Andric if (Op == "call") 13870b57cec5SDimitry Andric return emitDagCall(DI); 13880b57cec5SDimitry Andric if (Op == "name_replace") 13890b57cec5SDimitry Andric return emitDagNameReplace(DI); 13900b57cec5SDimitry Andric if (Op == "literal") 13910b57cec5SDimitry Andric return emitDagLiteral(DI); 13920b57cec5SDimitry Andric assert_with_loc(false, "Unknown operation!"); 13930b57cec5SDimitry Andric return std::make_pair(Type::getVoid(), ""); 13940b57cec5SDimitry Andric } 13950b57cec5SDimitry Andric 13960b57cec5SDimitry Andric std::pair<Type, std::string> Intrinsic::DagEmitter::emitDagOp(DagInit *DI) { 13970b57cec5SDimitry Andric std::string Op = cast<StringInit>(DI->getArg(0))->getAsUnquotedString(); 13980b57cec5SDimitry Andric if (DI->getNumArgs() == 2) { 13990b57cec5SDimitry Andric // Unary op. 14000b57cec5SDimitry Andric std::pair<Type, std::string> R = 14010b57cec5SDimitry Andric emitDagArg(DI->getArg(1), DI->getArgNameStr(1)); 14020b57cec5SDimitry Andric return std::make_pair(R.first, Op + R.second); 14030b57cec5SDimitry Andric } else { 14040b57cec5SDimitry Andric assert(DI->getNumArgs() == 3 && "Can only handle unary and binary ops!"); 14050b57cec5SDimitry Andric std::pair<Type, std::string> R1 = 14060b57cec5SDimitry Andric emitDagArg(DI->getArg(1), DI->getArgNameStr(1)); 14070b57cec5SDimitry Andric std::pair<Type, std::string> R2 = 14080b57cec5SDimitry Andric emitDagArg(DI->getArg(2), DI->getArgNameStr(2)); 14090b57cec5SDimitry Andric assert_with_loc(R1.first == R2.first, "Argument type mismatch!"); 14100b57cec5SDimitry Andric return std::make_pair(R1.first, R1.second + " " + Op + " " + R2.second); 14110b57cec5SDimitry Andric } 14120b57cec5SDimitry Andric } 14130b57cec5SDimitry Andric 14140b57cec5SDimitry Andric std::pair<Type, std::string> Intrinsic::DagEmitter::emitDagCall(DagInit *DI) { 14150b57cec5SDimitry Andric std::vector<Type> Types; 14160b57cec5SDimitry Andric std::vector<std::string> Values; 14170b57cec5SDimitry Andric for (unsigned I = 0; I < DI->getNumArgs() - 1; ++I) { 14180b57cec5SDimitry Andric std::pair<Type, std::string> R = 14190b57cec5SDimitry Andric emitDagArg(DI->getArg(I + 1), DI->getArgNameStr(I + 1)); 14200b57cec5SDimitry Andric Types.push_back(R.first); 14210b57cec5SDimitry Andric Values.push_back(R.second); 14220b57cec5SDimitry Andric } 14230b57cec5SDimitry Andric 14240b57cec5SDimitry Andric // Look up the called intrinsic. 14250b57cec5SDimitry Andric std::string N; 14260b57cec5SDimitry Andric if (StringInit *SI = dyn_cast<StringInit>(DI->getArg(0))) 14270b57cec5SDimitry Andric N = SI->getAsUnquotedString(); 14280b57cec5SDimitry Andric else 14290b57cec5SDimitry Andric N = emitDagArg(DI->getArg(0), "").second; 14300b57cec5SDimitry Andric Intrinsic &Callee = Intr.Emitter.getIntrinsic(N, Types); 14310b57cec5SDimitry Andric 14320b57cec5SDimitry Andric // Make sure the callee is known as an early def. 14330b57cec5SDimitry Andric Callee.setNeededEarly(); 14340b57cec5SDimitry Andric Intr.Dependencies.insert(&Callee); 14350b57cec5SDimitry Andric 14360b57cec5SDimitry Andric // Now create the call itself. 1437a7dea167SDimitry Andric std::string S = ""; 1438a7dea167SDimitry Andric if (!Callee.isBigEndianSafe()) 1439a7dea167SDimitry Andric S += CallPrefix.str(); 1440a7dea167SDimitry Andric S += Callee.getMangledName(true) + "("; 14410b57cec5SDimitry Andric for (unsigned I = 0; I < DI->getNumArgs() - 1; ++I) { 14420b57cec5SDimitry Andric if (I != 0) 14430b57cec5SDimitry Andric S += ", "; 14440b57cec5SDimitry Andric S += Values[I]; 14450b57cec5SDimitry Andric } 14460b57cec5SDimitry Andric S += ")"; 14470b57cec5SDimitry Andric 14480b57cec5SDimitry Andric return std::make_pair(Callee.getReturnType(), S); 14490b57cec5SDimitry Andric } 14500b57cec5SDimitry Andric 14510b57cec5SDimitry Andric std::pair<Type, std::string> Intrinsic::DagEmitter::emitDagCast(DagInit *DI, 14520b57cec5SDimitry Andric bool IsBitCast){ 14530b57cec5SDimitry Andric // (cast MOD* VAL) -> cast VAL to type given by MOD. 14540b57cec5SDimitry Andric std::pair<Type, std::string> R = emitDagArg( 14550b57cec5SDimitry Andric DI->getArg(DI->getNumArgs() - 1), 14560b57cec5SDimitry Andric DI->getArgNameStr(DI->getNumArgs() - 1)); 14570b57cec5SDimitry Andric Type castToType = R.first; 14580b57cec5SDimitry Andric for (unsigned ArgIdx = 0; ArgIdx < DI->getNumArgs() - 1; ++ArgIdx) { 14590b57cec5SDimitry Andric 14600b57cec5SDimitry Andric // MOD can take several forms: 14610b57cec5SDimitry Andric // 1. $X - take the type of parameter / variable X. 14620b57cec5SDimitry Andric // 2. The value "R" - take the type of the return type. 14630b57cec5SDimitry Andric // 3. a type string 14640b57cec5SDimitry Andric // 4. The value "U" or "S" to switch the signedness. 14650b57cec5SDimitry Andric // 5. The value "H" or "D" to half or double the bitwidth. 14660b57cec5SDimitry Andric // 6. The value "8" to convert to 8-bit (signed) integer lanes. 14670b57cec5SDimitry Andric if (!DI->getArgNameStr(ArgIdx).empty()) { 14680b57cec5SDimitry Andric assert_with_loc(Intr.Variables.find(DI->getArgNameStr(ArgIdx)) != 14690b57cec5SDimitry Andric Intr.Variables.end(), 14700b57cec5SDimitry Andric "Variable not found"); 14710b57cec5SDimitry Andric castToType = Intr.Variables[DI->getArgNameStr(ArgIdx)].getType(); 14720b57cec5SDimitry Andric } else { 14730b57cec5SDimitry Andric StringInit *SI = dyn_cast<StringInit>(DI->getArg(ArgIdx)); 14740b57cec5SDimitry Andric assert_with_loc(SI, "Expected string type or $Name for cast type"); 14750b57cec5SDimitry Andric 14760b57cec5SDimitry Andric if (SI->getAsUnquotedString() == "R") { 14770b57cec5SDimitry Andric castToType = Intr.getReturnType(); 14780b57cec5SDimitry Andric } else if (SI->getAsUnquotedString() == "U") { 14790b57cec5SDimitry Andric castToType.makeUnsigned(); 14800b57cec5SDimitry Andric } else if (SI->getAsUnquotedString() == "S") { 14810b57cec5SDimitry Andric castToType.makeSigned(); 14820b57cec5SDimitry Andric } else if (SI->getAsUnquotedString() == "H") { 14830b57cec5SDimitry Andric castToType.halveLanes(); 14840b57cec5SDimitry Andric } else if (SI->getAsUnquotedString() == "D") { 14850b57cec5SDimitry Andric castToType.doubleLanes(); 14860b57cec5SDimitry Andric } else if (SI->getAsUnquotedString() == "8") { 14870b57cec5SDimitry Andric castToType.makeInteger(8, true); 14880b57cec5SDimitry Andric } else { 14890b57cec5SDimitry Andric castToType = Type::fromTypedefName(SI->getAsUnquotedString()); 14900b57cec5SDimitry Andric assert_with_loc(!castToType.isVoid(), "Unknown typedef"); 14910b57cec5SDimitry Andric } 14920b57cec5SDimitry Andric } 14930b57cec5SDimitry Andric } 14940b57cec5SDimitry Andric 14950b57cec5SDimitry Andric std::string S; 14960b57cec5SDimitry Andric if (IsBitCast) { 14970b57cec5SDimitry Andric // Emit a reinterpret cast. The second operand must be an lvalue, so create 14980b57cec5SDimitry Andric // a temporary. 14990b57cec5SDimitry Andric std::string N = "reint"; 15000b57cec5SDimitry Andric unsigned I = 0; 15010b57cec5SDimitry Andric while (Intr.Variables.find(N) != Intr.Variables.end()) 15020b57cec5SDimitry Andric N = "reint" + utostr(++I); 15030b57cec5SDimitry Andric Intr.Variables[N] = Variable(R.first, N + Intr.VariablePostfix); 15040b57cec5SDimitry Andric 15050b57cec5SDimitry Andric Intr.OS << R.first.str() << " " << Intr.Variables[N].getName() << " = " 15060b57cec5SDimitry Andric << R.second << ";"; 15070b57cec5SDimitry Andric Intr.emitNewLine(); 15080b57cec5SDimitry Andric 15090b57cec5SDimitry Andric S = "*(" + castToType.str() + " *) &" + Intr.Variables[N].getName() + ""; 15100b57cec5SDimitry Andric } else { 15110b57cec5SDimitry Andric // Emit a normal (static) cast. 15120b57cec5SDimitry Andric S = "(" + castToType.str() + ")(" + R.second + ")"; 15130b57cec5SDimitry Andric } 15140b57cec5SDimitry Andric 15150b57cec5SDimitry Andric return std::make_pair(castToType, S); 15160b57cec5SDimitry Andric } 15170b57cec5SDimitry Andric 15180b57cec5SDimitry Andric std::pair<Type, std::string> Intrinsic::DagEmitter::emitDagShuffle(DagInit *DI){ 15190b57cec5SDimitry Andric // See the documentation in arm_neon.td for a description of these operators. 15200b57cec5SDimitry Andric class LowHalf : public SetTheory::Operator { 15210b57cec5SDimitry Andric public: 15220b57cec5SDimitry Andric void apply(SetTheory &ST, DagInit *Expr, SetTheory::RecSet &Elts, 15230b57cec5SDimitry Andric ArrayRef<SMLoc> Loc) override { 15240b57cec5SDimitry Andric SetTheory::RecSet Elts2; 15250b57cec5SDimitry Andric ST.evaluate(Expr->arg_begin(), Expr->arg_end(), Elts2, Loc); 15260b57cec5SDimitry Andric Elts.insert(Elts2.begin(), Elts2.begin() + (Elts2.size() / 2)); 15270b57cec5SDimitry Andric } 15280b57cec5SDimitry Andric }; 15290b57cec5SDimitry Andric 15300b57cec5SDimitry Andric class HighHalf : public SetTheory::Operator { 15310b57cec5SDimitry Andric public: 15320b57cec5SDimitry Andric void apply(SetTheory &ST, DagInit *Expr, SetTheory::RecSet &Elts, 15330b57cec5SDimitry Andric ArrayRef<SMLoc> Loc) override { 15340b57cec5SDimitry Andric SetTheory::RecSet Elts2; 15350b57cec5SDimitry Andric ST.evaluate(Expr->arg_begin(), Expr->arg_end(), Elts2, Loc); 15360b57cec5SDimitry Andric Elts.insert(Elts2.begin() + (Elts2.size() / 2), Elts2.end()); 15370b57cec5SDimitry Andric } 15380b57cec5SDimitry Andric }; 15390b57cec5SDimitry Andric 15400b57cec5SDimitry Andric class Rev : public SetTheory::Operator { 15410b57cec5SDimitry Andric unsigned ElementSize; 15420b57cec5SDimitry Andric 15430b57cec5SDimitry Andric public: 15440b57cec5SDimitry Andric Rev(unsigned ElementSize) : ElementSize(ElementSize) {} 15450b57cec5SDimitry Andric 15460b57cec5SDimitry Andric void apply(SetTheory &ST, DagInit *Expr, SetTheory::RecSet &Elts, 15470b57cec5SDimitry Andric ArrayRef<SMLoc> Loc) override { 15480b57cec5SDimitry Andric SetTheory::RecSet Elts2; 15490b57cec5SDimitry Andric ST.evaluate(Expr->arg_begin() + 1, Expr->arg_end(), Elts2, Loc); 15500b57cec5SDimitry Andric 15510b57cec5SDimitry Andric int64_t VectorSize = cast<IntInit>(Expr->getArg(0))->getValue(); 15520b57cec5SDimitry Andric VectorSize /= ElementSize; 15530b57cec5SDimitry Andric 15540b57cec5SDimitry Andric std::vector<Record *> Revved; 15550b57cec5SDimitry Andric for (unsigned VI = 0; VI < Elts2.size(); VI += VectorSize) { 15560b57cec5SDimitry Andric for (int LI = VectorSize - 1; LI >= 0; --LI) { 15570b57cec5SDimitry Andric Revved.push_back(Elts2[VI + LI]); 15580b57cec5SDimitry Andric } 15590b57cec5SDimitry Andric } 15600b57cec5SDimitry Andric 15610b57cec5SDimitry Andric Elts.insert(Revved.begin(), Revved.end()); 15620b57cec5SDimitry Andric } 15630b57cec5SDimitry Andric }; 15640b57cec5SDimitry Andric 15650b57cec5SDimitry Andric class MaskExpander : public SetTheory::Expander { 15660b57cec5SDimitry Andric unsigned N; 15670b57cec5SDimitry Andric 15680b57cec5SDimitry Andric public: 15690b57cec5SDimitry Andric MaskExpander(unsigned N) : N(N) {} 15700b57cec5SDimitry Andric 15710b57cec5SDimitry Andric void expand(SetTheory &ST, Record *R, SetTheory::RecSet &Elts) override { 15720b57cec5SDimitry Andric unsigned Addend = 0; 15730b57cec5SDimitry Andric if (R->getName() == "mask0") 15740b57cec5SDimitry Andric Addend = 0; 15750b57cec5SDimitry Andric else if (R->getName() == "mask1") 15760b57cec5SDimitry Andric Addend = N; 15770b57cec5SDimitry Andric else 15780b57cec5SDimitry Andric return; 15790b57cec5SDimitry Andric for (unsigned I = 0; I < N; ++I) 15800b57cec5SDimitry Andric Elts.insert(R->getRecords().getDef("sv" + utostr(I + Addend))); 15810b57cec5SDimitry Andric } 15820b57cec5SDimitry Andric }; 15830b57cec5SDimitry Andric 15840b57cec5SDimitry Andric // (shuffle arg1, arg2, sequence) 15850b57cec5SDimitry Andric std::pair<Type, std::string> Arg1 = 15860b57cec5SDimitry Andric emitDagArg(DI->getArg(0), DI->getArgNameStr(0)); 15870b57cec5SDimitry Andric std::pair<Type, std::string> Arg2 = 15880b57cec5SDimitry Andric emitDagArg(DI->getArg(1), DI->getArgNameStr(1)); 15890b57cec5SDimitry Andric assert_with_loc(Arg1.first == Arg2.first, 15900b57cec5SDimitry Andric "Different types in arguments to shuffle!"); 15910b57cec5SDimitry Andric 15920b57cec5SDimitry Andric SetTheory ST; 15930b57cec5SDimitry Andric SetTheory::RecSet Elts; 1594a7dea167SDimitry Andric ST.addOperator("lowhalf", std::make_unique<LowHalf>()); 1595a7dea167SDimitry Andric ST.addOperator("highhalf", std::make_unique<HighHalf>()); 15960b57cec5SDimitry Andric ST.addOperator("rev", 1597a7dea167SDimitry Andric std::make_unique<Rev>(Arg1.first.getElementSizeInBits())); 15980b57cec5SDimitry Andric ST.addExpander("MaskExpand", 1599a7dea167SDimitry Andric std::make_unique<MaskExpander>(Arg1.first.getNumElements())); 16000b57cec5SDimitry Andric ST.evaluate(DI->getArg(2), Elts, None); 16010b57cec5SDimitry Andric 16020b57cec5SDimitry Andric std::string S = "__builtin_shufflevector(" + Arg1.second + ", " + Arg2.second; 16030b57cec5SDimitry Andric for (auto &E : Elts) { 16040b57cec5SDimitry Andric StringRef Name = E->getName(); 16050b57cec5SDimitry Andric assert_with_loc(Name.startswith("sv"), 16060b57cec5SDimitry Andric "Incorrect element kind in shuffle mask!"); 16070b57cec5SDimitry Andric S += ", " + Name.drop_front(2).str(); 16080b57cec5SDimitry Andric } 16090b57cec5SDimitry Andric S += ")"; 16100b57cec5SDimitry Andric 16110b57cec5SDimitry Andric // Recalculate the return type - the shuffle may have halved or doubled it. 16120b57cec5SDimitry Andric Type T(Arg1.first); 16130b57cec5SDimitry Andric if (Elts.size() > T.getNumElements()) { 16140b57cec5SDimitry Andric assert_with_loc( 16150b57cec5SDimitry Andric Elts.size() == T.getNumElements() * 2, 16160b57cec5SDimitry Andric "Can only double or half the number of elements in a shuffle!"); 16170b57cec5SDimitry Andric T.doubleLanes(); 16180b57cec5SDimitry Andric } else if (Elts.size() < T.getNumElements()) { 16190b57cec5SDimitry Andric assert_with_loc( 16200b57cec5SDimitry Andric Elts.size() == T.getNumElements() / 2, 16210b57cec5SDimitry Andric "Can only double or half the number of elements in a shuffle!"); 16220b57cec5SDimitry Andric T.halveLanes(); 16230b57cec5SDimitry Andric } 16240b57cec5SDimitry Andric 16250b57cec5SDimitry Andric return std::make_pair(T, S); 16260b57cec5SDimitry Andric } 16270b57cec5SDimitry Andric 16280b57cec5SDimitry Andric std::pair<Type, std::string> Intrinsic::DagEmitter::emitDagDup(DagInit *DI) { 16290b57cec5SDimitry Andric assert_with_loc(DI->getNumArgs() == 1, "dup() expects one argument"); 16300b57cec5SDimitry Andric std::pair<Type, std::string> A = emitDagArg(DI->getArg(0), 16310b57cec5SDimitry Andric DI->getArgNameStr(0)); 16320b57cec5SDimitry Andric assert_with_loc(A.first.isScalar(), "dup() expects a scalar argument"); 16330b57cec5SDimitry Andric 16340b57cec5SDimitry Andric Type T = Intr.getBaseType(); 16350b57cec5SDimitry Andric assert_with_loc(T.isVector(), "dup() used but default type is scalar!"); 16360b57cec5SDimitry Andric std::string S = "(" + T.str() + ") {"; 16370b57cec5SDimitry Andric for (unsigned I = 0; I < T.getNumElements(); ++I) { 16380b57cec5SDimitry Andric if (I != 0) 16390b57cec5SDimitry Andric S += ", "; 16400b57cec5SDimitry Andric S += A.second; 16410b57cec5SDimitry Andric } 16420b57cec5SDimitry Andric S += "}"; 16430b57cec5SDimitry Andric 16440b57cec5SDimitry Andric return std::make_pair(T, S); 16450b57cec5SDimitry Andric } 16460b57cec5SDimitry Andric 16470b57cec5SDimitry Andric std::pair<Type, std::string> Intrinsic::DagEmitter::emitDagDupTyped(DagInit *DI) { 16480b57cec5SDimitry Andric assert_with_loc(DI->getNumArgs() == 2, "dup_typed() expects two arguments"); 16490b57cec5SDimitry Andric std::pair<Type, std::string> A = emitDagArg(DI->getArg(0), 16500b57cec5SDimitry Andric DI->getArgNameStr(0)); 16510b57cec5SDimitry Andric std::pair<Type, std::string> B = emitDagArg(DI->getArg(1), 16520b57cec5SDimitry Andric DI->getArgNameStr(1)); 16530b57cec5SDimitry Andric assert_with_loc(B.first.isScalar(), 16540b57cec5SDimitry Andric "dup_typed() requires a scalar as the second argument"); 16550b57cec5SDimitry Andric 16560b57cec5SDimitry Andric Type T = A.first; 16570b57cec5SDimitry Andric assert_with_loc(T.isVector(), "dup_typed() used but target type is scalar!"); 16580b57cec5SDimitry Andric std::string S = "(" + T.str() + ") {"; 16590b57cec5SDimitry Andric for (unsigned I = 0; I < T.getNumElements(); ++I) { 16600b57cec5SDimitry Andric if (I != 0) 16610b57cec5SDimitry Andric S += ", "; 16620b57cec5SDimitry Andric S += B.second; 16630b57cec5SDimitry Andric } 16640b57cec5SDimitry Andric S += "}"; 16650b57cec5SDimitry Andric 16660b57cec5SDimitry Andric return std::make_pair(T, S); 16670b57cec5SDimitry Andric } 16680b57cec5SDimitry Andric 16690b57cec5SDimitry Andric std::pair<Type, std::string> Intrinsic::DagEmitter::emitDagSplat(DagInit *DI) { 16700b57cec5SDimitry Andric assert_with_loc(DI->getNumArgs() == 2, "splat() expects two arguments"); 16710b57cec5SDimitry Andric std::pair<Type, std::string> A = emitDagArg(DI->getArg(0), 16720b57cec5SDimitry Andric DI->getArgNameStr(0)); 16730b57cec5SDimitry Andric std::pair<Type, std::string> B = emitDagArg(DI->getArg(1), 16740b57cec5SDimitry Andric DI->getArgNameStr(1)); 16750b57cec5SDimitry Andric 16760b57cec5SDimitry Andric assert_with_loc(B.first.isScalar(), 16770b57cec5SDimitry Andric "splat() requires a scalar int as the second argument"); 16780b57cec5SDimitry Andric 16790b57cec5SDimitry Andric std::string S = "__builtin_shufflevector(" + A.second + ", " + A.second; 16800b57cec5SDimitry Andric for (unsigned I = 0; I < Intr.getBaseType().getNumElements(); ++I) { 16810b57cec5SDimitry Andric S += ", " + B.second; 16820b57cec5SDimitry Andric } 16830b57cec5SDimitry Andric S += ")"; 16840b57cec5SDimitry Andric 16850b57cec5SDimitry Andric return std::make_pair(Intr.getBaseType(), S); 16860b57cec5SDimitry Andric } 16870b57cec5SDimitry Andric 16880b57cec5SDimitry Andric std::pair<Type, std::string> Intrinsic::DagEmitter::emitDagSaveTemp(DagInit *DI) { 16890b57cec5SDimitry Andric assert_with_loc(DI->getNumArgs() == 2, "save_temp() expects two arguments"); 16900b57cec5SDimitry Andric std::pair<Type, std::string> A = emitDagArg(DI->getArg(1), 16910b57cec5SDimitry Andric DI->getArgNameStr(1)); 16920b57cec5SDimitry Andric 16930b57cec5SDimitry Andric assert_with_loc(!A.first.isVoid(), 16940b57cec5SDimitry Andric "Argument to save_temp() must have non-void type!"); 16950b57cec5SDimitry Andric 16960b57cec5SDimitry Andric std::string N = DI->getArgNameStr(0); 16970b57cec5SDimitry Andric assert_with_loc(!N.empty(), 16980b57cec5SDimitry Andric "save_temp() expects a name as the first argument"); 16990b57cec5SDimitry Andric 17000b57cec5SDimitry Andric assert_with_loc(Intr.Variables.find(N) == Intr.Variables.end(), 17010b57cec5SDimitry Andric "Variable already defined!"); 17020b57cec5SDimitry Andric Intr.Variables[N] = Variable(A.first, N + Intr.VariablePostfix); 17030b57cec5SDimitry Andric 17040b57cec5SDimitry Andric std::string S = 17050b57cec5SDimitry Andric A.first.str() + " " + Intr.Variables[N].getName() + " = " + A.second; 17060b57cec5SDimitry Andric 17070b57cec5SDimitry Andric return std::make_pair(Type::getVoid(), S); 17080b57cec5SDimitry Andric } 17090b57cec5SDimitry Andric 17100b57cec5SDimitry Andric std::pair<Type, std::string> 17110b57cec5SDimitry Andric Intrinsic::DagEmitter::emitDagNameReplace(DagInit *DI) { 17120b57cec5SDimitry Andric std::string S = Intr.Name; 17130b57cec5SDimitry Andric 17140b57cec5SDimitry Andric assert_with_loc(DI->getNumArgs() == 2, "name_replace requires 2 arguments!"); 17150b57cec5SDimitry Andric std::string ToReplace = cast<StringInit>(DI->getArg(0))->getAsUnquotedString(); 17160b57cec5SDimitry Andric std::string ReplaceWith = cast<StringInit>(DI->getArg(1))->getAsUnquotedString(); 17170b57cec5SDimitry Andric 17180b57cec5SDimitry Andric size_t Idx = S.find(ToReplace); 17190b57cec5SDimitry Andric 17200b57cec5SDimitry Andric assert_with_loc(Idx != std::string::npos, "name should contain '" + ToReplace + "'!"); 17210b57cec5SDimitry Andric S.replace(Idx, ToReplace.size(), ReplaceWith); 17220b57cec5SDimitry Andric 17230b57cec5SDimitry Andric return std::make_pair(Type::getVoid(), S); 17240b57cec5SDimitry Andric } 17250b57cec5SDimitry Andric 17260b57cec5SDimitry Andric std::pair<Type, std::string> Intrinsic::DagEmitter::emitDagLiteral(DagInit *DI){ 17270b57cec5SDimitry Andric std::string Ty = cast<StringInit>(DI->getArg(0))->getAsUnquotedString(); 17280b57cec5SDimitry Andric std::string Value = cast<StringInit>(DI->getArg(1))->getAsUnquotedString(); 17290b57cec5SDimitry Andric return std::make_pair(Type::fromTypedefName(Ty), Value); 17300b57cec5SDimitry Andric } 17310b57cec5SDimitry Andric 17320b57cec5SDimitry Andric std::pair<Type, std::string> 17330b57cec5SDimitry Andric Intrinsic::DagEmitter::emitDagArg(Init *Arg, std::string ArgName) { 17340b57cec5SDimitry Andric if (!ArgName.empty()) { 17350b57cec5SDimitry Andric assert_with_loc(!Arg->isComplete(), 17360b57cec5SDimitry Andric "Arguments must either be DAGs or names, not both!"); 17370b57cec5SDimitry Andric assert_with_loc(Intr.Variables.find(ArgName) != Intr.Variables.end(), 17380b57cec5SDimitry Andric "Variable not defined!"); 17390b57cec5SDimitry Andric Variable &V = Intr.Variables[ArgName]; 17400b57cec5SDimitry Andric return std::make_pair(V.getType(), V.getName()); 17410b57cec5SDimitry Andric } 17420b57cec5SDimitry Andric 17430b57cec5SDimitry Andric assert(Arg && "Neither ArgName nor Arg?!"); 17440b57cec5SDimitry Andric DagInit *DI = dyn_cast<DagInit>(Arg); 17450b57cec5SDimitry Andric assert_with_loc(DI, "Arguments must either be DAGs or names!"); 17460b57cec5SDimitry Andric 17470b57cec5SDimitry Andric return emitDag(DI); 17480b57cec5SDimitry Andric } 17490b57cec5SDimitry Andric 17500b57cec5SDimitry Andric std::string Intrinsic::generate() { 1751a7dea167SDimitry Andric // Avoid duplicated code for big and little endian 1752a7dea167SDimitry Andric if (isBigEndianSafe()) { 1753a7dea167SDimitry Andric generateImpl(false, "", ""); 1754a7dea167SDimitry Andric return OS.str(); 1755a7dea167SDimitry Andric } 17560b57cec5SDimitry Andric // Little endian intrinsics are simple and don't require any argument 17570b57cec5SDimitry Andric // swapping. 17580b57cec5SDimitry Andric OS << "#ifdef __LITTLE_ENDIAN__\n"; 17590b57cec5SDimitry Andric 17600b57cec5SDimitry Andric generateImpl(false, "", ""); 17610b57cec5SDimitry Andric 17620b57cec5SDimitry Andric OS << "#else\n"; 17630b57cec5SDimitry Andric 17640b57cec5SDimitry Andric // Big endian intrinsics are more complex. The user intended these 17650b57cec5SDimitry Andric // intrinsics to operate on a vector "as-if" loaded by (V)LDR, 17660b57cec5SDimitry Andric // but we load as-if (V)LD1. So we should swap all arguments and 17670b57cec5SDimitry Andric // swap the return value too. 17680b57cec5SDimitry Andric // 17690b57cec5SDimitry Andric // If we call sub-intrinsics, we should call a version that does 17700b57cec5SDimitry Andric // not re-swap the arguments! 17710b57cec5SDimitry Andric generateImpl(true, "", "__noswap_"); 17720b57cec5SDimitry Andric 17730b57cec5SDimitry Andric // If we're needed early, create a non-swapping variant for 17740b57cec5SDimitry Andric // big-endian. 17750b57cec5SDimitry Andric if (NeededEarly) { 17760b57cec5SDimitry Andric generateImpl(false, "__noswap_", "__noswap_"); 17770b57cec5SDimitry Andric } 17780b57cec5SDimitry Andric OS << "#endif\n\n"; 17790b57cec5SDimitry Andric 17800b57cec5SDimitry Andric return OS.str(); 17810b57cec5SDimitry Andric } 17820b57cec5SDimitry Andric 17830b57cec5SDimitry Andric void Intrinsic::generateImpl(bool ReverseArguments, 17840b57cec5SDimitry Andric StringRef NamePrefix, StringRef CallPrefix) { 17850b57cec5SDimitry Andric CurrentRecord = R; 17860b57cec5SDimitry Andric 17870b57cec5SDimitry Andric // If we call a macro, our local variables may be corrupted due to 17880b57cec5SDimitry Andric // lack of proper lexical scoping. So, add a globally unique postfix 17890b57cec5SDimitry Andric // to every variable. 17900b57cec5SDimitry Andric // 17910b57cec5SDimitry Andric // indexBody() should have set up the Dependencies set by now. 17920b57cec5SDimitry Andric for (auto *I : Dependencies) 17930b57cec5SDimitry Andric if (I->UseMacro) { 17940b57cec5SDimitry Andric VariablePostfix = "_" + utostr(Emitter.getUniqueNumber()); 17950b57cec5SDimitry Andric break; 17960b57cec5SDimitry Andric } 17970b57cec5SDimitry Andric 17980b57cec5SDimitry Andric initVariables(); 17990b57cec5SDimitry Andric 18000b57cec5SDimitry Andric emitPrototype(NamePrefix); 18010b57cec5SDimitry Andric 18020b57cec5SDimitry Andric if (IsUnavailable) { 18030b57cec5SDimitry Andric OS << " __attribute__((unavailable));"; 18040b57cec5SDimitry Andric } else { 18050b57cec5SDimitry Andric emitOpeningBrace(); 18060b57cec5SDimitry Andric emitShadowedArgs(); 18070b57cec5SDimitry Andric if (ReverseArguments) 18080b57cec5SDimitry Andric emitArgumentReversal(); 18090b57cec5SDimitry Andric emitBody(CallPrefix); 18100b57cec5SDimitry Andric if (ReverseArguments) 18110b57cec5SDimitry Andric emitReturnReversal(); 18120b57cec5SDimitry Andric emitReturn(); 18130b57cec5SDimitry Andric emitClosingBrace(); 18140b57cec5SDimitry Andric } 18150b57cec5SDimitry Andric OS << "\n"; 18160b57cec5SDimitry Andric 18170b57cec5SDimitry Andric CurrentRecord = nullptr; 18180b57cec5SDimitry Andric } 18190b57cec5SDimitry Andric 18200b57cec5SDimitry Andric void Intrinsic::indexBody() { 18210b57cec5SDimitry Andric CurrentRecord = R; 18220b57cec5SDimitry Andric 18230b57cec5SDimitry Andric initVariables(); 18240b57cec5SDimitry Andric emitBody(""); 18250b57cec5SDimitry Andric OS.str(""); 18260b57cec5SDimitry Andric 18270b57cec5SDimitry Andric CurrentRecord = nullptr; 18280b57cec5SDimitry Andric } 18290b57cec5SDimitry Andric 18300b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 18310b57cec5SDimitry Andric // NeonEmitter implementation 18320b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 18330b57cec5SDimitry Andric 18340b57cec5SDimitry Andric Intrinsic &NeonEmitter::getIntrinsic(StringRef Name, ArrayRef<Type> Types) { 18350b57cec5SDimitry Andric // First, look up the name in the intrinsic map. 18360b57cec5SDimitry Andric assert_with_loc(IntrinsicMap.find(Name.str()) != IntrinsicMap.end(), 18370b57cec5SDimitry Andric ("Intrinsic '" + Name + "' not found!").str()); 18380b57cec5SDimitry Andric auto &V = IntrinsicMap.find(Name.str())->second; 18390b57cec5SDimitry Andric std::vector<Intrinsic *> GoodVec; 18400b57cec5SDimitry Andric 18410b57cec5SDimitry Andric // Create a string to print if we end up failing. 18420b57cec5SDimitry Andric std::string ErrMsg = "looking up intrinsic '" + Name.str() + "("; 18430b57cec5SDimitry Andric for (unsigned I = 0; I < Types.size(); ++I) { 18440b57cec5SDimitry Andric if (I != 0) 18450b57cec5SDimitry Andric ErrMsg += ", "; 18460b57cec5SDimitry Andric ErrMsg += Types[I].str(); 18470b57cec5SDimitry Andric } 18480b57cec5SDimitry Andric ErrMsg += ")'\n"; 18490b57cec5SDimitry Andric ErrMsg += "Available overloads:\n"; 18500b57cec5SDimitry Andric 18510b57cec5SDimitry Andric // Now, look through each intrinsic implementation and see if the types are 18520b57cec5SDimitry Andric // compatible. 18530b57cec5SDimitry Andric for (auto &I : V) { 18540b57cec5SDimitry Andric ErrMsg += " - " + I.getReturnType().str() + " " + I.getMangledName(); 18550b57cec5SDimitry Andric ErrMsg += "("; 18560b57cec5SDimitry Andric for (unsigned A = 0; A < I.getNumParams(); ++A) { 18570b57cec5SDimitry Andric if (A != 0) 18580b57cec5SDimitry Andric ErrMsg += ", "; 18590b57cec5SDimitry Andric ErrMsg += I.getParamType(A).str(); 18600b57cec5SDimitry Andric } 18610b57cec5SDimitry Andric ErrMsg += ")\n"; 18620b57cec5SDimitry Andric 18630b57cec5SDimitry Andric if (I.getNumParams() != Types.size()) 18640b57cec5SDimitry Andric continue; 18650b57cec5SDimitry Andric 18660b57cec5SDimitry Andric bool Good = true; 18670b57cec5SDimitry Andric for (unsigned Arg = 0; Arg < Types.size(); ++Arg) { 18680b57cec5SDimitry Andric if (I.getParamType(Arg) != Types[Arg]) { 18690b57cec5SDimitry Andric Good = false; 18700b57cec5SDimitry Andric break; 18710b57cec5SDimitry Andric } 18720b57cec5SDimitry Andric } 18730b57cec5SDimitry Andric if (Good) 18740b57cec5SDimitry Andric GoodVec.push_back(&I); 18750b57cec5SDimitry Andric } 18760b57cec5SDimitry Andric 18770b57cec5SDimitry Andric assert_with_loc(!GoodVec.empty(), 18780b57cec5SDimitry Andric "No compatible intrinsic found - " + ErrMsg); 18790b57cec5SDimitry Andric assert_with_loc(GoodVec.size() == 1, "Multiple overloads found - " + ErrMsg); 18800b57cec5SDimitry Andric 18810b57cec5SDimitry Andric return *GoodVec.front(); 18820b57cec5SDimitry Andric } 18830b57cec5SDimitry Andric 18840b57cec5SDimitry Andric void NeonEmitter::createIntrinsic(Record *R, 18850b57cec5SDimitry Andric SmallVectorImpl<Intrinsic *> &Out) { 18860b57cec5SDimitry Andric std::string Name = R->getValueAsString("Name"); 18870b57cec5SDimitry Andric std::string Proto = R->getValueAsString("Prototype"); 18880b57cec5SDimitry Andric std::string Types = R->getValueAsString("Types"); 18890b57cec5SDimitry Andric Record *OperationRec = R->getValueAsDef("Operation"); 18900b57cec5SDimitry Andric bool CartesianProductOfTypes = R->getValueAsBit("CartesianProductOfTypes"); 18910b57cec5SDimitry Andric bool BigEndianSafe = R->getValueAsBit("BigEndianSafe"); 18920b57cec5SDimitry Andric std::string Guard = R->getValueAsString("ArchGuard"); 18930b57cec5SDimitry Andric bool IsUnavailable = OperationRec->getValueAsBit("Unavailable"); 18940b57cec5SDimitry Andric 18950b57cec5SDimitry Andric // Set the global current record. This allows assert_with_loc to produce 18960b57cec5SDimitry Andric // decent location information even when highly nested. 18970b57cec5SDimitry Andric CurrentRecord = R; 18980b57cec5SDimitry Andric 18990b57cec5SDimitry Andric ListInit *Body = OperationRec->getValueAsListInit("Ops"); 19000b57cec5SDimitry Andric 19010b57cec5SDimitry Andric std::vector<TypeSpec> TypeSpecs = TypeSpec::fromTypeSpecs(Types); 19020b57cec5SDimitry Andric 19030b57cec5SDimitry Andric ClassKind CK = ClassNone; 19040b57cec5SDimitry Andric if (R->getSuperClasses().size() >= 2) 19050b57cec5SDimitry Andric CK = ClassMap[R->getSuperClasses()[1].first]; 19060b57cec5SDimitry Andric 19070b57cec5SDimitry Andric std::vector<std::pair<TypeSpec, TypeSpec>> NewTypeSpecs; 19080b57cec5SDimitry Andric for (auto TS : TypeSpecs) { 19090b57cec5SDimitry Andric if (CartesianProductOfTypes) { 1910*480093f4SDimitry Andric Type DefaultT(TS, "."); 19110b57cec5SDimitry Andric for (auto SrcTS : TypeSpecs) { 1912*480093f4SDimitry Andric Type DefaultSrcT(SrcTS, "."); 19130b57cec5SDimitry Andric if (TS == SrcTS || 19140b57cec5SDimitry Andric DefaultSrcT.getSizeInBits() != DefaultT.getSizeInBits()) 19150b57cec5SDimitry Andric continue; 19160b57cec5SDimitry Andric NewTypeSpecs.push_back(std::make_pair(TS, SrcTS)); 19170b57cec5SDimitry Andric } 19180b57cec5SDimitry Andric } else { 19190b57cec5SDimitry Andric NewTypeSpecs.push_back(std::make_pair(TS, TS)); 19200b57cec5SDimitry Andric } 19210b57cec5SDimitry Andric } 19220b57cec5SDimitry Andric 19230b57cec5SDimitry Andric llvm::sort(NewTypeSpecs); 19240b57cec5SDimitry Andric NewTypeSpecs.erase(std::unique(NewTypeSpecs.begin(), NewTypeSpecs.end()), 19250b57cec5SDimitry Andric NewTypeSpecs.end()); 19260b57cec5SDimitry Andric auto &Entry = IntrinsicMap[Name]; 19270b57cec5SDimitry Andric 19280b57cec5SDimitry Andric for (auto &I : NewTypeSpecs) { 19290b57cec5SDimitry Andric Entry.emplace_back(R, Name, Proto, I.first, I.second, CK, Body, *this, 19300b57cec5SDimitry Andric Guard, IsUnavailable, BigEndianSafe); 19310b57cec5SDimitry Andric Out.push_back(&Entry.back()); 19320b57cec5SDimitry Andric } 19330b57cec5SDimitry Andric 19340b57cec5SDimitry Andric CurrentRecord = nullptr; 19350b57cec5SDimitry Andric } 19360b57cec5SDimitry Andric 19370b57cec5SDimitry Andric /// genBuiltinsDef: Generate the BuiltinsARM.def and BuiltinsAArch64.def 19380b57cec5SDimitry Andric /// declaration of builtins, checking for unique builtin declarations. 19390b57cec5SDimitry Andric void NeonEmitter::genBuiltinsDef(raw_ostream &OS, 19400b57cec5SDimitry Andric SmallVectorImpl<Intrinsic *> &Defs) { 19410b57cec5SDimitry Andric OS << "#ifdef GET_NEON_BUILTINS\n"; 19420b57cec5SDimitry Andric 19430b57cec5SDimitry Andric // We only want to emit a builtin once, and we want to emit them in 19440b57cec5SDimitry Andric // alphabetical order, so use a std::set. 19450b57cec5SDimitry Andric std::set<std::string> Builtins; 19460b57cec5SDimitry Andric 19470b57cec5SDimitry Andric for (auto *Def : Defs) { 19480b57cec5SDimitry Andric if (Def->hasBody()) 19490b57cec5SDimitry Andric continue; 19500b57cec5SDimitry Andric 19510b57cec5SDimitry Andric std::string S = "BUILTIN(__builtin_neon_" + Def->getMangledName() + ", \""; 19520b57cec5SDimitry Andric 19530b57cec5SDimitry Andric S += Def->getBuiltinTypeStr(); 19540b57cec5SDimitry Andric S += "\", \"n\")"; 19550b57cec5SDimitry Andric 19560b57cec5SDimitry Andric Builtins.insert(S); 19570b57cec5SDimitry Andric } 19580b57cec5SDimitry Andric 19590b57cec5SDimitry Andric for (auto &S : Builtins) 19600b57cec5SDimitry Andric OS << S << "\n"; 19610b57cec5SDimitry Andric OS << "#endif\n\n"; 19620b57cec5SDimitry Andric } 19630b57cec5SDimitry Andric 19640b57cec5SDimitry Andric /// Generate the ARM and AArch64 overloaded type checking code for 19650b57cec5SDimitry Andric /// SemaChecking.cpp, checking for unique builtin declarations. 19660b57cec5SDimitry Andric void NeonEmitter::genOverloadTypeCheckCode(raw_ostream &OS, 19670b57cec5SDimitry Andric SmallVectorImpl<Intrinsic *> &Defs) { 19680b57cec5SDimitry Andric OS << "#ifdef GET_NEON_OVERLOAD_CHECK\n"; 19690b57cec5SDimitry Andric 19700b57cec5SDimitry Andric // We record each overload check line before emitting because subsequent Inst 19710b57cec5SDimitry Andric // definitions may extend the number of permitted types (i.e. augment the 19720b57cec5SDimitry Andric // Mask). Use std::map to avoid sorting the table by hash number. 19730b57cec5SDimitry Andric struct OverloadInfo { 19740b57cec5SDimitry Andric uint64_t Mask; 19750b57cec5SDimitry Andric int PtrArgNum; 19760b57cec5SDimitry Andric bool HasConstPtr; 19770b57cec5SDimitry Andric OverloadInfo() : Mask(0ULL), PtrArgNum(0), HasConstPtr(false) {} 19780b57cec5SDimitry Andric }; 19790b57cec5SDimitry Andric std::map<std::string, OverloadInfo> OverloadMap; 19800b57cec5SDimitry Andric 19810b57cec5SDimitry Andric for (auto *Def : Defs) { 19820b57cec5SDimitry Andric // If the def has a body (that is, it has Operation DAGs), it won't call 19830b57cec5SDimitry Andric // __builtin_neon_* so we don't need to generate a definition for it. 19840b57cec5SDimitry Andric if (Def->hasBody()) 19850b57cec5SDimitry Andric continue; 19860b57cec5SDimitry Andric // Functions which have a scalar argument cannot be overloaded, no need to 19870b57cec5SDimitry Andric // check them if we are emitting the type checking code. 19880b57cec5SDimitry Andric if (Def->protoHasScalar()) 19890b57cec5SDimitry Andric continue; 19900b57cec5SDimitry Andric 19910b57cec5SDimitry Andric uint64_t Mask = 0ULL; 1992*480093f4SDimitry Andric Mask |= 1ULL << Def->getPolymorphicKeyType().getNeonEnum(); 19930b57cec5SDimitry Andric 19940b57cec5SDimitry Andric // Check if the function has a pointer or const pointer argument. 19950b57cec5SDimitry Andric int PtrArgNum = -1; 19960b57cec5SDimitry Andric bool HasConstPtr = false; 19970b57cec5SDimitry Andric for (unsigned I = 0; I < Def->getNumParams(); ++I) { 1998*480093f4SDimitry Andric const auto &Type = Def->getParamType(I); 1999*480093f4SDimitry Andric if (Type.isPointer()) { 20000b57cec5SDimitry Andric PtrArgNum = I; 2001*480093f4SDimitry Andric HasConstPtr = Type.isConstPointer(); 20020b57cec5SDimitry Andric } 20030b57cec5SDimitry Andric } 2004*480093f4SDimitry Andric 20050b57cec5SDimitry Andric // For sret builtins, adjust the pointer argument index. 20060b57cec5SDimitry Andric if (PtrArgNum >= 0 && Def->getReturnType().getNumVectors() > 1) 20070b57cec5SDimitry Andric PtrArgNum += 1; 20080b57cec5SDimitry Andric 20090b57cec5SDimitry Andric std::string Name = Def->getName(); 20100b57cec5SDimitry Andric // Omit type checking for the pointer arguments of vld1_lane, vld1_dup, 20110b57cec5SDimitry Andric // and vst1_lane intrinsics. Using a pointer to the vector element 20120b57cec5SDimitry Andric // type with one of those operations causes codegen to select an aligned 20130b57cec5SDimitry Andric // load/store instruction. If you want an unaligned operation, 20140b57cec5SDimitry Andric // the pointer argument needs to have less alignment than element type, 20150b57cec5SDimitry Andric // so just accept any pointer type. 20160b57cec5SDimitry Andric if (Name == "vld1_lane" || Name == "vld1_dup" || Name == "vst1_lane") { 20170b57cec5SDimitry Andric PtrArgNum = -1; 20180b57cec5SDimitry Andric HasConstPtr = false; 20190b57cec5SDimitry Andric } 20200b57cec5SDimitry Andric 20210b57cec5SDimitry Andric if (Mask) { 20220b57cec5SDimitry Andric std::string Name = Def->getMangledName(); 20230b57cec5SDimitry Andric OverloadMap.insert(std::make_pair(Name, OverloadInfo())); 20240b57cec5SDimitry Andric OverloadInfo &OI = OverloadMap[Name]; 20250b57cec5SDimitry Andric OI.Mask |= Mask; 20260b57cec5SDimitry Andric OI.PtrArgNum |= PtrArgNum; 20270b57cec5SDimitry Andric OI.HasConstPtr = HasConstPtr; 20280b57cec5SDimitry Andric } 20290b57cec5SDimitry Andric } 20300b57cec5SDimitry Andric 20310b57cec5SDimitry Andric for (auto &I : OverloadMap) { 20320b57cec5SDimitry Andric OverloadInfo &OI = I.second; 20330b57cec5SDimitry Andric 20340b57cec5SDimitry Andric OS << "case NEON::BI__builtin_neon_" << I.first << ": "; 20350b57cec5SDimitry Andric OS << "mask = 0x" << Twine::utohexstr(OI.Mask) << "ULL"; 20360b57cec5SDimitry Andric if (OI.PtrArgNum >= 0) 20370b57cec5SDimitry Andric OS << "; PtrArgNum = " << OI.PtrArgNum; 20380b57cec5SDimitry Andric if (OI.HasConstPtr) 20390b57cec5SDimitry Andric OS << "; HasConstPtr = true"; 20400b57cec5SDimitry Andric OS << "; break;\n"; 20410b57cec5SDimitry Andric } 20420b57cec5SDimitry Andric OS << "#endif\n\n"; 20430b57cec5SDimitry Andric } 20440b57cec5SDimitry Andric 20450b57cec5SDimitry Andric void NeonEmitter::genIntrinsicRangeCheckCode(raw_ostream &OS, 20460b57cec5SDimitry Andric SmallVectorImpl<Intrinsic *> &Defs) { 20470b57cec5SDimitry Andric OS << "#ifdef GET_NEON_IMMEDIATE_CHECK\n"; 20480b57cec5SDimitry Andric 20490b57cec5SDimitry Andric std::set<std::string> Emitted; 20500b57cec5SDimitry Andric 20510b57cec5SDimitry Andric for (auto *Def : Defs) { 20520b57cec5SDimitry Andric if (Def->hasBody()) 20530b57cec5SDimitry Andric continue; 20540b57cec5SDimitry Andric // Functions which do not have an immediate do not need to have range 20550b57cec5SDimitry Andric // checking code emitted. 20560b57cec5SDimitry Andric if (!Def->hasImmediate()) 20570b57cec5SDimitry Andric continue; 20580b57cec5SDimitry Andric if (Emitted.find(Def->getMangledName()) != Emitted.end()) 20590b57cec5SDimitry Andric continue; 20600b57cec5SDimitry Andric 20610b57cec5SDimitry Andric std::string LowerBound, UpperBound; 20620b57cec5SDimitry Andric 20630b57cec5SDimitry Andric Record *R = Def->getRecord(); 20640b57cec5SDimitry Andric if (R->getValueAsBit("isVCVT_N")) { 20650b57cec5SDimitry Andric // VCVT between floating- and fixed-point values takes an immediate 20660b57cec5SDimitry Andric // in the range [1, 32) for f32 or [1, 64) for f64 or [1, 16) for f16. 20670b57cec5SDimitry Andric LowerBound = "1"; 20680b57cec5SDimitry Andric if (Def->getBaseType().getElementSizeInBits() == 16 || 20690b57cec5SDimitry Andric Def->getName().find('h') != std::string::npos) 20700b57cec5SDimitry Andric // VCVTh operating on FP16 intrinsics in range [1, 16) 20710b57cec5SDimitry Andric UpperBound = "15"; 20720b57cec5SDimitry Andric else if (Def->getBaseType().getElementSizeInBits() == 32) 20730b57cec5SDimitry Andric UpperBound = "31"; 20740b57cec5SDimitry Andric else 20750b57cec5SDimitry Andric UpperBound = "63"; 20760b57cec5SDimitry Andric } else if (R->getValueAsBit("isScalarShift")) { 20770b57cec5SDimitry Andric // Right shifts have an 'r' in the name, left shifts do not. Convert 20780b57cec5SDimitry Andric // instructions have the same bounds and right shifts. 20790b57cec5SDimitry Andric if (Def->getName().find('r') != std::string::npos || 20800b57cec5SDimitry Andric Def->getName().find("cvt") != std::string::npos) 20810b57cec5SDimitry Andric LowerBound = "1"; 20820b57cec5SDimitry Andric 20830b57cec5SDimitry Andric UpperBound = utostr(Def->getReturnType().getElementSizeInBits() - 1); 20840b57cec5SDimitry Andric } else if (R->getValueAsBit("isShift")) { 20850b57cec5SDimitry Andric // Builtins which are overloaded by type will need to have their upper 20860b57cec5SDimitry Andric // bound computed at Sema time based on the type constant. 20870b57cec5SDimitry Andric 20880b57cec5SDimitry Andric // Right shifts have an 'r' in the name, left shifts do not. 20890b57cec5SDimitry Andric if (Def->getName().find('r') != std::string::npos) 20900b57cec5SDimitry Andric LowerBound = "1"; 20910b57cec5SDimitry Andric UpperBound = "RFT(TV, true)"; 20920b57cec5SDimitry Andric } else if (Def->getClassKind(true) == ClassB) { 20930b57cec5SDimitry Andric // ClassB intrinsics have a type (and hence lane number) that is only 20940b57cec5SDimitry Andric // known at runtime. 20950b57cec5SDimitry Andric if (R->getValueAsBit("isLaneQ")) 20960b57cec5SDimitry Andric UpperBound = "RFT(TV, false, true)"; 20970b57cec5SDimitry Andric else 20980b57cec5SDimitry Andric UpperBound = "RFT(TV, false, false)"; 20990b57cec5SDimitry Andric } else { 21000b57cec5SDimitry Andric // The immediate generally refers to a lane in the preceding argument. 21010b57cec5SDimitry Andric assert(Def->getImmediateIdx() > 0); 21020b57cec5SDimitry Andric Type T = Def->getParamType(Def->getImmediateIdx() - 1); 21030b57cec5SDimitry Andric UpperBound = utostr(T.getNumElements() - 1); 21040b57cec5SDimitry Andric } 21050b57cec5SDimitry Andric 21060b57cec5SDimitry Andric // Calculate the index of the immediate that should be range checked. 21070b57cec5SDimitry Andric unsigned Idx = Def->getNumParams(); 21080b57cec5SDimitry Andric if (Def->hasImmediate()) 21090b57cec5SDimitry Andric Idx = Def->getGeneratedParamIdx(Def->getImmediateIdx()); 21100b57cec5SDimitry Andric 21110b57cec5SDimitry Andric OS << "case NEON::BI__builtin_neon_" << Def->getMangledName() << ": " 21120b57cec5SDimitry Andric << "i = " << Idx << ";"; 21130b57cec5SDimitry Andric if (!LowerBound.empty()) 21140b57cec5SDimitry Andric OS << " l = " << LowerBound << ";"; 21150b57cec5SDimitry Andric if (!UpperBound.empty()) 21160b57cec5SDimitry Andric OS << " u = " << UpperBound << ";"; 21170b57cec5SDimitry Andric OS << " break;\n"; 21180b57cec5SDimitry Andric 21190b57cec5SDimitry Andric Emitted.insert(Def->getMangledName()); 21200b57cec5SDimitry Andric } 21210b57cec5SDimitry Andric 21220b57cec5SDimitry Andric OS << "#endif\n\n"; 21230b57cec5SDimitry Andric } 21240b57cec5SDimitry Andric 21250b57cec5SDimitry Andric /// runHeader - Emit a file with sections defining: 21260b57cec5SDimitry Andric /// 1. the NEON section of BuiltinsARM.def and BuiltinsAArch64.def. 21270b57cec5SDimitry Andric /// 2. the SemaChecking code for the type overload checking. 21280b57cec5SDimitry Andric /// 3. the SemaChecking code for validation of intrinsic immediate arguments. 21290b57cec5SDimitry Andric void NeonEmitter::runHeader(raw_ostream &OS) { 21300b57cec5SDimitry Andric std::vector<Record *> RV = Records.getAllDerivedDefinitions("Inst"); 21310b57cec5SDimitry Andric 21320b57cec5SDimitry Andric SmallVector<Intrinsic *, 128> Defs; 21330b57cec5SDimitry Andric for (auto *R : RV) 21340b57cec5SDimitry Andric createIntrinsic(R, Defs); 21350b57cec5SDimitry Andric 21360b57cec5SDimitry Andric // Generate shared BuiltinsXXX.def 21370b57cec5SDimitry Andric genBuiltinsDef(OS, Defs); 21380b57cec5SDimitry Andric 21390b57cec5SDimitry Andric // Generate ARM overloaded type checking code for SemaChecking.cpp 21400b57cec5SDimitry Andric genOverloadTypeCheckCode(OS, Defs); 21410b57cec5SDimitry Andric 21420b57cec5SDimitry Andric // Generate ARM range checking code for shift/lane immediates. 21430b57cec5SDimitry Andric genIntrinsicRangeCheckCode(OS, Defs); 21440b57cec5SDimitry Andric } 21450b57cec5SDimitry Andric 21460b57cec5SDimitry Andric /// run - Read the records in arm_neon.td and output arm_neon.h. arm_neon.h 21470b57cec5SDimitry Andric /// is comprised of type definitions and function declarations. 21480b57cec5SDimitry Andric void NeonEmitter::run(raw_ostream &OS) { 21490b57cec5SDimitry Andric OS << "/*===---- arm_neon.h - ARM Neon intrinsics " 21500b57cec5SDimitry Andric "------------------------------" 21510b57cec5SDimitry Andric "---===\n" 21520b57cec5SDimitry Andric " *\n" 21530b57cec5SDimitry Andric " * Permission is hereby granted, free of charge, to any person " 21540b57cec5SDimitry Andric "obtaining " 21550b57cec5SDimitry Andric "a copy\n" 21560b57cec5SDimitry Andric " * of this software and associated documentation files (the " 21570b57cec5SDimitry Andric "\"Software\")," 21580b57cec5SDimitry Andric " to deal\n" 21590b57cec5SDimitry Andric " * in the Software without restriction, including without limitation " 21600b57cec5SDimitry Andric "the " 21610b57cec5SDimitry Andric "rights\n" 21620b57cec5SDimitry Andric " * to use, copy, modify, merge, publish, distribute, sublicense, " 21630b57cec5SDimitry Andric "and/or sell\n" 21640b57cec5SDimitry Andric " * copies of the Software, and to permit persons to whom the Software " 21650b57cec5SDimitry Andric "is\n" 21660b57cec5SDimitry Andric " * furnished to do so, subject to the following conditions:\n" 21670b57cec5SDimitry Andric " *\n" 21680b57cec5SDimitry Andric " * The above copyright notice and this permission notice shall be " 21690b57cec5SDimitry Andric "included in\n" 21700b57cec5SDimitry Andric " * all copies or substantial portions of the Software.\n" 21710b57cec5SDimitry Andric " *\n" 21720b57cec5SDimitry Andric " * THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, " 21730b57cec5SDimitry Andric "EXPRESS OR\n" 21740b57cec5SDimitry Andric " * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF " 21750b57cec5SDimitry Andric "MERCHANTABILITY,\n" 21760b57cec5SDimitry Andric " * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT " 21770b57cec5SDimitry Andric "SHALL THE\n" 21780b57cec5SDimitry Andric " * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR " 21790b57cec5SDimitry Andric "OTHER\n" 21800b57cec5SDimitry Andric " * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, " 21810b57cec5SDimitry Andric "ARISING FROM,\n" 21820b57cec5SDimitry Andric " * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER " 21830b57cec5SDimitry Andric "DEALINGS IN\n" 21840b57cec5SDimitry Andric " * THE SOFTWARE.\n" 21850b57cec5SDimitry Andric " *\n" 21860b57cec5SDimitry Andric " *===-----------------------------------------------------------------" 21870b57cec5SDimitry Andric "---" 21880b57cec5SDimitry Andric "---===\n" 21890b57cec5SDimitry Andric " */\n\n"; 21900b57cec5SDimitry Andric 21910b57cec5SDimitry Andric OS << "#ifndef __ARM_NEON_H\n"; 21920b57cec5SDimitry Andric OS << "#define __ARM_NEON_H\n\n"; 21930b57cec5SDimitry Andric 21940b57cec5SDimitry Andric OS << "#if !defined(__ARM_NEON)\n"; 21950b57cec5SDimitry Andric OS << "#error \"NEON support not enabled\"\n"; 21960b57cec5SDimitry Andric OS << "#endif\n\n"; 21970b57cec5SDimitry Andric 21980b57cec5SDimitry Andric OS << "#include <stdint.h>\n\n"; 21990b57cec5SDimitry Andric 22000b57cec5SDimitry Andric // Emit NEON-specific scalar typedefs. 22010b57cec5SDimitry Andric OS << "typedef float float32_t;\n"; 22020b57cec5SDimitry Andric OS << "typedef __fp16 float16_t;\n"; 22030b57cec5SDimitry Andric 22040b57cec5SDimitry Andric OS << "#ifdef __aarch64__\n"; 22050b57cec5SDimitry Andric OS << "typedef double float64_t;\n"; 22060b57cec5SDimitry Andric OS << "#endif\n\n"; 22070b57cec5SDimitry Andric 22080b57cec5SDimitry Andric // For now, signedness of polynomial types depends on target 22090b57cec5SDimitry Andric OS << "#ifdef __aarch64__\n"; 22100b57cec5SDimitry Andric OS << "typedef uint8_t poly8_t;\n"; 22110b57cec5SDimitry Andric OS << "typedef uint16_t poly16_t;\n"; 22120b57cec5SDimitry Andric OS << "typedef uint64_t poly64_t;\n"; 22130b57cec5SDimitry Andric OS << "typedef __uint128_t poly128_t;\n"; 22140b57cec5SDimitry Andric OS << "#else\n"; 22150b57cec5SDimitry Andric OS << "typedef int8_t poly8_t;\n"; 22160b57cec5SDimitry Andric OS << "typedef int16_t poly16_t;\n"; 22170b57cec5SDimitry Andric OS << "#endif\n"; 22180b57cec5SDimitry Andric 22190b57cec5SDimitry Andric // Emit Neon vector typedefs. 22200b57cec5SDimitry Andric std::string TypedefTypes( 22210b57cec5SDimitry Andric "cQcsQsiQilQlUcQUcUsQUsUiQUiUlQUlhQhfQfdQdPcQPcPsQPsPlQPl"); 22220b57cec5SDimitry Andric std::vector<TypeSpec> TDTypeVec = TypeSpec::fromTypeSpecs(TypedefTypes); 22230b57cec5SDimitry Andric 22240b57cec5SDimitry Andric // Emit vector typedefs. 22250b57cec5SDimitry Andric bool InIfdef = false; 22260b57cec5SDimitry Andric for (auto &TS : TDTypeVec) { 22270b57cec5SDimitry Andric bool IsA64 = false; 2228*480093f4SDimitry Andric Type T(TS, "."); 2229*480093f4SDimitry Andric if (T.isDouble() || (T.isPoly() && T.getElementSizeInBits() == 64)) 22300b57cec5SDimitry Andric IsA64 = true; 22310b57cec5SDimitry Andric 22320b57cec5SDimitry Andric if (InIfdef && !IsA64) { 22330b57cec5SDimitry Andric OS << "#endif\n"; 22340b57cec5SDimitry Andric InIfdef = false; 22350b57cec5SDimitry Andric } 22360b57cec5SDimitry Andric if (!InIfdef && IsA64) { 22370b57cec5SDimitry Andric OS << "#ifdef __aarch64__\n"; 22380b57cec5SDimitry Andric InIfdef = true; 22390b57cec5SDimitry Andric } 22400b57cec5SDimitry Andric 22410b57cec5SDimitry Andric if (T.isPoly()) 22420b57cec5SDimitry Andric OS << "typedef __attribute__((neon_polyvector_type("; 22430b57cec5SDimitry Andric else 22440b57cec5SDimitry Andric OS << "typedef __attribute__((neon_vector_type("; 22450b57cec5SDimitry Andric 22460b57cec5SDimitry Andric Type T2 = T; 22470b57cec5SDimitry Andric T2.makeScalar(); 22480b57cec5SDimitry Andric OS << T.getNumElements() << "))) "; 22490b57cec5SDimitry Andric OS << T2.str(); 22500b57cec5SDimitry Andric OS << " " << T.str() << ";\n"; 22510b57cec5SDimitry Andric } 22520b57cec5SDimitry Andric if (InIfdef) 22530b57cec5SDimitry Andric OS << "#endif\n"; 22540b57cec5SDimitry Andric OS << "\n"; 22550b57cec5SDimitry Andric 22560b57cec5SDimitry Andric // Emit struct typedefs. 22570b57cec5SDimitry Andric InIfdef = false; 22580b57cec5SDimitry Andric for (unsigned NumMembers = 2; NumMembers <= 4; ++NumMembers) { 22590b57cec5SDimitry Andric for (auto &TS : TDTypeVec) { 22600b57cec5SDimitry Andric bool IsA64 = false; 2261*480093f4SDimitry Andric Type T(TS, "."); 2262*480093f4SDimitry Andric if (T.isDouble() || (T.isPoly() && T.getElementSizeInBits() == 64)) 22630b57cec5SDimitry Andric IsA64 = true; 22640b57cec5SDimitry Andric 22650b57cec5SDimitry Andric if (InIfdef && !IsA64) { 22660b57cec5SDimitry Andric OS << "#endif\n"; 22670b57cec5SDimitry Andric InIfdef = false; 22680b57cec5SDimitry Andric } 22690b57cec5SDimitry Andric if (!InIfdef && IsA64) { 22700b57cec5SDimitry Andric OS << "#ifdef __aarch64__\n"; 22710b57cec5SDimitry Andric InIfdef = true; 22720b57cec5SDimitry Andric } 22730b57cec5SDimitry Andric 2274*480093f4SDimitry Andric const char Mods[] = { static_cast<char>('2' + (NumMembers - 2)), 0}; 2275*480093f4SDimitry Andric Type VT(TS, Mods); 22760b57cec5SDimitry Andric OS << "typedef struct " << VT.str() << " {\n"; 22770b57cec5SDimitry Andric OS << " " << T.str() << " val"; 22780b57cec5SDimitry Andric OS << "[" << NumMembers << "]"; 22790b57cec5SDimitry Andric OS << ";\n} "; 22800b57cec5SDimitry Andric OS << VT.str() << ";\n"; 22810b57cec5SDimitry Andric OS << "\n"; 22820b57cec5SDimitry Andric } 22830b57cec5SDimitry Andric } 22840b57cec5SDimitry Andric if (InIfdef) 22850b57cec5SDimitry Andric OS << "#endif\n"; 22860b57cec5SDimitry Andric OS << "\n"; 22870b57cec5SDimitry Andric 22880b57cec5SDimitry Andric OS << "#define __ai static __inline__ __attribute__((__always_inline__, " 22890b57cec5SDimitry Andric "__nodebug__))\n\n"; 22900b57cec5SDimitry Andric 22910b57cec5SDimitry Andric SmallVector<Intrinsic *, 128> Defs; 22920b57cec5SDimitry Andric std::vector<Record *> RV = Records.getAllDerivedDefinitions("Inst"); 22930b57cec5SDimitry Andric for (auto *R : RV) 22940b57cec5SDimitry Andric createIntrinsic(R, Defs); 22950b57cec5SDimitry Andric 22960b57cec5SDimitry Andric for (auto *I : Defs) 22970b57cec5SDimitry Andric I->indexBody(); 22980b57cec5SDimitry Andric 2299a7dea167SDimitry Andric llvm::stable_sort(Defs, llvm::deref<std::less<>>()); 23000b57cec5SDimitry Andric 23010b57cec5SDimitry Andric // Only emit a def when its requirements have been met. 23020b57cec5SDimitry Andric // FIXME: This loop could be made faster, but it's fast enough for now. 23030b57cec5SDimitry Andric bool MadeProgress = true; 23040b57cec5SDimitry Andric std::string InGuard; 23050b57cec5SDimitry Andric while (!Defs.empty() && MadeProgress) { 23060b57cec5SDimitry Andric MadeProgress = false; 23070b57cec5SDimitry Andric 23080b57cec5SDimitry Andric for (SmallVector<Intrinsic *, 128>::iterator I = Defs.begin(); 23090b57cec5SDimitry Andric I != Defs.end(); /*No step*/) { 23100b57cec5SDimitry Andric bool DependenciesSatisfied = true; 23110b57cec5SDimitry Andric for (auto *II : (*I)->getDependencies()) { 23120b57cec5SDimitry Andric if (llvm::is_contained(Defs, II)) 23130b57cec5SDimitry Andric DependenciesSatisfied = false; 23140b57cec5SDimitry Andric } 23150b57cec5SDimitry Andric if (!DependenciesSatisfied) { 23160b57cec5SDimitry Andric // Try the next one. 23170b57cec5SDimitry Andric ++I; 23180b57cec5SDimitry Andric continue; 23190b57cec5SDimitry Andric } 23200b57cec5SDimitry Andric 23210b57cec5SDimitry Andric // Emit #endif/#if pair if needed. 23220b57cec5SDimitry Andric if ((*I)->getGuard() != InGuard) { 23230b57cec5SDimitry Andric if (!InGuard.empty()) 23240b57cec5SDimitry Andric OS << "#endif\n"; 23250b57cec5SDimitry Andric InGuard = (*I)->getGuard(); 23260b57cec5SDimitry Andric if (!InGuard.empty()) 23270b57cec5SDimitry Andric OS << "#if " << InGuard << "\n"; 23280b57cec5SDimitry Andric } 23290b57cec5SDimitry Andric 23300b57cec5SDimitry Andric // Actually generate the intrinsic code. 23310b57cec5SDimitry Andric OS << (*I)->generate(); 23320b57cec5SDimitry Andric 23330b57cec5SDimitry Andric MadeProgress = true; 23340b57cec5SDimitry Andric I = Defs.erase(I); 23350b57cec5SDimitry Andric } 23360b57cec5SDimitry Andric } 23370b57cec5SDimitry Andric assert(Defs.empty() && "Some requirements were not satisfied!"); 23380b57cec5SDimitry Andric if (!InGuard.empty()) 23390b57cec5SDimitry Andric OS << "#endif\n"; 23400b57cec5SDimitry Andric 23410b57cec5SDimitry Andric OS << "\n"; 23420b57cec5SDimitry Andric OS << "#undef __ai\n\n"; 23430b57cec5SDimitry Andric OS << "#endif /* __ARM_NEON_H */\n"; 23440b57cec5SDimitry Andric } 23450b57cec5SDimitry Andric 23460b57cec5SDimitry Andric /// run - Read the records in arm_fp16.td and output arm_fp16.h. arm_fp16.h 23470b57cec5SDimitry Andric /// is comprised of type definitions and function declarations. 23480b57cec5SDimitry Andric void NeonEmitter::runFP16(raw_ostream &OS) { 23490b57cec5SDimitry Andric OS << "/*===---- arm_fp16.h - ARM FP16 intrinsics " 23500b57cec5SDimitry Andric "------------------------------" 23510b57cec5SDimitry Andric "---===\n" 23520b57cec5SDimitry Andric " *\n" 23530b57cec5SDimitry Andric " * Permission is hereby granted, free of charge, to any person " 23540b57cec5SDimitry Andric "obtaining a copy\n" 23550b57cec5SDimitry Andric " * of this software and associated documentation files (the " 23560b57cec5SDimitry Andric "\"Software\"), to deal\n" 23570b57cec5SDimitry Andric " * in the Software without restriction, including without limitation " 23580b57cec5SDimitry Andric "the rights\n" 23590b57cec5SDimitry Andric " * to use, copy, modify, merge, publish, distribute, sublicense, " 23600b57cec5SDimitry Andric "and/or sell\n" 23610b57cec5SDimitry Andric " * copies of the Software, and to permit persons to whom the Software " 23620b57cec5SDimitry Andric "is\n" 23630b57cec5SDimitry Andric " * furnished to do so, subject to the following conditions:\n" 23640b57cec5SDimitry Andric " *\n" 23650b57cec5SDimitry Andric " * The above copyright notice and this permission notice shall be " 23660b57cec5SDimitry Andric "included in\n" 23670b57cec5SDimitry Andric " * all copies or substantial portions of the Software.\n" 23680b57cec5SDimitry Andric " *\n" 23690b57cec5SDimitry Andric " * THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, " 23700b57cec5SDimitry Andric "EXPRESS OR\n" 23710b57cec5SDimitry Andric " * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF " 23720b57cec5SDimitry Andric "MERCHANTABILITY,\n" 23730b57cec5SDimitry Andric " * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT " 23740b57cec5SDimitry Andric "SHALL THE\n" 23750b57cec5SDimitry Andric " * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR " 23760b57cec5SDimitry Andric "OTHER\n" 23770b57cec5SDimitry Andric " * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, " 23780b57cec5SDimitry Andric "ARISING FROM,\n" 23790b57cec5SDimitry Andric " * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER " 23800b57cec5SDimitry Andric "DEALINGS IN\n" 23810b57cec5SDimitry Andric " * THE SOFTWARE.\n" 23820b57cec5SDimitry Andric " *\n" 23830b57cec5SDimitry Andric " *===-----------------------------------------------------------------" 23840b57cec5SDimitry Andric "---" 23850b57cec5SDimitry Andric "---===\n" 23860b57cec5SDimitry Andric " */\n\n"; 23870b57cec5SDimitry Andric 23880b57cec5SDimitry Andric OS << "#ifndef __ARM_FP16_H\n"; 23890b57cec5SDimitry Andric OS << "#define __ARM_FP16_H\n\n"; 23900b57cec5SDimitry Andric 23910b57cec5SDimitry Andric OS << "#include <stdint.h>\n\n"; 23920b57cec5SDimitry Andric 23930b57cec5SDimitry Andric OS << "typedef __fp16 float16_t;\n"; 23940b57cec5SDimitry Andric 23950b57cec5SDimitry Andric OS << "#define __ai static __inline__ __attribute__((__always_inline__, " 23960b57cec5SDimitry Andric "__nodebug__))\n\n"; 23970b57cec5SDimitry Andric 23980b57cec5SDimitry Andric SmallVector<Intrinsic *, 128> Defs; 23990b57cec5SDimitry Andric std::vector<Record *> RV = Records.getAllDerivedDefinitions("Inst"); 24000b57cec5SDimitry Andric for (auto *R : RV) 24010b57cec5SDimitry Andric createIntrinsic(R, Defs); 24020b57cec5SDimitry Andric 24030b57cec5SDimitry Andric for (auto *I : Defs) 24040b57cec5SDimitry Andric I->indexBody(); 24050b57cec5SDimitry Andric 2406a7dea167SDimitry Andric llvm::stable_sort(Defs, llvm::deref<std::less<>>()); 24070b57cec5SDimitry Andric 24080b57cec5SDimitry Andric // Only emit a def when its requirements have been met. 24090b57cec5SDimitry Andric // FIXME: This loop could be made faster, but it's fast enough for now. 24100b57cec5SDimitry Andric bool MadeProgress = true; 24110b57cec5SDimitry Andric std::string InGuard; 24120b57cec5SDimitry Andric while (!Defs.empty() && MadeProgress) { 24130b57cec5SDimitry Andric MadeProgress = false; 24140b57cec5SDimitry Andric 24150b57cec5SDimitry Andric for (SmallVector<Intrinsic *, 128>::iterator I = Defs.begin(); 24160b57cec5SDimitry Andric I != Defs.end(); /*No step*/) { 24170b57cec5SDimitry Andric bool DependenciesSatisfied = true; 24180b57cec5SDimitry Andric for (auto *II : (*I)->getDependencies()) { 24190b57cec5SDimitry Andric if (llvm::is_contained(Defs, II)) 24200b57cec5SDimitry Andric DependenciesSatisfied = false; 24210b57cec5SDimitry Andric } 24220b57cec5SDimitry Andric if (!DependenciesSatisfied) { 24230b57cec5SDimitry Andric // Try the next one. 24240b57cec5SDimitry Andric ++I; 24250b57cec5SDimitry Andric continue; 24260b57cec5SDimitry Andric } 24270b57cec5SDimitry Andric 24280b57cec5SDimitry Andric // Emit #endif/#if pair if needed. 24290b57cec5SDimitry Andric if ((*I)->getGuard() != InGuard) { 24300b57cec5SDimitry Andric if (!InGuard.empty()) 24310b57cec5SDimitry Andric OS << "#endif\n"; 24320b57cec5SDimitry Andric InGuard = (*I)->getGuard(); 24330b57cec5SDimitry Andric if (!InGuard.empty()) 24340b57cec5SDimitry Andric OS << "#if " << InGuard << "\n"; 24350b57cec5SDimitry Andric } 24360b57cec5SDimitry Andric 24370b57cec5SDimitry Andric // Actually generate the intrinsic code. 24380b57cec5SDimitry Andric OS << (*I)->generate(); 24390b57cec5SDimitry Andric 24400b57cec5SDimitry Andric MadeProgress = true; 24410b57cec5SDimitry Andric I = Defs.erase(I); 24420b57cec5SDimitry Andric } 24430b57cec5SDimitry Andric } 24440b57cec5SDimitry Andric assert(Defs.empty() && "Some requirements were not satisfied!"); 24450b57cec5SDimitry Andric if (!InGuard.empty()) 24460b57cec5SDimitry Andric OS << "#endif\n"; 24470b57cec5SDimitry Andric 24480b57cec5SDimitry Andric OS << "\n"; 24490b57cec5SDimitry Andric OS << "#undef __ai\n\n"; 24500b57cec5SDimitry Andric OS << "#endif /* __ARM_FP16_H */\n"; 24510b57cec5SDimitry Andric } 24520b57cec5SDimitry Andric 2453a7dea167SDimitry Andric void clang::EmitNeon(RecordKeeper &Records, raw_ostream &OS) { 24540b57cec5SDimitry Andric NeonEmitter(Records).run(OS); 24550b57cec5SDimitry Andric } 24560b57cec5SDimitry Andric 2457a7dea167SDimitry Andric void clang::EmitFP16(RecordKeeper &Records, raw_ostream &OS) { 24580b57cec5SDimitry Andric NeonEmitter(Records).runFP16(OS); 24590b57cec5SDimitry Andric } 24600b57cec5SDimitry Andric 2461a7dea167SDimitry Andric void clang::EmitNeonSema(RecordKeeper &Records, raw_ostream &OS) { 24620b57cec5SDimitry Andric NeonEmitter(Records).runHeader(OS); 24630b57cec5SDimitry Andric } 24640b57cec5SDimitry Andric 2465a7dea167SDimitry Andric void clang::EmitNeonTest(RecordKeeper &Records, raw_ostream &OS) { 24660b57cec5SDimitry Andric llvm_unreachable("Neon test generation no longer implemented!"); 24670b57cec5SDimitry Andric } 2468