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 26*a7dea167SDimitry 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 1430b57cec5SDimitry Andric bool Float, Signed, Immediate, Void, Poly, Constant, Pointer; 1440b57cec5SDimitry Andric // ScalarForMangling and NoManglingQ are really not suited to live here as 1450b57cec5SDimitry Andric // they are not related to the type. But they live in the TypeSpec (not the 1460b57cec5SDimitry Andric // prototype), so this is really the only place to store them. 1470b57cec5SDimitry Andric bool ScalarForMangling, NoManglingQ; 1480b57cec5SDimitry Andric unsigned Bitwidth, ElementBitwidth, NumVectors; 1490b57cec5SDimitry Andric 1500b57cec5SDimitry Andric public: 1510b57cec5SDimitry Andric Type() 1520b57cec5SDimitry Andric : Float(false), Signed(false), Immediate(false), Void(true), Poly(false), 1530b57cec5SDimitry Andric Constant(false), Pointer(false), ScalarForMangling(false), 1540b57cec5SDimitry Andric NoManglingQ(false), Bitwidth(0), ElementBitwidth(0), NumVectors(0) {} 1550b57cec5SDimitry Andric 1560b57cec5SDimitry Andric Type(TypeSpec TS, char CharMod) 1570b57cec5SDimitry Andric : TS(std::move(TS)), Float(false), Signed(false), Immediate(false), 1580b57cec5SDimitry Andric Void(false), Poly(false), Constant(false), Pointer(false), 1590b57cec5SDimitry Andric ScalarForMangling(false), NoManglingQ(false), Bitwidth(0), 1600b57cec5SDimitry Andric ElementBitwidth(0), NumVectors(0) { 1610b57cec5SDimitry Andric applyModifier(CharMod); 1620b57cec5SDimitry Andric } 1630b57cec5SDimitry Andric 1640b57cec5SDimitry Andric /// Returns a type representing "void". 1650b57cec5SDimitry Andric static Type getVoid() { return Type(); } 1660b57cec5SDimitry Andric 1670b57cec5SDimitry Andric bool operator==(const Type &Other) const { return str() == Other.str(); } 1680b57cec5SDimitry Andric bool operator!=(const Type &Other) const { return !operator==(Other); } 1690b57cec5SDimitry Andric 1700b57cec5SDimitry Andric // 1710b57cec5SDimitry Andric // Query functions 1720b57cec5SDimitry Andric // 1730b57cec5SDimitry Andric bool isScalarForMangling() const { return ScalarForMangling; } 1740b57cec5SDimitry Andric bool noManglingQ() const { return NoManglingQ; } 1750b57cec5SDimitry Andric 1760b57cec5SDimitry Andric bool isPointer() const { return Pointer; } 1770b57cec5SDimitry Andric bool isFloating() const { return Float; } 1780b57cec5SDimitry Andric bool isInteger() const { return !Float && !Poly; } 1790b57cec5SDimitry Andric bool isSigned() const { return Signed; } 1800b57cec5SDimitry Andric bool isImmediate() const { return Immediate; } 1810b57cec5SDimitry Andric bool isScalar() const { return NumVectors == 0; } 1820b57cec5SDimitry Andric bool isVector() const { return NumVectors > 0; } 1830b57cec5SDimitry Andric bool isFloat() const { return Float && ElementBitwidth == 32; } 1840b57cec5SDimitry Andric bool isDouble() const { return Float && ElementBitwidth == 64; } 1850b57cec5SDimitry Andric bool isHalf() const { return Float && ElementBitwidth == 16; } 1860b57cec5SDimitry Andric bool isPoly() const { return Poly; } 1870b57cec5SDimitry Andric bool isChar() const { return ElementBitwidth == 8; } 1880b57cec5SDimitry Andric bool isShort() const { return !Float && ElementBitwidth == 16; } 1890b57cec5SDimitry Andric bool isInt() const { return !Float && ElementBitwidth == 32; } 1900b57cec5SDimitry Andric bool isLong() const { return !Float && ElementBitwidth == 64; } 1910b57cec5SDimitry Andric bool isVoid() const { return Void; } 1920b57cec5SDimitry Andric unsigned getNumElements() const { return Bitwidth / ElementBitwidth; } 1930b57cec5SDimitry Andric unsigned getSizeInBits() const { return Bitwidth; } 1940b57cec5SDimitry Andric unsigned getElementSizeInBits() const { return ElementBitwidth; } 1950b57cec5SDimitry Andric unsigned getNumVectors() const { return NumVectors; } 1960b57cec5SDimitry Andric 1970b57cec5SDimitry Andric // 1980b57cec5SDimitry Andric // Mutator functions 1990b57cec5SDimitry Andric // 2000b57cec5SDimitry Andric void makeUnsigned() { Signed = false; } 2010b57cec5SDimitry Andric void makeSigned() { Signed = true; } 2020b57cec5SDimitry Andric 2030b57cec5SDimitry Andric void makeInteger(unsigned ElemWidth, bool Sign) { 2040b57cec5SDimitry Andric Float = false; 2050b57cec5SDimitry Andric Poly = false; 2060b57cec5SDimitry Andric Signed = Sign; 2070b57cec5SDimitry Andric Immediate = false; 2080b57cec5SDimitry Andric ElementBitwidth = ElemWidth; 2090b57cec5SDimitry Andric } 2100b57cec5SDimitry Andric 2110b57cec5SDimitry Andric void makeImmediate(unsigned ElemWidth) { 2120b57cec5SDimitry Andric Float = false; 2130b57cec5SDimitry Andric Poly = false; 2140b57cec5SDimitry Andric Signed = true; 2150b57cec5SDimitry Andric Immediate = true; 2160b57cec5SDimitry Andric ElementBitwidth = ElemWidth; 2170b57cec5SDimitry Andric } 2180b57cec5SDimitry Andric 2190b57cec5SDimitry Andric void makeScalar() { 2200b57cec5SDimitry Andric Bitwidth = ElementBitwidth; 2210b57cec5SDimitry Andric NumVectors = 0; 2220b57cec5SDimitry Andric } 2230b57cec5SDimitry Andric 2240b57cec5SDimitry Andric void makeOneVector() { 2250b57cec5SDimitry Andric assert(isVector()); 2260b57cec5SDimitry Andric NumVectors = 1; 2270b57cec5SDimitry Andric } 2280b57cec5SDimitry Andric 2290b57cec5SDimitry Andric void doubleLanes() { 2300b57cec5SDimitry Andric assert_with_loc(Bitwidth != 128, "Can't get bigger than 128!"); 2310b57cec5SDimitry Andric Bitwidth = 128; 2320b57cec5SDimitry Andric } 2330b57cec5SDimitry Andric 2340b57cec5SDimitry Andric void halveLanes() { 2350b57cec5SDimitry Andric assert_with_loc(Bitwidth != 64, "Can't get smaller than 64!"); 2360b57cec5SDimitry Andric Bitwidth = 64; 2370b57cec5SDimitry Andric } 2380b57cec5SDimitry Andric 2390b57cec5SDimitry Andric /// Return the C string representation of a type, which is the typename 2400b57cec5SDimitry Andric /// defined in stdint.h or arm_neon.h. 2410b57cec5SDimitry Andric std::string str() const; 2420b57cec5SDimitry Andric 2430b57cec5SDimitry Andric /// Return the string representation of a type, which is an encoded 2440b57cec5SDimitry Andric /// string for passing to the BUILTIN() macro in Builtins.def. 2450b57cec5SDimitry Andric std::string builtin_str() const; 2460b57cec5SDimitry Andric 2470b57cec5SDimitry Andric /// Return the value in NeonTypeFlags for this type. 2480b57cec5SDimitry Andric unsigned getNeonEnum() const; 2490b57cec5SDimitry Andric 2500b57cec5SDimitry Andric /// Parse a type from a stdint.h or arm_neon.h typedef name, 2510b57cec5SDimitry Andric /// for example uint32x2_t or int64_t. 2520b57cec5SDimitry Andric static Type fromTypedefName(StringRef Name); 2530b57cec5SDimitry Andric 2540b57cec5SDimitry Andric private: 2550b57cec5SDimitry Andric /// Creates the type based on the typespec string in TS. 2560b57cec5SDimitry Andric /// Sets "Quad" to true if the "Q" or "H" modifiers were 2570b57cec5SDimitry Andric /// seen. This is needed by applyModifier as some modifiers 2580b57cec5SDimitry Andric /// only take effect if the type size was changed by "Q" or "H". 2590b57cec5SDimitry Andric void applyTypespec(bool &Quad); 2600b57cec5SDimitry Andric /// Applies a prototype modifier to the type. 2610b57cec5SDimitry Andric void applyModifier(char Mod); 2620b57cec5SDimitry Andric }; 2630b57cec5SDimitry Andric 2640b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 2650b57cec5SDimitry Andric // Variable 2660b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 2670b57cec5SDimitry Andric 2680b57cec5SDimitry Andric /// A variable is a simple class that just has a type and a name. 2690b57cec5SDimitry Andric class Variable { 2700b57cec5SDimitry Andric Type T; 2710b57cec5SDimitry Andric std::string N; 2720b57cec5SDimitry Andric 2730b57cec5SDimitry Andric public: 2740b57cec5SDimitry Andric Variable() : T(Type::getVoid()), N("") {} 2750b57cec5SDimitry Andric Variable(Type T, std::string N) : T(std::move(T)), N(std::move(N)) {} 2760b57cec5SDimitry Andric 2770b57cec5SDimitry Andric Type getType() const { return T; } 2780b57cec5SDimitry Andric std::string getName() const { return "__" + N; } 2790b57cec5SDimitry Andric }; 2800b57cec5SDimitry Andric 2810b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 2820b57cec5SDimitry Andric // Intrinsic 2830b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 2840b57cec5SDimitry Andric 2850b57cec5SDimitry Andric /// The main grunt class. This represents an instantiation of an intrinsic with 2860b57cec5SDimitry Andric /// a particular typespec and prototype. 2870b57cec5SDimitry Andric class Intrinsic { 2880b57cec5SDimitry Andric friend class DagEmitter; 2890b57cec5SDimitry Andric 2900b57cec5SDimitry Andric /// The Record this intrinsic was created from. 2910b57cec5SDimitry Andric Record *R; 2920b57cec5SDimitry Andric /// The unmangled name and prototype. 2930b57cec5SDimitry Andric std::string Name, Proto; 2940b57cec5SDimitry Andric /// The input and output typespecs. InTS == OutTS except when 2950b57cec5SDimitry Andric /// CartesianProductOfTypes is 1 - this is the case for vreinterpret. 2960b57cec5SDimitry Andric TypeSpec OutTS, InTS; 2970b57cec5SDimitry Andric /// The base class kind. Most intrinsics use ClassS, which has full type 2980b57cec5SDimitry Andric /// info for integers (s32/u32). Some use ClassI, which doesn't care about 2990b57cec5SDimitry Andric /// signedness (i32), while some (ClassB) have no type at all, only a width 3000b57cec5SDimitry Andric /// (32). 3010b57cec5SDimitry Andric ClassKind CK; 3020b57cec5SDimitry Andric /// The list of DAGs for the body. May be empty, in which case we should 3030b57cec5SDimitry Andric /// emit a builtin call. 3040b57cec5SDimitry Andric ListInit *Body; 3050b57cec5SDimitry Andric /// The architectural #ifdef guard. 3060b57cec5SDimitry Andric std::string Guard; 3070b57cec5SDimitry Andric /// Set if the Unavailable bit is 1. This means we don't generate a body, 3080b57cec5SDimitry Andric /// just an "unavailable" attribute on a declaration. 3090b57cec5SDimitry Andric bool IsUnavailable; 3100b57cec5SDimitry Andric /// Is this intrinsic safe for big-endian? or does it need its arguments 3110b57cec5SDimitry Andric /// reversing? 3120b57cec5SDimitry Andric bool BigEndianSafe; 3130b57cec5SDimitry Andric 3140b57cec5SDimitry Andric /// The types of return value [0] and parameters [1..]. 3150b57cec5SDimitry Andric std::vector<Type> Types; 3160b57cec5SDimitry Andric /// The local variables defined. 3170b57cec5SDimitry Andric std::map<std::string, Variable> Variables; 3180b57cec5SDimitry Andric /// NeededEarly - set if any other intrinsic depends on this intrinsic. 3190b57cec5SDimitry Andric bool NeededEarly; 3200b57cec5SDimitry Andric /// UseMacro - set if we should implement using a macro or unset for a 3210b57cec5SDimitry Andric /// function. 3220b57cec5SDimitry Andric bool UseMacro; 3230b57cec5SDimitry Andric /// The set of intrinsics that this intrinsic uses/requires. 3240b57cec5SDimitry Andric std::set<Intrinsic *> Dependencies; 3250b57cec5SDimitry Andric /// The "base type", which is Type('d', OutTS). InBaseType is only 3260b57cec5SDimitry Andric /// different if CartesianProductOfTypes = 1 (for vreinterpret). 3270b57cec5SDimitry Andric Type BaseType, InBaseType; 3280b57cec5SDimitry Andric /// The return variable. 3290b57cec5SDimitry Andric Variable RetVar; 3300b57cec5SDimitry Andric /// A postfix to apply to every variable. Defaults to "". 3310b57cec5SDimitry Andric std::string VariablePostfix; 3320b57cec5SDimitry Andric 3330b57cec5SDimitry Andric NeonEmitter &Emitter; 3340b57cec5SDimitry Andric std::stringstream OS; 3350b57cec5SDimitry Andric 336*a7dea167SDimitry Andric bool isBigEndianSafe() const { 337*a7dea167SDimitry Andric if (BigEndianSafe) 338*a7dea167SDimitry Andric return true; 339*a7dea167SDimitry Andric 340*a7dea167SDimitry Andric for (const auto &T : Types){ 341*a7dea167SDimitry Andric if (T.isVector() && T.getNumElements() > 1) 342*a7dea167SDimitry Andric return false; 343*a7dea167SDimitry Andric } 344*a7dea167SDimitry Andric return true; 345*a7dea167SDimitry Andric } 346*a7dea167SDimitry Andric 3470b57cec5SDimitry Andric public: 3480b57cec5SDimitry Andric Intrinsic(Record *R, StringRef Name, StringRef Proto, TypeSpec OutTS, 3490b57cec5SDimitry Andric TypeSpec InTS, ClassKind CK, ListInit *Body, NeonEmitter &Emitter, 3500b57cec5SDimitry Andric StringRef Guard, bool IsUnavailable, bool BigEndianSafe) 3510b57cec5SDimitry Andric : R(R), Name(Name.str()), Proto(Proto.str()), OutTS(OutTS), InTS(InTS), 3520b57cec5SDimitry Andric CK(CK), Body(Body), Guard(Guard.str()), IsUnavailable(IsUnavailable), 3530b57cec5SDimitry Andric BigEndianSafe(BigEndianSafe), NeededEarly(false), UseMacro(false), 3540b57cec5SDimitry Andric BaseType(OutTS, 'd'), InBaseType(InTS, 'd'), Emitter(Emitter) { 3550b57cec5SDimitry Andric // If this builtin takes an immediate argument, we need to #define it rather 3560b57cec5SDimitry Andric // than use a standard declaration, so that SemaChecking can range check 3570b57cec5SDimitry Andric // the immediate passed by the user. 3580b57cec5SDimitry Andric if (Proto.find('i') != std::string::npos) 3590b57cec5SDimitry Andric UseMacro = true; 3600b57cec5SDimitry Andric 3610b57cec5SDimitry Andric // Pointer arguments need to use macros to avoid hiding aligned attributes 3620b57cec5SDimitry Andric // from the pointer type. 3630b57cec5SDimitry Andric if (Proto.find('p') != std::string::npos || 3640b57cec5SDimitry Andric Proto.find('c') != std::string::npos) 3650b57cec5SDimitry Andric UseMacro = true; 3660b57cec5SDimitry Andric 3670b57cec5SDimitry Andric // It is not permitted to pass or return an __fp16 by value, so intrinsics 3680b57cec5SDimitry Andric // taking a scalar float16_t must be implemented as macros. 3690b57cec5SDimitry Andric if (OutTS.find('h') != std::string::npos && 3700b57cec5SDimitry Andric Proto.find('s') != std::string::npos) 3710b57cec5SDimitry Andric UseMacro = true; 3720b57cec5SDimitry Andric 3730b57cec5SDimitry Andric // Modify the TypeSpec per-argument to get a concrete Type, and create 3740b57cec5SDimitry Andric // known variables for each. 3750b57cec5SDimitry Andric // Types[0] is the return value. 3760b57cec5SDimitry Andric Types.emplace_back(OutTS, Proto[0]); 3770b57cec5SDimitry Andric for (unsigned I = 1; I < Proto.size(); ++I) 3780b57cec5SDimitry Andric Types.emplace_back(InTS, Proto[I]); 3790b57cec5SDimitry Andric } 3800b57cec5SDimitry Andric 3810b57cec5SDimitry Andric /// Get the Record that this intrinsic is based off. 3820b57cec5SDimitry Andric Record *getRecord() const { return R; } 3830b57cec5SDimitry Andric /// Get the set of Intrinsics that this intrinsic calls. 3840b57cec5SDimitry Andric /// this is the set of immediate dependencies, NOT the 3850b57cec5SDimitry Andric /// transitive closure. 3860b57cec5SDimitry Andric const std::set<Intrinsic *> &getDependencies() const { return Dependencies; } 3870b57cec5SDimitry Andric /// Get the architectural guard string (#ifdef). 3880b57cec5SDimitry Andric std::string getGuard() const { return Guard; } 3890b57cec5SDimitry Andric /// Get the non-mangled name. 3900b57cec5SDimitry Andric std::string getName() const { return Name; } 3910b57cec5SDimitry Andric 3920b57cec5SDimitry Andric /// Return true if the intrinsic takes an immediate operand. 3930b57cec5SDimitry Andric bool hasImmediate() const { 3940b57cec5SDimitry Andric return Proto.find('i') != std::string::npos; 3950b57cec5SDimitry Andric } 3960b57cec5SDimitry Andric 3970b57cec5SDimitry Andric /// Return the parameter index of the immediate operand. 3980b57cec5SDimitry Andric unsigned getImmediateIdx() const { 3990b57cec5SDimitry Andric assert(hasImmediate()); 4000b57cec5SDimitry Andric unsigned Idx = Proto.find('i'); 4010b57cec5SDimitry Andric assert(Idx > 0 && "Can't return an immediate!"); 4020b57cec5SDimitry Andric return Idx - 1; 4030b57cec5SDimitry Andric } 4040b57cec5SDimitry Andric 4050b57cec5SDimitry Andric /// Return true if the intrinsic takes an splat operand. 4060b57cec5SDimitry Andric bool hasSplat() const { return Proto.find('a') != std::string::npos; } 4070b57cec5SDimitry Andric 4080b57cec5SDimitry Andric /// Return the parameter index of the splat operand. 4090b57cec5SDimitry Andric unsigned getSplatIdx() const { 4100b57cec5SDimitry Andric assert(hasSplat()); 4110b57cec5SDimitry Andric unsigned Idx = Proto.find('a'); 4120b57cec5SDimitry Andric assert(Idx > 0 && "Can't return a splat!"); 4130b57cec5SDimitry Andric return Idx - 1; 4140b57cec5SDimitry Andric } 4150b57cec5SDimitry Andric 4160b57cec5SDimitry Andric unsigned getNumParams() const { return Proto.size() - 1; } 4170b57cec5SDimitry Andric Type getReturnType() const { return Types[0]; } 4180b57cec5SDimitry Andric Type getParamType(unsigned I) const { return Types[I + 1]; } 4190b57cec5SDimitry Andric Type getBaseType() const { return BaseType; } 4200b57cec5SDimitry Andric /// Return the raw prototype string. 4210b57cec5SDimitry Andric std::string getProto() const { return Proto; } 4220b57cec5SDimitry Andric 4230b57cec5SDimitry Andric /// Return true if the prototype has a scalar argument. 4240b57cec5SDimitry Andric /// This does not return true for the "splat" code ('a'). 4250b57cec5SDimitry Andric bool protoHasScalar() const; 4260b57cec5SDimitry Andric 4270b57cec5SDimitry Andric /// Return the index that parameter PIndex will sit at 4280b57cec5SDimitry Andric /// in a generated function call. This is often just PIndex, 4290b57cec5SDimitry Andric /// but may not be as things such as multiple-vector operands 4300b57cec5SDimitry Andric /// and sret parameters need to be taken into accont. 4310b57cec5SDimitry Andric unsigned getGeneratedParamIdx(unsigned PIndex) { 4320b57cec5SDimitry Andric unsigned Idx = 0; 4330b57cec5SDimitry Andric if (getReturnType().getNumVectors() > 1) 4340b57cec5SDimitry Andric // Multiple vectors are passed as sret. 4350b57cec5SDimitry Andric ++Idx; 4360b57cec5SDimitry Andric 4370b57cec5SDimitry Andric for (unsigned I = 0; I < PIndex; ++I) 4380b57cec5SDimitry Andric Idx += std::max(1U, getParamType(I).getNumVectors()); 4390b57cec5SDimitry Andric 4400b57cec5SDimitry Andric return Idx; 4410b57cec5SDimitry Andric } 4420b57cec5SDimitry Andric 4430b57cec5SDimitry Andric bool hasBody() const { return Body && !Body->getValues().empty(); } 4440b57cec5SDimitry Andric 4450b57cec5SDimitry Andric void setNeededEarly() { NeededEarly = true; } 4460b57cec5SDimitry Andric 4470b57cec5SDimitry Andric bool operator<(const Intrinsic &Other) const { 4480b57cec5SDimitry Andric // Sort lexicographically on a two-tuple (Guard, Name) 4490b57cec5SDimitry Andric if (Guard != Other.Guard) 4500b57cec5SDimitry Andric return Guard < Other.Guard; 4510b57cec5SDimitry Andric return Name < Other.Name; 4520b57cec5SDimitry Andric } 4530b57cec5SDimitry Andric 4540b57cec5SDimitry Andric ClassKind getClassKind(bool UseClassBIfScalar = false) { 4550b57cec5SDimitry Andric if (UseClassBIfScalar && !protoHasScalar()) 4560b57cec5SDimitry Andric return ClassB; 4570b57cec5SDimitry Andric return CK; 4580b57cec5SDimitry Andric } 4590b57cec5SDimitry Andric 4600b57cec5SDimitry Andric /// Return the name, mangled with type information. 4610b57cec5SDimitry Andric /// If ForceClassS is true, use ClassS (u32/s32) instead 4620b57cec5SDimitry Andric /// of the intrinsic's own type class. 4630b57cec5SDimitry Andric std::string getMangledName(bool ForceClassS = false) const; 4640b57cec5SDimitry Andric /// Return the type code for a builtin function call. 4650b57cec5SDimitry Andric std::string getInstTypeCode(Type T, ClassKind CK) const; 4660b57cec5SDimitry Andric /// Return the type string for a BUILTIN() macro in Builtins.def. 4670b57cec5SDimitry Andric std::string getBuiltinTypeStr(); 4680b57cec5SDimitry Andric 4690b57cec5SDimitry Andric /// Generate the intrinsic, returning code. 4700b57cec5SDimitry Andric std::string generate(); 4710b57cec5SDimitry Andric /// Perform type checking and populate the dependency graph, but 4720b57cec5SDimitry Andric /// don't generate code yet. 4730b57cec5SDimitry Andric void indexBody(); 4740b57cec5SDimitry Andric 4750b57cec5SDimitry Andric private: 4760b57cec5SDimitry Andric std::string mangleName(std::string Name, ClassKind CK) const; 4770b57cec5SDimitry Andric 4780b57cec5SDimitry Andric void initVariables(); 4790b57cec5SDimitry Andric std::string replaceParamsIn(std::string S); 4800b57cec5SDimitry Andric 4810b57cec5SDimitry Andric void emitBodyAsBuiltinCall(); 4820b57cec5SDimitry Andric 4830b57cec5SDimitry Andric void generateImpl(bool ReverseArguments, 4840b57cec5SDimitry Andric StringRef NamePrefix, StringRef CallPrefix); 4850b57cec5SDimitry Andric void emitReturn(); 4860b57cec5SDimitry Andric void emitBody(StringRef CallPrefix); 4870b57cec5SDimitry Andric void emitShadowedArgs(); 4880b57cec5SDimitry Andric void emitArgumentReversal(); 4890b57cec5SDimitry Andric void emitReturnReversal(); 4900b57cec5SDimitry Andric void emitReverseVariable(Variable &Dest, Variable &Src); 4910b57cec5SDimitry Andric void emitNewLine(); 4920b57cec5SDimitry Andric void emitClosingBrace(); 4930b57cec5SDimitry Andric void emitOpeningBrace(); 4940b57cec5SDimitry Andric void emitPrototype(StringRef NamePrefix); 4950b57cec5SDimitry Andric 4960b57cec5SDimitry Andric class DagEmitter { 4970b57cec5SDimitry Andric Intrinsic &Intr; 4980b57cec5SDimitry Andric StringRef CallPrefix; 4990b57cec5SDimitry Andric 5000b57cec5SDimitry Andric public: 5010b57cec5SDimitry Andric DagEmitter(Intrinsic &Intr, StringRef CallPrefix) : 5020b57cec5SDimitry Andric Intr(Intr), CallPrefix(CallPrefix) { 5030b57cec5SDimitry Andric } 5040b57cec5SDimitry Andric std::pair<Type, std::string> emitDagArg(Init *Arg, std::string ArgName); 5050b57cec5SDimitry Andric std::pair<Type, std::string> emitDagSaveTemp(DagInit *DI); 5060b57cec5SDimitry Andric std::pair<Type, std::string> emitDagSplat(DagInit *DI); 5070b57cec5SDimitry Andric std::pair<Type, std::string> emitDagDup(DagInit *DI); 5080b57cec5SDimitry Andric std::pair<Type, std::string> emitDagDupTyped(DagInit *DI); 5090b57cec5SDimitry Andric std::pair<Type, std::string> emitDagShuffle(DagInit *DI); 5100b57cec5SDimitry Andric std::pair<Type, std::string> emitDagCast(DagInit *DI, bool IsBitCast); 5110b57cec5SDimitry Andric std::pair<Type, std::string> emitDagCall(DagInit *DI); 5120b57cec5SDimitry Andric std::pair<Type, std::string> emitDagNameReplace(DagInit *DI); 5130b57cec5SDimitry Andric std::pair<Type, std::string> emitDagLiteral(DagInit *DI); 5140b57cec5SDimitry Andric std::pair<Type, std::string> emitDagOp(DagInit *DI); 5150b57cec5SDimitry Andric std::pair<Type, std::string> emitDag(DagInit *DI); 5160b57cec5SDimitry Andric }; 5170b57cec5SDimitry Andric }; 5180b57cec5SDimitry Andric 5190b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 5200b57cec5SDimitry Andric // NeonEmitter 5210b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 5220b57cec5SDimitry Andric 5230b57cec5SDimitry Andric class NeonEmitter { 5240b57cec5SDimitry Andric RecordKeeper &Records; 5250b57cec5SDimitry Andric DenseMap<Record *, ClassKind> ClassMap; 5260b57cec5SDimitry Andric std::map<std::string, std::deque<Intrinsic>> IntrinsicMap; 5270b57cec5SDimitry Andric unsigned UniqueNumber; 5280b57cec5SDimitry Andric 5290b57cec5SDimitry Andric void createIntrinsic(Record *R, SmallVectorImpl<Intrinsic *> &Out); 5300b57cec5SDimitry Andric void genBuiltinsDef(raw_ostream &OS, SmallVectorImpl<Intrinsic *> &Defs); 5310b57cec5SDimitry Andric void genOverloadTypeCheckCode(raw_ostream &OS, 5320b57cec5SDimitry Andric SmallVectorImpl<Intrinsic *> &Defs); 5330b57cec5SDimitry Andric void genIntrinsicRangeCheckCode(raw_ostream &OS, 5340b57cec5SDimitry Andric SmallVectorImpl<Intrinsic *> &Defs); 5350b57cec5SDimitry Andric 5360b57cec5SDimitry Andric public: 5370b57cec5SDimitry Andric /// Called by Intrinsic - this attempts to get an intrinsic that takes 5380b57cec5SDimitry Andric /// the given types as arguments. 5390b57cec5SDimitry Andric Intrinsic &getIntrinsic(StringRef Name, ArrayRef<Type> Types); 5400b57cec5SDimitry Andric 5410b57cec5SDimitry Andric /// Called by Intrinsic - returns a globally-unique number. 5420b57cec5SDimitry Andric unsigned getUniqueNumber() { return UniqueNumber++; } 5430b57cec5SDimitry Andric 5440b57cec5SDimitry Andric NeonEmitter(RecordKeeper &R) : Records(R), UniqueNumber(0) { 5450b57cec5SDimitry Andric Record *SI = R.getClass("SInst"); 5460b57cec5SDimitry Andric Record *II = R.getClass("IInst"); 5470b57cec5SDimitry Andric Record *WI = R.getClass("WInst"); 5480b57cec5SDimitry Andric Record *SOpI = R.getClass("SOpInst"); 5490b57cec5SDimitry Andric Record *IOpI = R.getClass("IOpInst"); 5500b57cec5SDimitry Andric Record *WOpI = R.getClass("WOpInst"); 5510b57cec5SDimitry Andric Record *LOpI = R.getClass("LOpInst"); 5520b57cec5SDimitry Andric Record *NoTestOpI = R.getClass("NoTestOpInst"); 5530b57cec5SDimitry Andric 5540b57cec5SDimitry Andric ClassMap[SI] = ClassS; 5550b57cec5SDimitry Andric ClassMap[II] = ClassI; 5560b57cec5SDimitry Andric ClassMap[WI] = ClassW; 5570b57cec5SDimitry Andric ClassMap[SOpI] = ClassS; 5580b57cec5SDimitry Andric ClassMap[IOpI] = ClassI; 5590b57cec5SDimitry Andric ClassMap[WOpI] = ClassW; 5600b57cec5SDimitry Andric ClassMap[LOpI] = ClassL; 5610b57cec5SDimitry Andric ClassMap[NoTestOpI] = ClassNoTest; 5620b57cec5SDimitry Andric } 5630b57cec5SDimitry Andric 5640b57cec5SDimitry Andric // run - Emit arm_neon.h.inc 5650b57cec5SDimitry Andric void run(raw_ostream &o); 5660b57cec5SDimitry Andric 5670b57cec5SDimitry Andric // runFP16 - Emit arm_fp16.h.inc 5680b57cec5SDimitry Andric void runFP16(raw_ostream &o); 5690b57cec5SDimitry Andric 5700b57cec5SDimitry Andric // runHeader - Emit all the __builtin prototypes used in arm_neon.h 5710b57cec5SDimitry Andric // and arm_fp16.h 5720b57cec5SDimitry Andric void runHeader(raw_ostream &o); 5730b57cec5SDimitry Andric 5740b57cec5SDimitry Andric // runTests - Emit tests for all the Neon intrinsics. 5750b57cec5SDimitry Andric void runTests(raw_ostream &o); 5760b57cec5SDimitry Andric }; 5770b57cec5SDimitry Andric 5780b57cec5SDimitry Andric } // end anonymous namespace 5790b57cec5SDimitry Andric 5800b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 5810b57cec5SDimitry Andric // Type implementation 5820b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 5830b57cec5SDimitry Andric 5840b57cec5SDimitry Andric std::string Type::str() const { 5850b57cec5SDimitry Andric if (Void) 5860b57cec5SDimitry Andric return "void"; 5870b57cec5SDimitry Andric std::string S; 5880b57cec5SDimitry Andric 5890b57cec5SDimitry Andric if (!Signed && isInteger()) 5900b57cec5SDimitry Andric S += "u"; 5910b57cec5SDimitry Andric 5920b57cec5SDimitry Andric if (Poly) 5930b57cec5SDimitry Andric S += "poly"; 5940b57cec5SDimitry Andric else if (Float) 5950b57cec5SDimitry Andric S += "float"; 5960b57cec5SDimitry Andric else 5970b57cec5SDimitry Andric S += "int"; 5980b57cec5SDimitry Andric 5990b57cec5SDimitry Andric S += utostr(ElementBitwidth); 6000b57cec5SDimitry Andric if (isVector()) 6010b57cec5SDimitry Andric S += "x" + utostr(getNumElements()); 6020b57cec5SDimitry Andric if (NumVectors > 1) 6030b57cec5SDimitry Andric S += "x" + utostr(NumVectors); 6040b57cec5SDimitry Andric S += "_t"; 6050b57cec5SDimitry Andric 6060b57cec5SDimitry Andric if (Constant) 6070b57cec5SDimitry Andric S += " const"; 6080b57cec5SDimitry Andric if (Pointer) 6090b57cec5SDimitry Andric S += " *"; 6100b57cec5SDimitry Andric 6110b57cec5SDimitry Andric return S; 6120b57cec5SDimitry Andric } 6130b57cec5SDimitry Andric 6140b57cec5SDimitry Andric std::string Type::builtin_str() const { 6150b57cec5SDimitry Andric std::string S; 6160b57cec5SDimitry Andric if (isVoid()) 6170b57cec5SDimitry Andric return "v"; 6180b57cec5SDimitry Andric 6190b57cec5SDimitry Andric if (Pointer) 6200b57cec5SDimitry Andric // All pointers are void pointers. 6210b57cec5SDimitry Andric S += "v"; 6220b57cec5SDimitry Andric else if (isInteger()) 6230b57cec5SDimitry Andric switch (ElementBitwidth) { 6240b57cec5SDimitry Andric case 8: S += "c"; break; 6250b57cec5SDimitry Andric case 16: S += "s"; break; 6260b57cec5SDimitry Andric case 32: S += "i"; break; 6270b57cec5SDimitry Andric case 64: S += "Wi"; break; 6280b57cec5SDimitry Andric case 128: S += "LLLi"; break; 6290b57cec5SDimitry Andric default: llvm_unreachable("Unhandled case!"); 6300b57cec5SDimitry Andric } 6310b57cec5SDimitry Andric else 6320b57cec5SDimitry Andric switch (ElementBitwidth) { 6330b57cec5SDimitry Andric case 16: S += "h"; break; 6340b57cec5SDimitry Andric case 32: S += "f"; break; 6350b57cec5SDimitry Andric case 64: S += "d"; break; 6360b57cec5SDimitry Andric default: llvm_unreachable("Unhandled case!"); 6370b57cec5SDimitry Andric } 6380b57cec5SDimitry Andric 639*a7dea167SDimitry Andric if (isChar() && !Pointer && Signed) 6400b57cec5SDimitry Andric // Make chars explicitly signed. 6410b57cec5SDimitry Andric S = "S" + S; 6420b57cec5SDimitry Andric else if (isInteger() && !Pointer && !Signed) 6430b57cec5SDimitry Andric S = "U" + S; 6440b57cec5SDimitry Andric 6450b57cec5SDimitry Andric // Constant indices are "int", but have the "constant expression" modifier. 6460b57cec5SDimitry Andric if (isImmediate()) { 6470b57cec5SDimitry Andric assert(isInteger() && isSigned()); 6480b57cec5SDimitry Andric S = "I" + S; 6490b57cec5SDimitry Andric } 6500b57cec5SDimitry Andric 6510b57cec5SDimitry Andric if (isScalar()) { 6520b57cec5SDimitry Andric if (Constant) S += "C"; 6530b57cec5SDimitry Andric if (Pointer) S += "*"; 6540b57cec5SDimitry Andric return S; 6550b57cec5SDimitry Andric } 6560b57cec5SDimitry Andric 6570b57cec5SDimitry Andric std::string Ret; 6580b57cec5SDimitry Andric for (unsigned I = 0; I < NumVectors; ++I) 6590b57cec5SDimitry Andric Ret += "V" + utostr(getNumElements()) + S; 6600b57cec5SDimitry Andric 6610b57cec5SDimitry Andric return Ret; 6620b57cec5SDimitry Andric } 6630b57cec5SDimitry Andric 6640b57cec5SDimitry Andric unsigned Type::getNeonEnum() const { 6650b57cec5SDimitry Andric unsigned Addend; 6660b57cec5SDimitry Andric switch (ElementBitwidth) { 6670b57cec5SDimitry Andric case 8: Addend = 0; break; 6680b57cec5SDimitry Andric case 16: Addend = 1; break; 6690b57cec5SDimitry Andric case 32: Addend = 2; break; 6700b57cec5SDimitry Andric case 64: Addend = 3; break; 6710b57cec5SDimitry Andric case 128: Addend = 4; break; 6720b57cec5SDimitry Andric default: llvm_unreachable("Unhandled element bitwidth!"); 6730b57cec5SDimitry Andric } 6740b57cec5SDimitry Andric 6750b57cec5SDimitry Andric unsigned Base = (unsigned)NeonTypeFlags::Int8 + Addend; 6760b57cec5SDimitry Andric if (Poly) { 6770b57cec5SDimitry Andric // Adjustment needed because Poly32 doesn't exist. 6780b57cec5SDimitry Andric if (Addend >= 2) 6790b57cec5SDimitry Andric --Addend; 6800b57cec5SDimitry Andric Base = (unsigned)NeonTypeFlags::Poly8 + Addend; 6810b57cec5SDimitry Andric } 6820b57cec5SDimitry Andric if (Float) { 6830b57cec5SDimitry Andric assert(Addend != 0 && "Float8 doesn't exist!"); 6840b57cec5SDimitry Andric Base = (unsigned)NeonTypeFlags::Float16 + (Addend - 1); 6850b57cec5SDimitry Andric } 6860b57cec5SDimitry Andric 6870b57cec5SDimitry Andric if (Bitwidth == 128) 6880b57cec5SDimitry Andric Base |= (unsigned)NeonTypeFlags::QuadFlag; 6890b57cec5SDimitry Andric if (isInteger() && !Signed) 6900b57cec5SDimitry Andric Base |= (unsigned)NeonTypeFlags::UnsignedFlag; 6910b57cec5SDimitry Andric 6920b57cec5SDimitry Andric return Base; 6930b57cec5SDimitry Andric } 6940b57cec5SDimitry Andric 6950b57cec5SDimitry Andric Type Type::fromTypedefName(StringRef Name) { 6960b57cec5SDimitry Andric Type T; 6970b57cec5SDimitry Andric T.Void = false; 6980b57cec5SDimitry Andric T.Float = false; 6990b57cec5SDimitry Andric T.Poly = false; 7000b57cec5SDimitry Andric 7010b57cec5SDimitry Andric if (Name.front() == 'u') { 7020b57cec5SDimitry Andric T.Signed = false; 7030b57cec5SDimitry Andric Name = Name.drop_front(); 7040b57cec5SDimitry Andric } else { 7050b57cec5SDimitry Andric T.Signed = true; 7060b57cec5SDimitry Andric } 7070b57cec5SDimitry Andric 7080b57cec5SDimitry Andric if (Name.startswith("float")) { 7090b57cec5SDimitry Andric T.Float = true; 7100b57cec5SDimitry Andric Name = Name.drop_front(5); 7110b57cec5SDimitry Andric } else if (Name.startswith("poly")) { 7120b57cec5SDimitry Andric T.Poly = true; 7130b57cec5SDimitry Andric Name = Name.drop_front(4); 7140b57cec5SDimitry Andric } else { 7150b57cec5SDimitry Andric assert(Name.startswith("int")); 7160b57cec5SDimitry Andric Name = Name.drop_front(3); 7170b57cec5SDimitry Andric } 7180b57cec5SDimitry Andric 7190b57cec5SDimitry Andric unsigned I = 0; 7200b57cec5SDimitry Andric for (I = 0; I < Name.size(); ++I) { 7210b57cec5SDimitry Andric if (!isdigit(Name[I])) 7220b57cec5SDimitry Andric break; 7230b57cec5SDimitry Andric } 7240b57cec5SDimitry Andric Name.substr(0, I).getAsInteger(10, T.ElementBitwidth); 7250b57cec5SDimitry Andric Name = Name.drop_front(I); 7260b57cec5SDimitry Andric 7270b57cec5SDimitry Andric T.Bitwidth = T.ElementBitwidth; 7280b57cec5SDimitry Andric T.NumVectors = 1; 7290b57cec5SDimitry Andric 7300b57cec5SDimitry Andric if (Name.front() == 'x') { 7310b57cec5SDimitry Andric Name = Name.drop_front(); 7320b57cec5SDimitry Andric unsigned I = 0; 7330b57cec5SDimitry Andric for (I = 0; I < Name.size(); ++I) { 7340b57cec5SDimitry Andric if (!isdigit(Name[I])) 7350b57cec5SDimitry Andric break; 7360b57cec5SDimitry Andric } 7370b57cec5SDimitry Andric unsigned NumLanes; 7380b57cec5SDimitry Andric Name.substr(0, I).getAsInteger(10, NumLanes); 7390b57cec5SDimitry Andric Name = Name.drop_front(I); 7400b57cec5SDimitry Andric T.Bitwidth = T.ElementBitwidth * NumLanes; 7410b57cec5SDimitry Andric } else { 7420b57cec5SDimitry Andric // Was scalar. 7430b57cec5SDimitry Andric T.NumVectors = 0; 7440b57cec5SDimitry Andric } 7450b57cec5SDimitry Andric if (Name.front() == 'x') { 7460b57cec5SDimitry Andric Name = Name.drop_front(); 7470b57cec5SDimitry Andric unsigned I = 0; 7480b57cec5SDimitry Andric for (I = 0; I < Name.size(); ++I) { 7490b57cec5SDimitry Andric if (!isdigit(Name[I])) 7500b57cec5SDimitry Andric break; 7510b57cec5SDimitry Andric } 7520b57cec5SDimitry Andric Name.substr(0, I).getAsInteger(10, T.NumVectors); 7530b57cec5SDimitry Andric Name = Name.drop_front(I); 7540b57cec5SDimitry Andric } 7550b57cec5SDimitry Andric 7560b57cec5SDimitry Andric assert(Name.startswith("_t") && "Malformed typedef!"); 7570b57cec5SDimitry Andric return T; 7580b57cec5SDimitry Andric } 7590b57cec5SDimitry Andric 7600b57cec5SDimitry Andric void Type::applyTypespec(bool &Quad) { 7610b57cec5SDimitry Andric std::string S = TS; 7620b57cec5SDimitry Andric ScalarForMangling = false; 7630b57cec5SDimitry Andric Void = false; 7640b57cec5SDimitry Andric Poly = Float = false; 7650b57cec5SDimitry Andric ElementBitwidth = ~0U; 7660b57cec5SDimitry Andric Signed = true; 7670b57cec5SDimitry Andric NumVectors = 1; 7680b57cec5SDimitry Andric 7690b57cec5SDimitry Andric for (char I : S) { 7700b57cec5SDimitry Andric switch (I) { 7710b57cec5SDimitry Andric case 'S': 7720b57cec5SDimitry Andric ScalarForMangling = true; 7730b57cec5SDimitry Andric break; 7740b57cec5SDimitry Andric case 'H': 7750b57cec5SDimitry Andric NoManglingQ = true; 7760b57cec5SDimitry Andric Quad = true; 7770b57cec5SDimitry Andric break; 7780b57cec5SDimitry Andric case 'Q': 7790b57cec5SDimitry Andric Quad = true; 7800b57cec5SDimitry Andric break; 7810b57cec5SDimitry Andric case 'P': 7820b57cec5SDimitry Andric Poly = true; 7830b57cec5SDimitry Andric break; 7840b57cec5SDimitry Andric case 'U': 7850b57cec5SDimitry Andric Signed = false; 7860b57cec5SDimitry Andric break; 7870b57cec5SDimitry Andric case 'c': 7880b57cec5SDimitry Andric ElementBitwidth = 8; 7890b57cec5SDimitry Andric break; 7900b57cec5SDimitry Andric case 'h': 7910b57cec5SDimitry Andric Float = true; 7920b57cec5SDimitry Andric LLVM_FALLTHROUGH; 7930b57cec5SDimitry Andric case 's': 7940b57cec5SDimitry Andric ElementBitwidth = 16; 7950b57cec5SDimitry Andric break; 7960b57cec5SDimitry Andric case 'f': 7970b57cec5SDimitry Andric Float = true; 7980b57cec5SDimitry Andric LLVM_FALLTHROUGH; 7990b57cec5SDimitry Andric case 'i': 8000b57cec5SDimitry Andric ElementBitwidth = 32; 8010b57cec5SDimitry Andric break; 8020b57cec5SDimitry Andric case 'd': 8030b57cec5SDimitry Andric Float = true; 8040b57cec5SDimitry Andric LLVM_FALLTHROUGH; 8050b57cec5SDimitry Andric case 'l': 8060b57cec5SDimitry Andric ElementBitwidth = 64; 8070b57cec5SDimitry Andric break; 8080b57cec5SDimitry Andric case 'k': 8090b57cec5SDimitry Andric ElementBitwidth = 128; 8100b57cec5SDimitry Andric // Poly doesn't have a 128x1 type. 8110b57cec5SDimitry Andric if (Poly) 8120b57cec5SDimitry Andric NumVectors = 0; 8130b57cec5SDimitry Andric break; 8140b57cec5SDimitry Andric default: 8150b57cec5SDimitry Andric llvm_unreachable("Unhandled type code!"); 8160b57cec5SDimitry Andric } 8170b57cec5SDimitry Andric } 8180b57cec5SDimitry Andric assert(ElementBitwidth != ~0U && "Bad element bitwidth!"); 8190b57cec5SDimitry Andric 8200b57cec5SDimitry Andric Bitwidth = Quad ? 128 : 64; 8210b57cec5SDimitry Andric } 8220b57cec5SDimitry Andric 8230b57cec5SDimitry Andric void Type::applyModifier(char Mod) { 8240b57cec5SDimitry Andric bool AppliedQuad = false; 8250b57cec5SDimitry Andric applyTypespec(AppliedQuad); 8260b57cec5SDimitry Andric 8270b57cec5SDimitry Andric switch (Mod) { 8280b57cec5SDimitry Andric case 'v': 8290b57cec5SDimitry Andric Void = true; 8300b57cec5SDimitry Andric break; 8310b57cec5SDimitry Andric case 't': 8320b57cec5SDimitry Andric if (Poly) { 8330b57cec5SDimitry Andric Poly = false; 8340b57cec5SDimitry Andric Signed = false; 8350b57cec5SDimitry Andric } 8360b57cec5SDimitry Andric break; 8370b57cec5SDimitry Andric case 'b': 8380b57cec5SDimitry Andric Signed = false; 8390b57cec5SDimitry Andric Float = false; 8400b57cec5SDimitry Andric Poly = false; 8410b57cec5SDimitry Andric NumVectors = 0; 8420b57cec5SDimitry Andric Bitwidth = ElementBitwidth; 8430b57cec5SDimitry Andric break; 8440b57cec5SDimitry Andric case '$': 8450b57cec5SDimitry Andric Signed = true; 8460b57cec5SDimitry Andric Float = false; 8470b57cec5SDimitry Andric Poly = false; 8480b57cec5SDimitry Andric NumVectors = 0; 8490b57cec5SDimitry Andric Bitwidth = ElementBitwidth; 8500b57cec5SDimitry Andric break; 8510b57cec5SDimitry Andric case 'u': 8520b57cec5SDimitry Andric Signed = false; 8530b57cec5SDimitry Andric Poly = false; 8540b57cec5SDimitry Andric Float = false; 8550b57cec5SDimitry Andric break; 8560b57cec5SDimitry Andric case 'x': 8570b57cec5SDimitry Andric Signed = true; 8580b57cec5SDimitry Andric assert(!Poly && "'u' can't be used with poly types!"); 8590b57cec5SDimitry Andric Float = false; 8600b57cec5SDimitry Andric break; 8610b57cec5SDimitry Andric case 'o': 8620b57cec5SDimitry Andric Bitwidth = ElementBitwidth = 64; 8630b57cec5SDimitry Andric NumVectors = 0; 8640b57cec5SDimitry Andric Float = true; 8650b57cec5SDimitry Andric break; 8660b57cec5SDimitry Andric case 'y': 8670b57cec5SDimitry Andric Bitwidth = ElementBitwidth = 32; 8680b57cec5SDimitry Andric NumVectors = 0; 8690b57cec5SDimitry Andric Float = true; 8700b57cec5SDimitry Andric break; 8710b57cec5SDimitry Andric case 'Y': 8720b57cec5SDimitry Andric Bitwidth = ElementBitwidth = 16; 8730b57cec5SDimitry Andric NumVectors = 0; 8740b57cec5SDimitry Andric Float = true; 8750b57cec5SDimitry Andric break; 8760b57cec5SDimitry Andric case 'I': 8770b57cec5SDimitry Andric Bitwidth = ElementBitwidth = 32; 8780b57cec5SDimitry Andric NumVectors = 0; 8790b57cec5SDimitry Andric Float = false; 8800b57cec5SDimitry Andric Signed = true; 8810b57cec5SDimitry Andric break; 8820b57cec5SDimitry Andric case 'L': 8830b57cec5SDimitry Andric Bitwidth = ElementBitwidth = 64; 8840b57cec5SDimitry Andric NumVectors = 0; 8850b57cec5SDimitry Andric Float = false; 8860b57cec5SDimitry Andric Signed = true; 8870b57cec5SDimitry Andric break; 8880b57cec5SDimitry Andric case 'U': 8890b57cec5SDimitry Andric Bitwidth = ElementBitwidth = 32; 8900b57cec5SDimitry Andric NumVectors = 0; 8910b57cec5SDimitry Andric Float = false; 8920b57cec5SDimitry Andric Signed = false; 8930b57cec5SDimitry Andric break; 8940b57cec5SDimitry Andric case 'O': 8950b57cec5SDimitry Andric Bitwidth = ElementBitwidth = 64; 8960b57cec5SDimitry Andric NumVectors = 0; 8970b57cec5SDimitry Andric Float = false; 8980b57cec5SDimitry Andric Signed = false; 8990b57cec5SDimitry Andric break; 9000b57cec5SDimitry Andric case 'f': 9010b57cec5SDimitry Andric Float = true; 9020b57cec5SDimitry Andric ElementBitwidth = 32; 9030b57cec5SDimitry Andric break; 9040b57cec5SDimitry Andric case 'F': 9050b57cec5SDimitry Andric Float = true; 9060b57cec5SDimitry Andric ElementBitwidth = 64; 9070b57cec5SDimitry Andric break; 9080b57cec5SDimitry Andric case 'H': 9090b57cec5SDimitry Andric Float = true; 9100b57cec5SDimitry Andric ElementBitwidth = 16; 9110b57cec5SDimitry Andric break; 9120b57cec5SDimitry Andric case '0': 9130b57cec5SDimitry Andric Float = true; 9140b57cec5SDimitry Andric if (AppliedQuad) 9150b57cec5SDimitry Andric Bitwidth /= 2; 9160b57cec5SDimitry Andric ElementBitwidth = 16; 9170b57cec5SDimitry Andric break; 9180b57cec5SDimitry Andric case '1': 9190b57cec5SDimitry Andric Float = true; 9200b57cec5SDimitry Andric if (!AppliedQuad) 9210b57cec5SDimitry Andric Bitwidth *= 2; 9220b57cec5SDimitry Andric ElementBitwidth = 16; 9230b57cec5SDimitry Andric break; 9240b57cec5SDimitry Andric case 'g': 9250b57cec5SDimitry Andric if (AppliedQuad) 9260b57cec5SDimitry Andric Bitwidth /= 2; 9270b57cec5SDimitry Andric break; 9280b57cec5SDimitry Andric case 'j': 9290b57cec5SDimitry Andric if (!AppliedQuad) 9300b57cec5SDimitry Andric Bitwidth *= 2; 9310b57cec5SDimitry Andric break; 9320b57cec5SDimitry Andric case 'w': 9330b57cec5SDimitry Andric ElementBitwidth *= 2; 9340b57cec5SDimitry Andric Bitwidth *= 2; 9350b57cec5SDimitry Andric break; 9360b57cec5SDimitry Andric case 'n': 9370b57cec5SDimitry Andric ElementBitwidth *= 2; 9380b57cec5SDimitry Andric break; 9390b57cec5SDimitry Andric case 'i': 9400b57cec5SDimitry Andric Float = false; 9410b57cec5SDimitry Andric Poly = false; 9420b57cec5SDimitry Andric ElementBitwidth = Bitwidth = 32; 9430b57cec5SDimitry Andric NumVectors = 0; 9440b57cec5SDimitry Andric Signed = true; 9450b57cec5SDimitry Andric Immediate = true; 9460b57cec5SDimitry Andric break; 9470b57cec5SDimitry Andric case 'l': 9480b57cec5SDimitry Andric Float = false; 9490b57cec5SDimitry Andric Poly = false; 9500b57cec5SDimitry Andric ElementBitwidth = Bitwidth = 64; 9510b57cec5SDimitry Andric NumVectors = 0; 9520b57cec5SDimitry Andric Signed = false; 9530b57cec5SDimitry Andric Immediate = true; 9540b57cec5SDimitry Andric break; 9550b57cec5SDimitry Andric case 'z': 9560b57cec5SDimitry Andric ElementBitwidth /= 2; 9570b57cec5SDimitry Andric Bitwidth = ElementBitwidth; 9580b57cec5SDimitry Andric NumVectors = 0; 9590b57cec5SDimitry Andric break; 9600b57cec5SDimitry Andric case 'r': 9610b57cec5SDimitry Andric ElementBitwidth *= 2; 9620b57cec5SDimitry Andric Bitwidth = ElementBitwidth; 9630b57cec5SDimitry Andric NumVectors = 0; 9640b57cec5SDimitry Andric break; 9650b57cec5SDimitry Andric case 's': 9660b57cec5SDimitry Andric case 'a': 9670b57cec5SDimitry Andric Bitwidth = ElementBitwidth; 9680b57cec5SDimitry Andric NumVectors = 0; 9690b57cec5SDimitry Andric break; 9700b57cec5SDimitry Andric case 'k': 9710b57cec5SDimitry Andric Bitwidth *= 2; 9720b57cec5SDimitry Andric break; 9730b57cec5SDimitry Andric case 'c': 9740b57cec5SDimitry Andric Constant = true; 9750b57cec5SDimitry Andric LLVM_FALLTHROUGH; 9760b57cec5SDimitry Andric case 'p': 9770b57cec5SDimitry Andric Pointer = true; 9780b57cec5SDimitry Andric Bitwidth = ElementBitwidth; 9790b57cec5SDimitry Andric NumVectors = 0; 9800b57cec5SDimitry Andric break; 9810b57cec5SDimitry Andric case 'h': 9820b57cec5SDimitry Andric ElementBitwidth /= 2; 9830b57cec5SDimitry Andric break; 9840b57cec5SDimitry Andric case 'q': 9850b57cec5SDimitry Andric ElementBitwidth /= 2; 9860b57cec5SDimitry Andric Bitwidth *= 2; 9870b57cec5SDimitry Andric break; 9880b57cec5SDimitry Andric case 'e': 9890b57cec5SDimitry Andric ElementBitwidth /= 2; 9900b57cec5SDimitry Andric Signed = false; 9910b57cec5SDimitry Andric break; 9920b57cec5SDimitry Andric case 'm': 9930b57cec5SDimitry Andric ElementBitwidth /= 2; 9940b57cec5SDimitry Andric Bitwidth /= 2; 9950b57cec5SDimitry Andric break; 9960b57cec5SDimitry Andric case 'd': 9970b57cec5SDimitry Andric break; 9980b57cec5SDimitry Andric case '2': 9990b57cec5SDimitry Andric NumVectors = 2; 10000b57cec5SDimitry Andric break; 10010b57cec5SDimitry Andric case '3': 10020b57cec5SDimitry Andric NumVectors = 3; 10030b57cec5SDimitry Andric break; 10040b57cec5SDimitry Andric case '4': 10050b57cec5SDimitry Andric NumVectors = 4; 10060b57cec5SDimitry Andric break; 10070b57cec5SDimitry Andric case 'B': 10080b57cec5SDimitry Andric NumVectors = 2; 10090b57cec5SDimitry Andric if (!AppliedQuad) 10100b57cec5SDimitry Andric Bitwidth *= 2; 10110b57cec5SDimitry Andric break; 10120b57cec5SDimitry Andric case 'C': 10130b57cec5SDimitry Andric NumVectors = 3; 10140b57cec5SDimitry Andric if (!AppliedQuad) 10150b57cec5SDimitry Andric Bitwidth *= 2; 10160b57cec5SDimitry Andric break; 10170b57cec5SDimitry Andric case 'D': 10180b57cec5SDimitry Andric NumVectors = 4; 10190b57cec5SDimitry Andric if (!AppliedQuad) 10200b57cec5SDimitry Andric Bitwidth *= 2; 10210b57cec5SDimitry Andric break; 10220b57cec5SDimitry Andric case '7': 10230b57cec5SDimitry Andric if (AppliedQuad) 10240b57cec5SDimitry Andric Bitwidth /= 2; 10250b57cec5SDimitry Andric ElementBitwidth = 8; 10260b57cec5SDimitry Andric break; 10270b57cec5SDimitry Andric case '8': 10280b57cec5SDimitry Andric ElementBitwidth = 8; 10290b57cec5SDimitry Andric break; 10300b57cec5SDimitry Andric case '9': 10310b57cec5SDimitry Andric if (!AppliedQuad) 10320b57cec5SDimitry Andric Bitwidth *= 2; 10330b57cec5SDimitry Andric ElementBitwidth = 8; 10340b57cec5SDimitry Andric break; 10350b57cec5SDimitry Andric default: 10360b57cec5SDimitry Andric llvm_unreachable("Unhandled character!"); 10370b57cec5SDimitry Andric } 10380b57cec5SDimitry Andric } 10390b57cec5SDimitry Andric 10400b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 10410b57cec5SDimitry Andric // Intrinsic implementation 10420b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 10430b57cec5SDimitry Andric 10440b57cec5SDimitry Andric std::string Intrinsic::getInstTypeCode(Type T, ClassKind CK) const { 10450b57cec5SDimitry Andric char typeCode = '\0'; 10460b57cec5SDimitry Andric bool printNumber = true; 10470b57cec5SDimitry Andric 10480b57cec5SDimitry Andric if (CK == ClassB) 10490b57cec5SDimitry Andric return ""; 10500b57cec5SDimitry Andric 10510b57cec5SDimitry Andric if (T.isPoly()) 10520b57cec5SDimitry Andric typeCode = 'p'; 10530b57cec5SDimitry Andric else if (T.isInteger()) 10540b57cec5SDimitry Andric typeCode = T.isSigned() ? 's' : 'u'; 10550b57cec5SDimitry Andric else 10560b57cec5SDimitry Andric typeCode = 'f'; 10570b57cec5SDimitry Andric 10580b57cec5SDimitry Andric if (CK == ClassI) { 10590b57cec5SDimitry Andric switch (typeCode) { 10600b57cec5SDimitry Andric default: 10610b57cec5SDimitry Andric break; 10620b57cec5SDimitry Andric case 's': 10630b57cec5SDimitry Andric case 'u': 10640b57cec5SDimitry Andric case 'p': 10650b57cec5SDimitry Andric typeCode = 'i'; 10660b57cec5SDimitry Andric break; 10670b57cec5SDimitry Andric } 10680b57cec5SDimitry Andric } 10690b57cec5SDimitry Andric if (CK == ClassB) { 10700b57cec5SDimitry Andric typeCode = '\0'; 10710b57cec5SDimitry Andric } 10720b57cec5SDimitry Andric 10730b57cec5SDimitry Andric std::string S; 10740b57cec5SDimitry Andric if (typeCode != '\0') 10750b57cec5SDimitry Andric S.push_back(typeCode); 10760b57cec5SDimitry Andric if (printNumber) 10770b57cec5SDimitry Andric S += utostr(T.getElementSizeInBits()); 10780b57cec5SDimitry Andric 10790b57cec5SDimitry Andric return S; 10800b57cec5SDimitry Andric } 10810b57cec5SDimitry Andric 10820b57cec5SDimitry Andric static bool isFloatingPointProtoModifier(char Mod) { 10830b57cec5SDimitry Andric return Mod == 'F' || Mod == 'f' || Mod == 'H' || Mod == 'Y' || Mod == 'I'; 10840b57cec5SDimitry Andric } 10850b57cec5SDimitry Andric 10860b57cec5SDimitry Andric std::string Intrinsic::getBuiltinTypeStr() { 10870b57cec5SDimitry Andric ClassKind LocalCK = getClassKind(true); 10880b57cec5SDimitry Andric std::string S; 10890b57cec5SDimitry Andric 10900b57cec5SDimitry Andric Type RetT = getReturnType(); 10910b57cec5SDimitry Andric if ((LocalCK == ClassI || LocalCK == ClassW) && RetT.isScalar() && 10920b57cec5SDimitry Andric !RetT.isFloating()) 10930b57cec5SDimitry Andric RetT.makeInteger(RetT.getElementSizeInBits(), false); 10940b57cec5SDimitry Andric 10950b57cec5SDimitry Andric // Since the return value must be one type, return a vector type of the 10960b57cec5SDimitry Andric // appropriate width which we will bitcast. An exception is made for 10970b57cec5SDimitry Andric // returning structs of 2, 3, or 4 vectors which are returned in a sret-like 10980b57cec5SDimitry Andric // fashion, storing them to a pointer arg. 10990b57cec5SDimitry Andric if (RetT.getNumVectors() > 1) { 11000b57cec5SDimitry Andric S += "vv*"; // void result with void* first argument 11010b57cec5SDimitry Andric } else { 11020b57cec5SDimitry Andric if (RetT.isPoly()) 11030b57cec5SDimitry Andric RetT.makeInteger(RetT.getElementSizeInBits(), false); 11040b57cec5SDimitry Andric if (!RetT.isScalar() && !RetT.isSigned()) 11050b57cec5SDimitry Andric RetT.makeSigned(); 11060b57cec5SDimitry Andric 11070b57cec5SDimitry Andric bool ForcedVectorFloatingType = isFloatingPointProtoModifier(Proto[0]); 11080b57cec5SDimitry Andric if (LocalCK == ClassB && !RetT.isScalar() && !ForcedVectorFloatingType) 11090b57cec5SDimitry Andric // Cast to vector of 8-bit elements. 11100b57cec5SDimitry Andric RetT.makeInteger(8, true); 11110b57cec5SDimitry Andric 11120b57cec5SDimitry Andric S += RetT.builtin_str(); 11130b57cec5SDimitry Andric } 11140b57cec5SDimitry Andric 11150b57cec5SDimitry Andric for (unsigned I = 0; I < getNumParams(); ++I) { 11160b57cec5SDimitry Andric Type T = getParamType(I); 11170b57cec5SDimitry Andric if (T.isPoly()) 11180b57cec5SDimitry Andric T.makeInteger(T.getElementSizeInBits(), false); 11190b57cec5SDimitry Andric 11200b57cec5SDimitry Andric bool ForcedFloatingType = isFloatingPointProtoModifier(Proto[I + 1]); 11210b57cec5SDimitry Andric if (LocalCK == ClassB && !T.isScalar() && !ForcedFloatingType) 11220b57cec5SDimitry Andric T.makeInteger(8, true); 11230b57cec5SDimitry Andric // Halves always get converted to 8-bit elements. 11240b57cec5SDimitry Andric if (T.isHalf() && T.isVector() && !T.isScalarForMangling()) 11250b57cec5SDimitry Andric T.makeInteger(8, true); 11260b57cec5SDimitry Andric 11270b57cec5SDimitry Andric if (LocalCK == ClassI) 11280b57cec5SDimitry Andric T.makeSigned(); 11290b57cec5SDimitry Andric 11300b57cec5SDimitry Andric if (hasImmediate() && getImmediateIdx() == I) 11310b57cec5SDimitry Andric T.makeImmediate(32); 11320b57cec5SDimitry Andric 11330b57cec5SDimitry Andric S += T.builtin_str(); 11340b57cec5SDimitry Andric } 11350b57cec5SDimitry Andric 11360b57cec5SDimitry Andric // Extra constant integer to hold type class enum for this function, e.g. s8 11370b57cec5SDimitry Andric if (LocalCK == ClassB) 11380b57cec5SDimitry Andric S += "i"; 11390b57cec5SDimitry Andric 11400b57cec5SDimitry Andric return S; 11410b57cec5SDimitry Andric } 11420b57cec5SDimitry Andric 11430b57cec5SDimitry Andric std::string Intrinsic::getMangledName(bool ForceClassS) const { 11440b57cec5SDimitry Andric // Check if the prototype has a scalar operand with the type of the vector 11450b57cec5SDimitry Andric // elements. If not, bitcasting the args will take care of arg checking. 11460b57cec5SDimitry Andric // The actual signedness etc. will be taken care of with special enums. 11470b57cec5SDimitry Andric ClassKind LocalCK = CK; 11480b57cec5SDimitry Andric if (!protoHasScalar()) 11490b57cec5SDimitry Andric LocalCK = ClassB; 11500b57cec5SDimitry Andric 11510b57cec5SDimitry Andric return mangleName(Name, ForceClassS ? ClassS : LocalCK); 11520b57cec5SDimitry Andric } 11530b57cec5SDimitry Andric 11540b57cec5SDimitry Andric std::string Intrinsic::mangleName(std::string Name, ClassKind LocalCK) const { 11550b57cec5SDimitry Andric std::string typeCode = getInstTypeCode(BaseType, LocalCK); 11560b57cec5SDimitry Andric std::string S = Name; 11570b57cec5SDimitry Andric 11580b57cec5SDimitry Andric if (Name == "vcvt_f16_f32" || Name == "vcvt_f32_f16" || 11590b57cec5SDimitry Andric Name == "vcvt_f32_f64" || Name == "vcvt_f64_f32") 11600b57cec5SDimitry Andric return Name; 11610b57cec5SDimitry Andric 11620b57cec5SDimitry Andric if (!typeCode.empty()) { 11630b57cec5SDimitry Andric // If the name ends with _xN (N = 2,3,4), insert the typeCode before _xN. 11640b57cec5SDimitry Andric if (Name.size() >= 3 && isdigit(Name.back()) && 11650b57cec5SDimitry Andric Name[Name.length() - 2] == 'x' && Name[Name.length() - 3] == '_') 11660b57cec5SDimitry Andric S.insert(S.length() - 3, "_" + typeCode); 11670b57cec5SDimitry Andric else 11680b57cec5SDimitry Andric S += "_" + typeCode; 11690b57cec5SDimitry Andric } 11700b57cec5SDimitry Andric 11710b57cec5SDimitry Andric if (BaseType != InBaseType) { 11720b57cec5SDimitry Andric // A reinterpret - out the input base type at the end. 11730b57cec5SDimitry Andric S += "_" + getInstTypeCode(InBaseType, LocalCK); 11740b57cec5SDimitry Andric } 11750b57cec5SDimitry Andric 11760b57cec5SDimitry Andric if (LocalCK == ClassB) 11770b57cec5SDimitry Andric S += "_v"; 11780b57cec5SDimitry Andric 11790b57cec5SDimitry Andric // Insert a 'q' before the first '_' character so that it ends up before 11800b57cec5SDimitry Andric // _lane or _n on vector-scalar operations. 11810b57cec5SDimitry Andric if (BaseType.getSizeInBits() == 128 && !BaseType.noManglingQ()) { 11820b57cec5SDimitry Andric size_t Pos = S.find('_'); 11830b57cec5SDimitry Andric S.insert(Pos, "q"); 11840b57cec5SDimitry Andric } 11850b57cec5SDimitry Andric 11860b57cec5SDimitry Andric char Suffix = '\0'; 11870b57cec5SDimitry Andric if (BaseType.isScalarForMangling()) { 11880b57cec5SDimitry Andric switch (BaseType.getElementSizeInBits()) { 11890b57cec5SDimitry Andric case 8: Suffix = 'b'; break; 11900b57cec5SDimitry Andric case 16: Suffix = 'h'; break; 11910b57cec5SDimitry Andric case 32: Suffix = 's'; break; 11920b57cec5SDimitry Andric case 64: Suffix = 'd'; break; 11930b57cec5SDimitry Andric default: llvm_unreachable("Bad suffix!"); 11940b57cec5SDimitry Andric } 11950b57cec5SDimitry Andric } 11960b57cec5SDimitry Andric if (Suffix != '\0') { 11970b57cec5SDimitry Andric size_t Pos = S.find('_'); 11980b57cec5SDimitry Andric S.insert(Pos, &Suffix, 1); 11990b57cec5SDimitry Andric } 12000b57cec5SDimitry Andric 12010b57cec5SDimitry Andric return S; 12020b57cec5SDimitry Andric } 12030b57cec5SDimitry Andric 12040b57cec5SDimitry Andric std::string Intrinsic::replaceParamsIn(std::string S) { 12050b57cec5SDimitry Andric while (S.find('$') != std::string::npos) { 12060b57cec5SDimitry Andric size_t Pos = S.find('$'); 12070b57cec5SDimitry Andric size_t End = Pos + 1; 12080b57cec5SDimitry Andric while (isalpha(S[End])) 12090b57cec5SDimitry Andric ++End; 12100b57cec5SDimitry Andric 12110b57cec5SDimitry Andric std::string VarName = S.substr(Pos + 1, End - Pos - 1); 12120b57cec5SDimitry Andric assert_with_loc(Variables.find(VarName) != Variables.end(), 12130b57cec5SDimitry Andric "Variable not defined!"); 12140b57cec5SDimitry Andric S.replace(Pos, End - Pos, Variables.find(VarName)->second.getName()); 12150b57cec5SDimitry Andric } 12160b57cec5SDimitry Andric 12170b57cec5SDimitry Andric return S; 12180b57cec5SDimitry Andric } 12190b57cec5SDimitry Andric 12200b57cec5SDimitry Andric void Intrinsic::initVariables() { 12210b57cec5SDimitry Andric Variables.clear(); 12220b57cec5SDimitry Andric 12230b57cec5SDimitry Andric // Modify the TypeSpec per-argument to get a concrete Type, and create 12240b57cec5SDimitry Andric // known variables for each. 12250b57cec5SDimitry Andric for (unsigned I = 1; I < Proto.size(); ++I) { 12260b57cec5SDimitry Andric char NameC = '0' + (I - 1); 12270b57cec5SDimitry Andric std::string Name = "p"; 12280b57cec5SDimitry Andric Name.push_back(NameC); 12290b57cec5SDimitry Andric 12300b57cec5SDimitry Andric Variables[Name] = Variable(Types[I], Name + VariablePostfix); 12310b57cec5SDimitry Andric } 12320b57cec5SDimitry Andric RetVar = Variable(Types[0], "ret" + VariablePostfix); 12330b57cec5SDimitry Andric } 12340b57cec5SDimitry Andric 12350b57cec5SDimitry Andric void Intrinsic::emitPrototype(StringRef NamePrefix) { 12360b57cec5SDimitry Andric if (UseMacro) 12370b57cec5SDimitry Andric OS << "#define "; 12380b57cec5SDimitry Andric else 12390b57cec5SDimitry Andric OS << "__ai " << Types[0].str() << " "; 12400b57cec5SDimitry Andric 12410b57cec5SDimitry Andric OS << NamePrefix.str() << mangleName(Name, ClassS) << "("; 12420b57cec5SDimitry Andric 12430b57cec5SDimitry Andric for (unsigned I = 0; I < getNumParams(); ++I) { 12440b57cec5SDimitry Andric if (I != 0) 12450b57cec5SDimitry Andric OS << ", "; 12460b57cec5SDimitry Andric 12470b57cec5SDimitry Andric char NameC = '0' + I; 12480b57cec5SDimitry Andric std::string Name = "p"; 12490b57cec5SDimitry Andric Name.push_back(NameC); 12500b57cec5SDimitry Andric assert(Variables.find(Name) != Variables.end()); 12510b57cec5SDimitry Andric Variable &V = Variables[Name]; 12520b57cec5SDimitry Andric 12530b57cec5SDimitry Andric if (!UseMacro) 12540b57cec5SDimitry Andric OS << V.getType().str() << " "; 12550b57cec5SDimitry Andric OS << V.getName(); 12560b57cec5SDimitry Andric } 12570b57cec5SDimitry Andric 12580b57cec5SDimitry Andric OS << ")"; 12590b57cec5SDimitry Andric } 12600b57cec5SDimitry Andric 12610b57cec5SDimitry Andric void Intrinsic::emitOpeningBrace() { 12620b57cec5SDimitry Andric if (UseMacro) 12630b57cec5SDimitry Andric OS << " __extension__ ({"; 12640b57cec5SDimitry Andric else 12650b57cec5SDimitry Andric OS << " {"; 12660b57cec5SDimitry Andric emitNewLine(); 12670b57cec5SDimitry Andric } 12680b57cec5SDimitry Andric 12690b57cec5SDimitry Andric void Intrinsic::emitClosingBrace() { 12700b57cec5SDimitry Andric if (UseMacro) 12710b57cec5SDimitry Andric OS << "})"; 12720b57cec5SDimitry Andric else 12730b57cec5SDimitry Andric OS << "}"; 12740b57cec5SDimitry Andric } 12750b57cec5SDimitry Andric 12760b57cec5SDimitry Andric void Intrinsic::emitNewLine() { 12770b57cec5SDimitry Andric if (UseMacro) 12780b57cec5SDimitry Andric OS << " \\\n"; 12790b57cec5SDimitry Andric else 12800b57cec5SDimitry Andric OS << "\n"; 12810b57cec5SDimitry Andric } 12820b57cec5SDimitry Andric 12830b57cec5SDimitry Andric void Intrinsic::emitReverseVariable(Variable &Dest, Variable &Src) { 12840b57cec5SDimitry Andric if (Dest.getType().getNumVectors() > 1) { 12850b57cec5SDimitry Andric emitNewLine(); 12860b57cec5SDimitry Andric 12870b57cec5SDimitry Andric for (unsigned K = 0; K < Dest.getType().getNumVectors(); ++K) { 12880b57cec5SDimitry Andric OS << " " << Dest.getName() << ".val[" << K << "] = " 12890b57cec5SDimitry Andric << "__builtin_shufflevector(" 12900b57cec5SDimitry Andric << Src.getName() << ".val[" << K << "], " 12910b57cec5SDimitry Andric << Src.getName() << ".val[" << K << "]"; 12920b57cec5SDimitry Andric for (int J = Dest.getType().getNumElements() - 1; J >= 0; --J) 12930b57cec5SDimitry Andric OS << ", " << J; 12940b57cec5SDimitry Andric OS << ");"; 12950b57cec5SDimitry Andric emitNewLine(); 12960b57cec5SDimitry Andric } 12970b57cec5SDimitry Andric } else { 12980b57cec5SDimitry Andric OS << " " << Dest.getName() 12990b57cec5SDimitry Andric << " = __builtin_shufflevector(" << Src.getName() << ", " << Src.getName(); 13000b57cec5SDimitry Andric for (int J = Dest.getType().getNumElements() - 1; J >= 0; --J) 13010b57cec5SDimitry Andric OS << ", " << J; 13020b57cec5SDimitry Andric OS << ");"; 13030b57cec5SDimitry Andric emitNewLine(); 13040b57cec5SDimitry Andric } 13050b57cec5SDimitry Andric } 13060b57cec5SDimitry Andric 13070b57cec5SDimitry Andric void Intrinsic::emitArgumentReversal() { 1308*a7dea167SDimitry Andric if (isBigEndianSafe()) 13090b57cec5SDimitry Andric return; 13100b57cec5SDimitry Andric 13110b57cec5SDimitry Andric // Reverse all vector arguments. 13120b57cec5SDimitry Andric for (unsigned I = 0; I < getNumParams(); ++I) { 13130b57cec5SDimitry Andric std::string Name = "p" + utostr(I); 13140b57cec5SDimitry Andric std::string NewName = "rev" + utostr(I); 13150b57cec5SDimitry Andric 13160b57cec5SDimitry Andric Variable &V = Variables[Name]; 13170b57cec5SDimitry Andric Variable NewV(V.getType(), NewName + VariablePostfix); 13180b57cec5SDimitry Andric 13190b57cec5SDimitry Andric if (!NewV.getType().isVector() || NewV.getType().getNumElements() == 1) 13200b57cec5SDimitry Andric continue; 13210b57cec5SDimitry Andric 13220b57cec5SDimitry Andric OS << " " << NewV.getType().str() << " " << NewV.getName() << ";"; 13230b57cec5SDimitry Andric emitReverseVariable(NewV, V); 13240b57cec5SDimitry Andric V = NewV; 13250b57cec5SDimitry Andric } 13260b57cec5SDimitry Andric } 13270b57cec5SDimitry Andric 13280b57cec5SDimitry Andric void Intrinsic::emitReturnReversal() { 1329*a7dea167SDimitry Andric if (isBigEndianSafe()) 13300b57cec5SDimitry Andric return; 13310b57cec5SDimitry Andric if (!getReturnType().isVector() || getReturnType().isVoid() || 13320b57cec5SDimitry Andric getReturnType().getNumElements() == 1) 13330b57cec5SDimitry Andric return; 13340b57cec5SDimitry Andric emitReverseVariable(RetVar, RetVar); 13350b57cec5SDimitry Andric } 13360b57cec5SDimitry Andric 13370b57cec5SDimitry Andric void Intrinsic::emitShadowedArgs() { 13380b57cec5SDimitry Andric // Macro arguments are not type-checked like inline function arguments, 13390b57cec5SDimitry Andric // so assign them to local temporaries to get the right type checking. 13400b57cec5SDimitry Andric if (!UseMacro) 13410b57cec5SDimitry Andric return; 13420b57cec5SDimitry Andric 13430b57cec5SDimitry Andric for (unsigned I = 0; I < getNumParams(); ++I) { 13440b57cec5SDimitry Andric // Do not create a temporary for an immediate argument. 13450b57cec5SDimitry Andric // That would defeat the whole point of using a macro! 13460b57cec5SDimitry Andric if (hasImmediate() && Proto[I+1] == 'i') 13470b57cec5SDimitry Andric continue; 13480b57cec5SDimitry Andric // Do not create a temporary for pointer arguments. The input 13490b57cec5SDimitry Andric // pointer may have an alignment hint. 13500b57cec5SDimitry Andric if (getParamType(I).isPointer()) 13510b57cec5SDimitry Andric continue; 13520b57cec5SDimitry Andric 13530b57cec5SDimitry Andric std::string Name = "p" + utostr(I); 13540b57cec5SDimitry Andric 13550b57cec5SDimitry Andric assert(Variables.find(Name) != Variables.end()); 13560b57cec5SDimitry Andric Variable &V = Variables[Name]; 13570b57cec5SDimitry Andric 13580b57cec5SDimitry Andric std::string NewName = "s" + utostr(I); 13590b57cec5SDimitry Andric Variable V2(V.getType(), NewName + VariablePostfix); 13600b57cec5SDimitry Andric 13610b57cec5SDimitry Andric OS << " " << V2.getType().str() << " " << V2.getName() << " = " 13620b57cec5SDimitry Andric << V.getName() << ";"; 13630b57cec5SDimitry Andric emitNewLine(); 13640b57cec5SDimitry Andric 13650b57cec5SDimitry Andric V = V2; 13660b57cec5SDimitry Andric } 13670b57cec5SDimitry Andric } 13680b57cec5SDimitry Andric 13690b57cec5SDimitry Andric // We don't check 'a' in this function, because for builtin function the 13700b57cec5SDimitry Andric // argument matching to 'a' uses a vector type splatted from a scalar type. 13710b57cec5SDimitry Andric bool Intrinsic::protoHasScalar() const { 13720b57cec5SDimitry Andric return (Proto.find('s') != std::string::npos || 13730b57cec5SDimitry Andric Proto.find('z') != std::string::npos || 13740b57cec5SDimitry Andric Proto.find('r') != std::string::npos || 13750b57cec5SDimitry Andric Proto.find('b') != std::string::npos || 13760b57cec5SDimitry Andric Proto.find('$') != std::string::npos || 13770b57cec5SDimitry Andric Proto.find('y') != std::string::npos || 13780b57cec5SDimitry Andric Proto.find('o') != std::string::npos); 13790b57cec5SDimitry Andric } 13800b57cec5SDimitry Andric 13810b57cec5SDimitry Andric void Intrinsic::emitBodyAsBuiltinCall() { 13820b57cec5SDimitry Andric std::string S; 13830b57cec5SDimitry Andric 13840b57cec5SDimitry Andric // If this builtin returns a struct 2, 3, or 4 vectors, pass it as an implicit 13850b57cec5SDimitry Andric // sret-like argument. 13860b57cec5SDimitry Andric bool SRet = getReturnType().getNumVectors() >= 2; 13870b57cec5SDimitry Andric 13880b57cec5SDimitry Andric StringRef N = Name; 13890b57cec5SDimitry Andric if (hasSplat()) { 13900b57cec5SDimitry Andric // Call the non-splat builtin: chop off the "_n" suffix from the name. 13910b57cec5SDimitry Andric assert(N.endswith("_n")); 13920b57cec5SDimitry Andric N = N.drop_back(2); 13930b57cec5SDimitry Andric } 13940b57cec5SDimitry Andric 13950b57cec5SDimitry Andric ClassKind LocalCK = CK; 13960b57cec5SDimitry Andric if (!protoHasScalar()) 13970b57cec5SDimitry Andric LocalCK = ClassB; 13980b57cec5SDimitry Andric 13990b57cec5SDimitry Andric if (!getReturnType().isVoid() && !SRet) 14000b57cec5SDimitry Andric S += "(" + RetVar.getType().str() + ") "; 14010b57cec5SDimitry Andric 14020b57cec5SDimitry Andric S += "__builtin_neon_" + mangleName(N, LocalCK) + "("; 14030b57cec5SDimitry Andric 14040b57cec5SDimitry Andric if (SRet) 14050b57cec5SDimitry Andric S += "&" + RetVar.getName() + ", "; 14060b57cec5SDimitry Andric 14070b57cec5SDimitry Andric for (unsigned I = 0; I < getNumParams(); ++I) { 14080b57cec5SDimitry Andric Variable &V = Variables["p" + utostr(I)]; 14090b57cec5SDimitry Andric Type T = V.getType(); 14100b57cec5SDimitry Andric 14110b57cec5SDimitry Andric // Handle multiple-vector values specially, emitting each subvector as an 14120b57cec5SDimitry Andric // argument to the builtin. 14130b57cec5SDimitry Andric if (T.getNumVectors() > 1) { 14140b57cec5SDimitry Andric // Check if an explicit cast is needed. 14150b57cec5SDimitry Andric std::string Cast; 1416*a7dea167SDimitry Andric if (LocalCK == ClassB) { 14170b57cec5SDimitry Andric Type T2 = T; 14180b57cec5SDimitry Andric T2.makeOneVector(); 14190b57cec5SDimitry Andric T2.makeInteger(8, /*Signed=*/true); 14200b57cec5SDimitry Andric Cast = "(" + T2.str() + ")"; 14210b57cec5SDimitry Andric } 14220b57cec5SDimitry Andric 14230b57cec5SDimitry Andric for (unsigned J = 0; J < T.getNumVectors(); ++J) 14240b57cec5SDimitry Andric S += Cast + V.getName() + ".val[" + utostr(J) + "], "; 14250b57cec5SDimitry Andric continue; 14260b57cec5SDimitry Andric } 14270b57cec5SDimitry Andric 14280b57cec5SDimitry Andric std::string Arg; 14290b57cec5SDimitry Andric Type CastToType = T; 14300b57cec5SDimitry Andric if (hasSplat() && I == getSplatIdx()) { 14310b57cec5SDimitry Andric Arg = "(" + BaseType.str() + ") {"; 14320b57cec5SDimitry Andric for (unsigned J = 0; J < BaseType.getNumElements(); ++J) { 14330b57cec5SDimitry Andric if (J != 0) 14340b57cec5SDimitry Andric Arg += ", "; 14350b57cec5SDimitry Andric Arg += V.getName(); 14360b57cec5SDimitry Andric } 14370b57cec5SDimitry Andric Arg += "}"; 14380b57cec5SDimitry Andric 14390b57cec5SDimitry Andric CastToType = BaseType; 14400b57cec5SDimitry Andric } else { 14410b57cec5SDimitry Andric Arg = V.getName(); 14420b57cec5SDimitry Andric } 14430b57cec5SDimitry Andric 14440b57cec5SDimitry Andric // Check if an explicit cast is needed. 1445*a7dea167SDimitry Andric if (CastToType.isVector() && 1446*a7dea167SDimitry Andric (LocalCK == ClassB || (T.isHalf() && !T.isScalarForMangling()))) { 14470b57cec5SDimitry Andric CastToType.makeInteger(8, true); 14480b57cec5SDimitry Andric Arg = "(" + CastToType.str() + ")" + Arg; 1449*a7dea167SDimitry Andric } else if (CastToType.isVector() && LocalCK == ClassI) { 1450*a7dea167SDimitry Andric CastToType.makeSigned(); 1451*a7dea167SDimitry Andric Arg = "(" + CastToType.str() + ")" + Arg; 14520b57cec5SDimitry Andric } 14530b57cec5SDimitry Andric 14540b57cec5SDimitry Andric S += Arg + ", "; 14550b57cec5SDimitry Andric } 14560b57cec5SDimitry Andric 14570b57cec5SDimitry Andric // Extra constant integer to hold type class enum for this function, e.g. s8 14580b57cec5SDimitry Andric if (getClassKind(true) == ClassB) { 14590b57cec5SDimitry Andric Type ThisTy = getReturnType(); 14600b57cec5SDimitry Andric if (Proto[0] == 'v' || isFloatingPointProtoModifier(Proto[0])) 14610b57cec5SDimitry Andric ThisTy = getParamType(0); 14620b57cec5SDimitry Andric if (ThisTy.isPointer()) 14630b57cec5SDimitry Andric ThisTy = getParamType(1); 14640b57cec5SDimitry Andric 14650b57cec5SDimitry Andric S += utostr(ThisTy.getNeonEnum()); 14660b57cec5SDimitry Andric } else { 14670b57cec5SDimitry Andric // Remove extraneous ", ". 14680b57cec5SDimitry Andric S.pop_back(); 14690b57cec5SDimitry Andric S.pop_back(); 14700b57cec5SDimitry Andric } 14710b57cec5SDimitry Andric S += ");"; 14720b57cec5SDimitry Andric 14730b57cec5SDimitry Andric std::string RetExpr; 14740b57cec5SDimitry Andric if (!SRet && !RetVar.getType().isVoid()) 14750b57cec5SDimitry Andric RetExpr = RetVar.getName() + " = "; 14760b57cec5SDimitry Andric 14770b57cec5SDimitry Andric OS << " " << RetExpr << S; 14780b57cec5SDimitry Andric emitNewLine(); 14790b57cec5SDimitry Andric } 14800b57cec5SDimitry Andric 14810b57cec5SDimitry Andric void Intrinsic::emitBody(StringRef CallPrefix) { 14820b57cec5SDimitry Andric std::vector<std::string> Lines; 14830b57cec5SDimitry Andric 14840b57cec5SDimitry Andric assert(RetVar.getType() == Types[0]); 14850b57cec5SDimitry Andric // Create a return variable, if we're not void. 14860b57cec5SDimitry Andric if (!RetVar.getType().isVoid()) { 14870b57cec5SDimitry Andric OS << " " << RetVar.getType().str() << " " << RetVar.getName() << ";"; 14880b57cec5SDimitry Andric emitNewLine(); 14890b57cec5SDimitry Andric } 14900b57cec5SDimitry Andric 14910b57cec5SDimitry Andric if (!Body || Body->getValues().empty()) { 14920b57cec5SDimitry Andric // Nothing specific to output - must output a builtin. 14930b57cec5SDimitry Andric emitBodyAsBuiltinCall(); 14940b57cec5SDimitry Andric return; 14950b57cec5SDimitry Andric } 14960b57cec5SDimitry Andric 14970b57cec5SDimitry Andric // We have a list of "things to output". The last should be returned. 14980b57cec5SDimitry Andric for (auto *I : Body->getValues()) { 14990b57cec5SDimitry Andric if (StringInit *SI = dyn_cast<StringInit>(I)) { 15000b57cec5SDimitry Andric Lines.push_back(replaceParamsIn(SI->getAsString())); 15010b57cec5SDimitry Andric } else if (DagInit *DI = dyn_cast<DagInit>(I)) { 15020b57cec5SDimitry Andric DagEmitter DE(*this, CallPrefix); 15030b57cec5SDimitry Andric Lines.push_back(DE.emitDag(DI).second + ";"); 15040b57cec5SDimitry Andric } 15050b57cec5SDimitry Andric } 15060b57cec5SDimitry Andric 15070b57cec5SDimitry Andric assert(!Lines.empty() && "Empty def?"); 15080b57cec5SDimitry Andric if (!RetVar.getType().isVoid()) 15090b57cec5SDimitry Andric Lines.back().insert(0, RetVar.getName() + " = "); 15100b57cec5SDimitry Andric 15110b57cec5SDimitry Andric for (auto &L : Lines) { 15120b57cec5SDimitry Andric OS << " " << L; 15130b57cec5SDimitry Andric emitNewLine(); 15140b57cec5SDimitry Andric } 15150b57cec5SDimitry Andric } 15160b57cec5SDimitry Andric 15170b57cec5SDimitry Andric void Intrinsic::emitReturn() { 15180b57cec5SDimitry Andric if (RetVar.getType().isVoid()) 15190b57cec5SDimitry Andric return; 15200b57cec5SDimitry Andric if (UseMacro) 15210b57cec5SDimitry Andric OS << " " << RetVar.getName() << ";"; 15220b57cec5SDimitry Andric else 15230b57cec5SDimitry Andric OS << " return " << RetVar.getName() << ";"; 15240b57cec5SDimitry Andric emitNewLine(); 15250b57cec5SDimitry Andric } 15260b57cec5SDimitry Andric 15270b57cec5SDimitry Andric std::pair<Type, std::string> Intrinsic::DagEmitter::emitDag(DagInit *DI) { 15280b57cec5SDimitry Andric // At this point we should only be seeing a def. 15290b57cec5SDimitry Andric DefInit *DefI = cast<DefInit>(DI->getOperator()); 15300b57cec5SDimitry Andric std::string Op = DefI->getAsString(); 15310b57cec5SDimitry Andric 15320b57cec5SDimitry Andric if (Op == "cast" || Op == "bitcast") 15330b57cec5SDimitry Andric return emitDagCast(DI, Op == "bitcast"); 15340b57cec5SDimitry Andric if (Op == "shuffle") 15350b57cec5SDimitry Andric return emitDagShuffle(DI); 15360b57cec5SDimitry Andric if (Op == "dup") 15370b57cec5SDimitry Andric return emitDagDup(DI); 15380b57cec5SDimitry Andric if (Op == "dup_typed") 15390b57cec5SDimitry Andric return emitDagDupTyped(DI); 15400b57cec5SDimitry Andric if (Op == "splat") 15410b57cec5SDimitry Andric return emitDagSplat(DI); 15420b57cec5SDimitry Andric if (Op == "save_temp") 15430b57cec5SDimitry Andric return emitDagSaveTemp(DI); 15440b57cec5SDimitry Andric if (Op == "op") 15450b57cec5SDimitry Andric return emitDagOp(DI); 15460b57cec5SDimitry Andric if (Op == "call") 15470b57cec5SDimitry Andric return emitDagCall(DI); 15480b57cec5SDimitry Andric if (Op == "name_replace") 15490b57cec5SDimitry Andric return emitDagNameReplace(DI); 15500b57cec5SDimitry Andric if (Op == "literal") 15510b57cec5SDimitry Andric return emitDagLiteral(DI); 15520b57cec5SDimitry Andric assert_with_loc(false, "Unknown operation!"); 15530b57cec5SDimitry Andric return std::make_pair(Type::getVoid(), ""); 15540b57cec5SDimitry Andric } 15550b57cec5SDimitry Andric 15560b57cec5SDimitry Andric std::pair<Type, std::string> Intrinsic::DagEmitter::emitDagOp(DagInit *DI) { 15570b57cec5SDimitry Andric std::string Op = cast<StringInit>(DI->getArg(0))->getAsUnquotedString(); 15580b57cec5SDimitry Andric if (DI->getNumArgs() == 2) { 15590b57cec5SDimitry Andric // Unary op. 15600b57cec5SDimitry Andric std::pair<Type, std::string> R = 15610b57cec5SDimitry Andric emitDagArg(DI->getArg(1), DI->getArgNameStr(1)); 15620b57cec5SDimitry Andric return std::make_pair(R.first, Op + R.second); 15630b57cec5SDimitry Andric } else { 15640b57cec5SDimitry Andric assert(DI->getNumArgs() == 3 && "Can only handle unary and binary ops!"); 15650b57cec5SDimitry Andric std::pair<Type, std::string> R1 = 15660b57cec5SDimitry Andric emitDagArg(DI->getArg(1), DI->getArgNameStr(1)); 15670b57cec5SDimitry Andric std::pair<Type, std::string> R2 = 15680b57cec5SDimitry Andric emitDagArg(DI->getArg(2), DI->getArgNameStr(2)); 15690b57cec5SDimitry Andric assert_with_loc(R1.first == R2.first, "Argument type mismatch!"); 15700b57cec5SDimitry Andric return std::make_pair(R1.first, R1.second + " " + Op + " " + R2.second); 15710b57cec5SDimitry Andric } 15720b57cec5SDimitry Andric } 15730b57cec5SDimitry Andric 15740b57cec5SDimitry Andric std::pair<Type, std::string> Intrinsic::DagEmitter::emitDagCall(DagInit *DI) { 15750b57cec5SDimitry Andric std::vector<Type> Types; 15760b57cec5SDimitry Andric std::vector<std::string> Values; 15770b57cec5SDimitry Andric for (unsigned I = 0; I < DI->getNumArgs() - 1; ++I) { 15780b57cec5SDimitry Andric std::pair<Type, std::string> R = 15790b57cec5SDimitry Andric emitDagArg(DI->getArg(I + 1), DI->getArgNameStr(I + 1)); 15800b57cec5SDimitry Andric Types.push_back(R.first); 15810b57cec5SDimitry Andric Values.push_back(R.second); 15820b57cec5SDimitry Andric } 15830b57cec5SDimitry Andric 15840b57cec5SDimitry Andric // Look up the called intrinsic. 15850b57cec5SDimitry Andric std::string N; 15860b57cec5SDimitry Andric if (StringInit *SI = dyn_cast<StringInit>(DI->getArg(0))) 15870b57cec5SDimitry Andric N = SI->getAsUnquotedString(); 15880b57cec5SDimitry Andric else 15890b57cec5SDimitry Andric N = emitDagArg(DI->getArg(0), "").second; 15900b57cec5SDimitry Andric Intrinsic &Callee = Intr.Emitter.getIntrinsic(N, Types); 15910b57cec5SDimitry Andric 15920b57cec5SDimitry Andric // Make sure the callee is known as an early def. 15930b57cec5SDimitry Andric Callee.setNeededEarly(); 15940b57cec5SDimitry Andric Intr.Dependencies.insert(&Callee); 15950b57cec5SDimitry Andric 15960b57cec5SDimitry Andric // Now create the call itself. 1597*a7dea167SDimitry Andric std::string S = ""; 1598*a7dea167SDimitry Andric if (!Callee.isBigEndianSafe()) 1599*a7dea167SDimitry Andric S += CallPrefix.str(); 1600*a7dea167SDimitry Andric S += Callee.getMangledName(true) + "("; 16010b57cec5SDimitry Andric for (unsigned I = 0; I < DI->getNumArgs() - 1; ++I) { 16020b57cec5SDimitry Andric if (I != 0) 16030b57cec5SDimitry Andric S += ", "; 16040b57cec5SDimitry Andric S += Values[I]; 16050b57cec5SDimitry Andric } 16060b57cec5SDimitry Andric S += ")"; 16070b57cec5SDimitry Andric 16080b57cec5SDimitry Andric return std::make_pair(Callee.getReturnType(), S); 16090b57cec5SDimitry Andric } 16100b57cec5SDimitry Andric 16110b57cec5SDimitry Andric std::pair<Type, std::string> Intrinsic::DagEmitter::emitDagCast(DagInit *DI, 16120b57cec5SDimitry Andric bool IsBitCast){ 16130b57cec5SDimitry Andric // (cast MOD* VAL) -> cast VAL to type given by MOD. 16140b57cec5SDimitry Andric std::pair<Type, std::string> R = emitDagArg( 16150b57cec5SDimitry Andric DI->getArg(DI->getNumArgs() - 1), 16160b57cec5SDimitry Andric DI->getArgNameStr(DI->getNumArgs() - 1)); 16170b57cec5SDimitry Andric Type castToType = R.first; 16180b57cec5SDimitry Andric for (unsigned ArgIdx = 0; ArgIdx < DI->getNumArgs() - 1; ++ArgIdx) { 16190b57cec5SDimitry Andric 16200b57cec5SDimitry Andric // MOD can take several forms: 16210b57cec5SDimitry Andric // 1. $X - take the type of parameter / variable X. 16220b57cec5SDimitry Andric // 2. The value "R" - take the type of the return type. 16230b57cec5SDimitry Andric // 3. a type string 16240b57cec5SDimitry Andric // 4. The value "U" or "S" to switch the signedness. 16250b57cec5SDimitry Andric // 5. The value "H" or "D" to half or double the bitwidth. 16260b57cec5SDimitry Andric // 6. The value "8" to convert to 8-bit (signed) integer lanes. 16270b57cec5SDimitry Andric if (!DI->getArgNameStr(ArgIdx).empty()) { 16280b57cec5SDimitry Andric assert_with_loc(Intr.Variables.find(DI->getArgNameStr(ArgIdx)) != 16290b57cec5SDimitry Andric Intr.Variables.end(), 16300b57cec5SDimitry Andric "Variable not found"); 16310b57cec5SDimitry Andric castToType = Intr.Variables[DI->getArgNameStr(ArgIdx)].getType(); 16320b57cec5SDimitry Andric } else { 16330b57cec5SDimitry Andric StringInit *SI = dyn_cast<StringInit>(DI->getArg(ArgIdx)); 16340b57cec5SDimitry Andric assert_with_loc(SI, "Expected string type or $Name for cast type"); 16350b57cec5SDimitry Andric 16360b57cec5SDimitry Andric if (SI->getAsUnquotedString() == "R") { 16370b57cec5SDimitry Andric castToType = Intr.getReturnType(); 16380b57cec5SDimitry Andric } else if (SI->getAsUnquotedString() == "U") { 16390b57cec5SDimitry Andric castToType.makeUnsigned(); 16400b57cec5SDimitry Andric } else if (SI->getAsUnquotedString() == "S") { 16410b57cec5SDimitry Andric castToType.makeSigned(); 16420b57cec5SDimitry Andric } else if (SI->getAsUnquotedString() == "H") { 16430b57cec5SDimitry Andric castToType.halveLanes(); 16440b57cec5SDimitry Andric } else if (SI->getAsUnquotedString() == "D") { 16450b57cec5SDimitry Andric castToType.doubleLanes(); 16460b57cec5SDimitry Andric } else if (SI->getAsUnquotedString() == "8") { 16470b57cec5SDimitry Andric castToType.makeInteger(8, true); 16480b57cec5SDimitry Andric } else { 16490b57cec5SDimitry Andric castToType = Type::fromTypedefName(SI->getAsUnquotedString()); 16500b57cec5SDimitry Andric assert_with_loc(!castToType.isVoid(), "Unknown typedef"); 16510b57cec5SDimitry Andric } 16520b57cec5SDimitry Andric } 16530b57cec5SDimitry Andric } 16540b57cec5SDimitry Andric 16550b57cec5SDimitry Andric std::string S; 16560b57cec5SDimitry Andric if (IsBitCast) { 16570b57cec5SDimitry Andric // Emit a reinterpret cast. The second operand must be an lvalue, so create 16580b57cec5SDimitry Andric // a temporary. 16590b57cec5SDimitry Andric std::string N = "reint"; 16600b57cec5SDimitry Andric unsigned I = 0; 16610b57cec5SDimitry Andric while (Intr.Variables.find(N) != Intr.Variables.end()) 16620b57cec5SDimitry Andric N = "reint" + utostr(++I); 16630b57cec5SDimitry Andric Intr.Variables[N] = Variable(R.first, N + Intr.VariablePostfix); 16640b57cec5SDimitry Andric 16650b57cec5SDimitry Andric Intr.OS << R.first.str() << " " << Intr.Variables[N].getName() << " = " 16660b57cec5SDimitry Andric << R.second << ";"; 16670b57cec5SDimitry Andric Intr.emitNewLine(); 16680b57cec5SDimitry Andric 16690b57cec5SDimitry Andric S = "*(" + castToType.str() + " *) &" + Intr.Variables[N].getName() + ""; 16700b57cec5SDimitry Andric } else { 16710b57cec5SDimitry Andric // Emit a normal (static) cast. 16720b57cec5SDimitry Andric S = "(" + castToType.str() + ")(" + R.second + ")"; 16730b57cec5SDimitry Andric } 16740b57cec5SDimitry Andric 16750b57cec5SDimitry Andric return std::make_pair(castToType, S); 16760b57cec5SDimitry Andric } 16770b57cec5SDimitry Andric 16780b57cec5SDimitry Andric std::pair<Type, std::string> Intrinsic::DagEmitter::emitDagShuffle(DagInit *DI){ 16790b57cec5SDimitry Andric // See the documentation in arm_neon.td for a description of these operators. 16800b57cec5SDimitry Andric class LowHalf : public SetTheory::Operator { 16810b57cec5SDimitry Andric public: 16820b57cec5SDimitry Andric void apply(SetTheory &ST, DagInit *Expr, SetTheory::RecSet &Elts, 16830b57cec5SDimitry Andric ArrayRef<SMLoc> Loc) override { 16840b57cec5SDimitry Andric SetTheory::RecSet Elts2; 16850b57cec5SDimitry Andric ST.evaluate(Expr->arg_begin(), Expr->arg_end(), Elts2, Loc); 16860b57cec5SDimitry Andric Elts.insert(Elts2.begin(), Elts2.begin() + (Elts2.size() / 2)); 16870b57cec5SDimitry Andric } 16880b57cec5SDimitry Andric }; 16890b57cec5SDimitry Andric 16900b57cec5SDimitry Andric class HighHalf : public SetTheory::Operator { 16910b57cec5SDimitry Andric public: 16920b57cec5SDimitry Andric void apply(SetTheory &ST, DagInit *Expr, SetTheory::RecSet &Elts, 16930b57cec5SDimitry Andric ArrayRef<SMLoc> Loc) override { 16940b57cec5SDimitry Andric SetTheory::RecSet Elts2; 16950b57cec5SDimitry Andric ST.evaluate(Expr->arg_begin(), Expr->arg_end(), Elts2, Loc); 16960b57cec5SDimitry Andric Elts.insert(Elts2.begin() + (Elts2.size() / 2), Elts2.end()); 16970b57cec5SDimitry Andric } 16980b57cec5SDimitry Andric }; 16990b57cec5SDimitry Andric 17000b57cec5SDimitry Andric class Rev : public SetTheory::Operator { 17010b57cec5SDimitry Andric unsigned ElementSize; 17020b57cec5SDimitry Andric 17030b57cec5SDimitry Andric public: 17040b57cec5SDimitry Andric Rev(unsigned ElementSize) : ElementSize(ElementSize) {} 17050b57cec5SDimitry Andric 17060b57cec5SDimitry Andric void apply(SetTheory &ST, DagInit *Expr, SetTheory::RecSet &Elts, 17070b57cec5SDimitry Andric ArrayRef<SMLoc> Loc) override { 17080b57cec5SDimitry Andric SetTheory::RecSet Elts2; 17090b57cec5SDimitry Andric ST.evaluate(Expr->arg_begin() + 1, Expr->arg_end(), Elts2, Loc); 17100b57cec5SDimitry Andric 17110b57cec5SDimitry Andric int64_t VectorSize = cast<IntInit>(Expr->getArg(0))->getValue(); 17120b57cec5SDimitry Andric VectorSize /= ElementSize; 17130b57cec5SDimitry Andric 17140b57cec5SDimitry Andric std::vector<Record *> Revved; 17150b57cec5SDimitry Andric for (unsigned VI = 0; VI < Elts2.size(); VI += VectorSize) { 17160b57cec5SDimitry Andric for (int LI = VectorSize - 1; LI >= 0; --LI) { 17170b57cec5SDimitry Andric Revved.push_back(Elts2[VI + LI]); 17180b57cec5SDimitry Andric } 17190b57cec5SDimitry Andric } 17200b57cec5SDimitry Andric 17210b57cec5SDimitry Andric Elts.insert(Revved.begin(), Revved.end()); 17220b57cec5SDimitry Andric } 17230b57cec5SDimitry Andric }; 17240b57cec5SDimitry Andric 17250b57cec5SDimitry Andric class MaskExpander : public SetTheory::Expander { 17260b57cec5SDimitry Andric unsigned N; 17270b57cec5SDimitry Andric 17280b57cec5SDimitry Andric public: 17290b57cec5SDimitry Andric MaskExpander(unsigned N) : N(N) {} 17300b57cec5SDimitry Andric 17310b57cec5SDimitry Andric void expand(SetTheory &ST, Record *R, SetTheory::RecSet &Elts) override { 17320b57cec5SDimitry Andric unsigned Addend = 0; 17330b57cec5SDimitry Andric if (R->getName() == "mask0") 17340b57cec5SDimitry Andric Addend = 0; 17350b57cec5SDimitry Andric else if (R->getName() == "mask1") 17360b57cec5SDimitry Andric Addend = N; 17370b57cec5SDimitry Andric else 17380b57cec5SDimitry Andric return; 17390b57cec5SDimitry Andric for (unsigned I = 0; I < N; ++I) 17400b57cec5SDimitry Andric Elts.insert(R->getRecords().getDef("sv" + utostr(I + Addend))); 17410b57cec5SDimitry Andric } 17420b57cec5SDimitry Andric }; 17430b57cec5SDimitry Andric 17440b57cec5SDimitry Andric // (shuffle arg1, arg2, sequence) 17450b57cec5SDimitry Andric std::pair<Type, std::string> Arg1 = 17460b57cec5SDimitry Andric emitDagArg(DI->getArg(0), DI->getArgNameStr(0)); 17470b57cec5SDimitry Andric std::pair<Type, std::string> Arg2 = 17480b57cec5SDimitry Andric emitDagArg(DI->getArg(1), DI->getArgNameStr(1)); 17490b57cec5SDimitry Andric assert_with_loc(Arg1.first == Arg2.first, 17500b57cec5SDimitry Andric "Different types in arguments to shuffle!"); 17510b57cec5SDimitry Andric 17520b57cec5SDimitry Andric SetTheory ST; 17530b57cec5SDimitry Andric SetTheory::RecSet Elts; 1754*a7dea167SDimitry Andric ST.addOperator("lowhalf", std::make_unique<LowHalf>()); 1755*a7dea167SDimitry Andric ST.addOperator("highhalf", std::make_unique<HighHalf>()); 17560b57cec5SDimitry Andric ST.addOperator("rev", 1757*a7dea167SDimitry Andric std::make_unique<Rev>(Arg1.first.getElementSizeInBits())); 17580b57cec5SDimitry Andric ST.addExpander("MaskExpand", 1759*a7dea167SDimitry Andric std::make_unique<MaskExpander>(Arg1.first.getNumElements())); 17600b57cec5SDimitry Andric ST.evaluate(DI->getArg(2), Elts, None); 17610b57cec5SDimitry Andric 17620b57cec5SDimitry Andric std::string S = "__builtin_shufflevector(" + Arg1.second + ", " + Arg2.second; 17630b57cec5SDimitry Andric for (auto &E : Elts) { 17640b57cec5SDimitry Andric StringRef Name = E->getName(); 17650b57cec5SDimitry Andric assert_with_loc(Name.startswith("sv"), 17660b57cec5SDimitry Andric "Incorrect element kind in shuffle mask!"); 17670b57cec5SDimitry Andric S += ", " + Name.drop_front(2).str(); 17680b57cec5SDimitry Andric } 17690b57cec5SDimitry Andric S += ")"; 17700b57cec5SDimitry Andric 17710b57cec5SDimitry Andric // Recalculate the return type - the shuffle may have halved or doubled it. 17720b57cec5SDimitry Andric Type T(Arg1.first); 17730b57cec5SDimitry Andric if (Elts.size() > T.getNumElements()) { 17740b57cec5SDimitry Andric assert_with_loc( 17750b57cec5SDimitry Andric Elts.size() == T.getNumElements() * 2, 17760b57cec5SDimitry Andric "Can only double or half the number of elements in a shuffle!"); 17770b57cec5SDimitry Andric T.doubleLanes(); 17780b57cec5SDimitry Andric } else if (Elts.size() < T.getNumElements()) { 17790b57cec5SDimitry Andric assert_with_loc( 17800b57cec5SDimitry Andric Elts.size() == T.getNumElements() / 2, 17810b57cec5SDimitry Andric "Can only double or half the number of elements in a shuffle!"); 17820b57cec5SDimitry Andric T.halveLanes(); 17830b57cec5SDimitry Andric } 17840b57cec5SDimitry Andric 17850b57cec5SDimitry Andric return std::make_pair(T, S); 17860b57cec5SDimitry Andric } 17870b57cec5SDimitry Andric 17880b57cec5SDimitry Andric std::pair<Type, std::string> Intrinsic::DagEmitter::emitDagDup(DagInit *DI) { 17890b57cec5SDimitry Andric assert_with_loc(DI->getNumArgs() == 1, "dup() expects one argument"); 17900b57cec5SDimitry Andric std::pair<Type, std::string> A = emitDagArg(DI->getArg(0), 17910b57cec5SDimitry Andric DI->getArgNameStr(0)); 17920b57cec5SDimitry Andric assert_with_loc(A.first.isScalar(), "dup() expects a scalar argument"); 17930b57cec5SDimitry Andric 17940b57cec5SDimitry Andric Type T = Intr.getBaseType(); 17950b57cec5SDimitry Andric assert_with_loc(T.isVector(), "dup() used but default type is scalar!"); 17960b57cec5SDimitry Andric std::string S = "(" + T.str() + ") {"; 17970b57cec5SDimitry Andric for (unsigned I = 0; I < T.getNumElements(); ++I) { 17980b57cec5SDimitry Andric if (I != 0) 17990b57cec5SDimitry Andric S += ", "; 18000b57cec5SDimitry Andric S += A.second; 18010b57cec5SDimitry Andric } 18020b57cec5SDimitry Andric S += "}"; 18030b57cec5SDimitry Andric 18040b57cec5SDimitry Andric return std::make_pair(T, S); 18050b57cec5SDimitry Andric } 18060b57cec5SDimitry Andric 18070b57cec5SDimitry Andric std::pair<Type, std::string> Intrinsic::DagEmitter::emitDagDupTyped(DagInit *DI) { 18080b57cec5SDimitry Andric assert_with_loc(DI->getNumArgs() == 2, "dup_typed() expects two arguments"); 18090b57cec5SDimitry Andric std::pair<Type, std::string> A = emitDagArg(DI->getArg(0), 18100b57cec5SDimitry Andric DI->getArgNameStr(0)); 18110b57cec5SDimitry Andric std::pair<Type, std::string> B = emitDagArg(DI->getArg(1), 18120b57cec5SDimitry Andric DI->getArgNameStr(1)); 18130b57cec5SDimitry Andric assert_with_loc(B.first.isScalar(), 18140b57cec5SDimitry Andric "dup_typed() requires a scalar as the second argument"); 18150b57cec5SDimitry Andric 18160b57cec5SDimitry Andric Type T = A.first; 18170b57cec5SDimitry Andric assert_with_loc(T.isVector(), "dup_typed() used but target type is scalar!"); 18180b57cec5SDimitry Andric std::string S = "(" + T.str() + ") {"; 18190b57cec5SDimitry Andric for (unsigned I = 0; I < T.getNumElements(); ++I) { 18200b57cec5SDimitry Andric if (I != 0) 18210b57cec5SDimitry Andric S += ", "; 18220b57cec5SDimitry Andric S += B.second; 18230b57cec5SDimitry Andric } 18240b57cec5SDimitry Andric S += "}"; 18250b57cec5SDimitry Andric 18260b57cec5SDimitry Andric return std::make_pair(T, S); 18270b57cec5SDimitry Andric } 18280b57cec5SDimitry Andric 18290b57cec5SDimitry Andric std::pair<Type, std::string> Intrinsic::DagEmitter::emitDagSplat(DagInit *DI) { 18300b57cec5SDimitry Andric assert_with_loc(DI->getNumArgs() == 2, "splat() expects two arguments"); 18310b57cec5SDimitry Andric std::pair<Type, std::string> A = emitDagArg(DI->getArg(0), 18320b57cec5SDimitry Andric DI->getArgNameStr(0)); 18330b57cec5SDimitry Andric std::pair<Type, std::string> B = emitDagArg(DI->getArg(1), 18340b57cec5SDimitry Andric DI->getArgNameStr(1)); 18350b57cec5SDimitry Andric 18360b57cec5SDimitry Andric assert_with_loc(B.first.isScalar(), 18370b57cec5SDimitry Andric "splat() requires a scalar int as the second argument"); 18380b57cec5SDimitry Andric 18390b57cec5SDimitry Andric std::string S = "__builtin_shufflevector(" + A.second + ", " + A.second; 18400b57cec5SDimitry Andric for (unsigned I = 0; I < Intr.getBaseType().getNumElements(); ++I) { 18410b57cec5SDimitry Andric S += ", " + B.second; 18420b57cec5SDimitry Andric } 18430b57cec5SDimitry Andric S += ")"; 18440b57cec5SDimitry Andric 18450b57cec5SDimitry Andric return std::make_pair(Intr.getBaseType(), S); 18460b57cec5SDimitry Andric } 18470b57cec5SDimitry Andric 18480b57cec5SDimitry Andric std::pair<Type, std::string> Intrinsic::DagEmitter::emitDagSaveTemp(DagInit *DI) { 18490b57cec5SDimitry Andric assert_with_loc(DI->getNumArgs() == 2, "save_temp() expects two arguments"); 18500b57cec5SDimitry Andric std::pair<Type, std::string> A = emitDagArg(DI->getArg(1), 18510b57cec5SDimitry Andric DI->getArgNameStr(1)); 18520b57cec5SDimitry Andric 18530b57cec5SDimitry Andric assert_with_loc(!A.first.isVoid(), 18540b57cec5SDimitry Andric "Argument to save_temp() must have non-void type!"); 18550b57cec5SDimitry Andric 18560b57cec5SDimitry Andric std::string N = DI->getArgNameStr(0); 18570b57cec5SDimitry Andric assert_with_loc(!N.empty(), 18580b57cec5SDimitry Andric "save_temp() expects a name as the first argument"); 18590b57cec5SDimitry Andric 18600b57cec5SDimitry Andric assert_with_loc(Intr.Variables.find(N) == Intr.Variables.end(), 18610b57cec5SDimitry Andric "Variable already defined!"); 18620b57cec5SDimitry Andric Intr.Variables[N] = Variable(A.first, N + Intr.VariablePostfix); 18630b57cec5SDimitry Andric 18640b57cec5SDimitry Andric std::string S = 18650b57cec5SDimitry Andric A.first.str() + " " + Intr.Variables[N].getName() + " = " + A.second; 18660b57cec5SDimitry Andric 18670b57cec5SDimitry Andric return std::make_pair(Type::getVoid(), S); 18680b57cec5SDimitry Andric } 18690b57cec5SDimitry Andric 18700b57cec5SDimitry Andric std::pair<Type, std::string> 18710b57cec5SDimitry Andric Intrinsic::DagEmitter::emitDagNameReplace(DagInit *DI) { 18720b57cec5SDimitry Andric std::string S = Intr.Name; 18730b57cec5SDimitry Andric 18740b57cec5SDimitry Andric assert_with_loc(DI->getNumArgs() == 2, "name_replace requires 2 arguments!"); 18750b57cec5SDimitry Andric std::string ToReplace = cast<StringInit>(DI->getArg(0))->getAsUnquotedString(); 18760b57cec5SDimitry Andric std::string ReplaceWith = cast<StringInit>(DI->getArg(1))->getAsUnquotedString(); 18770b57cec5SDimitry Andric 18780b57cec5SDimitry Andric size_t Idx = S.find(ToReplace); 18790b57cec5SDimitry Andric 18800b57cec5SDimitry Andric assert_with_loc(Idx != std::string::npos, "name should contain '" + ToReplace + "'!"); 18810b57cec5SDimitry Andric S.replace(Idx, ToReplace.size(), ReplaceWith); 18820b57cec5SDimitry Andric 18830b57cec5SDimitry Andric return std::make_pair(Type::getVoid(), S); 18840b57cec5SDimitry Andric } 18850b57cec5SDimitry Andric 18860b57cec5SDimitry Andric std::pair<Type, std::string> Intrinsic::DagEmitter::emitDagLiteral(DagInit *DI){ 18870b57cec5SDimitry Andric std::string Ty = cast<StringInit>(DI->getArg(0))->getAsUnquotedString(); 18880b57cec5SDimitry Andric std::string Value = cast<StringInit>(DI->getArg(1))->getAsUnquotedString(); 18890b57cec5SDimitry Andric return std::make_pair(Type::fromTypedefName(Ty), Value); 18900b57cec5SDimitry Andric } 18910b57cec5SDimitry Andric 18920b57cec5SDimitry Andric std::pair<Type, std::string> 18930b57cec5SDimitry Andric Intrinsic::DagEmitter::emitDagArg(Init *Arg, std::string ArgName) { 18940b57cec5SDimitry Andric if (!ArgName.empty()) { 18950b57cec5SDimitry Andric assert_with_loc(!Arg->isComplete(), 18960b57cec5SDimitry Andric "Arguments must either be DAGs or names, not both!"); 18970b57cec5SDimitry Andric assert_with_loc(Intr.Variables.find(ArgName) != Intr.Variables.end(), 18980b57cec5SDimitry Andric "Variable not defined!"); 18990b57cec5SDimitry Andric Variable &V = Intr.Variables[ArgName]; 19000b57cec5SDimitry Andric return std::make_pair(V.getType(), V.getName()); 19010b57cec5SDimitry Andric } 19020b57cec5SDimitry Andric 19030b57cec5SDimitry Andric assert(Arg && "Neither ArgName nor Arg?!"); 19040b57cec5SDimitry Andric DagInit *DI = dyn_cast<DagInit>(Arg); 19050b57cec5SDimitry Andric assert_with_loc(DI, "Arguments must either be DAGs or names!"); 19060b57cec5SDimitry Andric 19070b57cec5SDimitry Andric return emitDag(DI); 19080b57cec5SDimitry Andric } 19090b57cec5SDimitry Andric 19100b57cec5SDimitry Andric std::string Intrinsic::generate() { 1911*a7dea167SDimitry Andric // Avoid duplicated code for big and little endian 1912*a7dea167SDimitry Andric if (isBigEndianSafe()) { 1913*a7dea167SDimitry Andric generateImpl(false, "", ""); 1914*a7dea167SDimitry Andric return OS.str(); 1915*a7dea167SDimitry Andric } 19160b57cec5SDimitry Andric // Little endian intrinsics are simple and don't require any argument 19170b57cec5SDimitry Andric // swapping. 19180b57cec5SDimitry Andric OS << "#ifdef __LITTLE_ENDIAN__\n"; 19190b57cec5SDimitry Andric 19200b57cec5SDimitry Andric generateImpl(false, "", ""); 19210b57cec5SDimitry Andric 19220b57cec5SDimitry Andric OS << "#else\n"; 19230b57cec5SDimitry Andric 19240b57cec5SDimitry Andric // Big endian intrinsics are more complex. The user intended these 19250b57cec5SDimitry Andric // intrinsics to operate on a vector "as-if" loaded by (V)LDR, 19260b57cec5SDimitry Andric // but we load as-if (V)LD1. So we should swap all arguments and 19270b57cec5SDimitry Andric // swap the return value too. 19280b57cec5SDimitry Andric // 19290b57cec5SDimitry Andric // If we call sub-intrinsics, we should call a version that does 19300b57cec5SDimitry Andric // not re-swap the arguments! 19310b57cec5SDimitry Andric generateImpl(true, "", "__noswap_"); 19320b57cec5SDimitry Andric 19330b57cec5SDimitry Andric // If we're needed early, create a non-swapping variant for 19340b57cec5SDimitry Andric // big-endian. 19350b57cec5SDimitry Andric if (NeededEarly) { 19360b57cec5SDimitry Andric generateImpl(false, "__noswap_", "__noswap_"); 19370b57cec5SDimitry Andric } 19380b57cec5SDimitry Andric OS << "#endif\n\n"; 19390b57cec5SDimitry Andric 19400b57cec5SDimitry Andric return OS.str(); 19410b57cec5SDimitry Andric } 19420b57cec5SDimitry Andric 19430b57cec5SDimitry Andric void Intrinsic::generateImpl(bool ReverseArguments, 19440b57cec5SDimitry Andric StringRef NamePrefix, StringRef CallPrefix) { 19450b57cec5SDimitry Andric CurrentRecord = R; 19460b57cec5SDimitry Andric 19470b57cec5SDimitry Andric // If we call a macro, our local variables may be corrupted due to 19480b57cec5SDimitry Andric // lack of proper lexical scoping. So, add a globally unique postfix 19490b57cec5SDimitry Andric // to every variable. 19500b57cec5SDimitry Andric // 19510b57cec5SDimitry Andric // indexBody() should have set up the Dependencies set by now. 19520b57cec5SDimitry Andric for (auto *I : Dependencies) 19530b57cec5SDimitry Andric if (I->UseMacro) { 19540b57cec5SDimitry Andric VariablePostfix = "_" + utostr(Emitter.getUniqueNumber()); 19550b57cec5SDimitry Andric break; 19560b57cec5SDimitry Andric } 19570b57cec5SDimitry Andric 19580b57cec5SDimitry Andric initVariables(); 19590b57cec5SDimitry Andric 19600b57cec5SDimitry Andric emitPrototype(NamePrefix); 19610b57cec5SDimitry Andric 19620b57cec5SDimitry Andric if (IsUnavailable) { 19630b57cec5SDimitry Andric OS << " __attribute__((unavailable));"; 19640b57cec5SDimitry Andric } else { 19650b57cec5SDimitry Andric emitOpeningBrace(); 19660b57cec5SDimitry Andric emitShadowedArgs(); 19670b57cec5SDimitry Andric if (ReverseArguments) 19680b57cec5SDimitry Andric emitArgumentReversal(); 19690b57cec5SDimitry Andric emitBody(CallPrefix); 19700b57cec5SDimitry Andric if (ReverseArguments) 19710b57cec5SDimitry Andric emitReturnReversal(); 19720b57cec5SDimitry Andric emitReturn(); 19730b57cec5SDimitry Andric emitClosingBrace(); 19740b57cec5SDimitry Andric } 19750b57cec5SDimitry Andric OS << "\n"; 19760b57cec5SDimitry Andric 19770b57cec5SDimitry Andric CurrentRecord = nullptr; 19780b57cec5SDimitry Andric } 19790b57cec5SDimitry Andric 19800b57cec5SDimitry Andric void Intrinsic::indexBody() { 19810b57cec5SDimitry Andric CurrentRecord = R; 19820b57cec5SDimitry Andric 19830b57cec5SDimitry Andric initVariables(); 19840b57cec5SDimitry Andric emitBody(""); 19850b57cec5SDimitry Andric OS.str(""); 19860b57cec5SDimitry Andric 19870b57cec5SDimitry Andric CurrentRecord = nullptr; 19880b57cec5SDimitry Andric } 19890b57cec5SDimitry Andric 19900b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 19910b57cec5SDimitry Andric // NeonEmitter implementation 19920b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 19930b57cec5SDimitry Andric 19940b57cec5SDimitry Andric Intrinsic &NeonEmitter::getIntrinsic(StringRef Name, ArrayRef<Type> Types) { 19950b57cec5SDimitry Andric // First, look up the name in the intrinsic map. 19960b57cec5SDimitry Andric assert_with_loc(IntrinsicMap.find(Name.str()) != IntrinsicMap.end(), 19970b57cec5SDimitry Andric ("Intrinsic '" + Name + "' not found!").str()); 19980b57cec5SDimitry Andric auto &V = IntrinsicMap.find(Name.str())->second; 19990b57cec5SDimitry Andric std::vector<Intrinsic *> GoodVec; 20000b57cec5SDimitry Andric 20010b57cec5SDimitry Andric // Create a string to print if we end up failing. 20020b57cec5SDimitry Andric std::string ErrMsg = "looking up intrinsic '" + Name.str() + "("; 20030b57cec5SDimitry Andric for (unsigned I = 0; I < Types.size(); ++I) { 20040b57cec5SDimitry Andric if (I != 0) 20050b57cec5SDimitry Andric ErrMsg += ", "; 20060b57cec5SDimitry Andric ErrMsg += Types[I].str(); 20070b57cec5SDimitry Andric } 20080b57cec5SDimitry Andric ErrMsg += ")'\n"; 20090b57cec5SDimitry Andric ErrMsg += "Available overloads:\n"; 20100b57cec5SDimitry Andric 20110b57cec5SDimitry Andric // Now, look through each intrinsic implementation and see if the types are 20120b57cec5SDimitry Andric // compatible. 20130b57cec5SDimitry Andric for (auto &I : V) { 20140b57cec5SDimitry Andric ErrMsg += " - " + I.getReturnType().str() + " " + I.getMangledName(); 20150b57cec5SDimitry Andric ErrMsg += "("; 20160b57cec5SDimitry Andric for (unsigned A = 0; A < I.getNumParams(); ++A) { 20170b57cec5SDimitry Andric if (A != 0) 20180b57cec5SDimitry Andric ErrMsg += ", "; 20190b57cec5SDimitry Andric ErrMsg += I.getParamType(A).str(); 20200b57cec5SDimitry Andric } 20210b57cec5SDimitry Andric ErrMsg += ")\n"; 20220b57cec5SDimitry Andric 20230b57cec5SDimitry Andric if (I.getNumParams() != Types.size()) 20240b57cec5SDimitry Andric continue; 20250b57cec5SDimitry Andric 20260b57cec5SDimitry Andric bool Good = true; 20270b57cec5SDimitry Andric for (unsigned Arg = 0; Arg < Types.size(); ++Arg) { 20280b57cec5SDimitry Andric if (I.getParamType(Arg) != Types[Arg]) { 20290b57cec5SDimitry Andric Good = false; 20300b57cec5SDimitry Andric break; 20310b57cec5SDimitry Andric } 20320b57cec5SDimitry Andric } 20330b57cec5SDimitry Andric if (Good) 20340b57cec5SDimitry Andric GoodVec.push_back(&I); 20350b57cec5SDimitry Andric } 20360b57cec5SDimitry Andric 20370b57cec5SDimitry Andric assert_with_loc(!GoodVec.empty(), 20380b57cec5SDimitry Andric "No compatible intrinsic found - " + ErrMsg); 20390b57cec5SDimitry Andric assert_with_loc(GoodVec.size() == 1, "Multiple overloads found - " + ErrMsg); 20400b57cec5SDimitry Andric 20410b57cec5SDimitry Andric return *GoodVec.front(); 20420b57cec5SDimitry Andric } 20430b57cec5SDimitry Andric 20440b57cec5SDimitry Andric void NeonEmitter::createIntrinsic(Record *R, 20450b57cec5SDimitry Andric SmallVectorImpl<Intrinsic *> &Out) { 20460b57cec5SDimitry Andric std::string Name = R->getValueAsString("Name"); 20470b57cec5SDimitry Andric std::string Proto = R->getValueAsString("Prototype"); 20480b57cec5SDimitry Andric std::string Types = R->getValueAsString("Types"); 20490b57cec5SDimitry Andric Record *OperationRec = R->getValueAsDef("Operation"); 20500b57cec5SDimitry Andric bool CartesianProductOfTypes = R->getValueAsBit("CartesianProductOfTypes"); 20510b57cec5SDimitry Andric bool BigEndianSafe = R->getValueAsBit("BigEndianSafe"); 20520b57cec5SDimitry Andric std::string Guard = R->getValueAsString("ArchGuard"); 20530b57cec5SDimitry Andric bool IsUnavailable = OperationRec->getValueAsBit("Unavailable"); 20540b57cec5SDimitry Andric 20550b57cec5SDimitry Andric // Set the global current record. This allows assert_with_loc to produce 20560b57cec5SDimitry Andric // decent location information even when highly nested. 20570b57cec5SDimitry Andric CurrentRecord = R; 20580b57cec5SDimitry Andric 20590b57cec5SDimitry Andric ListInit *Body = OperationRec->getValueAsListInit("Ops"); 20600b57cec5SDimitry Andric 20610b57cec5SDimitry Andric std::vector<TypeSpec> TypeSpecs = TypeSpec::fromTypeSpecs(Types); 20620b57cec5SDimitry Andric 20630b57cec5SDimitry Andric ClassKind CK = ClassNone; 20640b57cec5SDimitry Andric if (R->getSuperClasses().size() >= 2) 20650b57cec5SDimitry Andric CK = ClassMap[R->getSuperClasses()[1].first]; 20660b57cec5SDimitry Andric 20670b57cec5SDimitry Andric std::vector<std::pair<TypeSpec, TypeSpec>> NewTypeSpecs; 20680b57cec5SDimitry Andric for (auto TS : TypeSpecs) { 20690b57cec5SDimitry Andric if (CartesianProductOfTypes) { 20700b57cec5SDimitry Andric Type DefaultT(TS, 'd'); 20710b57cec5SDimitry Andric for (auto SrcTS : TypeSpecs) { 20720b57cec5SDimitry Andric Type DefaultSrcT(SrcTS, 'd'); 20730b57cec5SDimitry Andric if (TS == SrcTS || 20740b57cec5SDimitry Andric DefaultSrcT.getSizeInBits() != DefaultT.getSizeInBits()) 20750b57cec5SDimitry Andric continue; 20760b57cec5SDimitry Andric NewTypeSpecs.push_back(std::make_pair(TS, SrcTS)); 20770b57cec5SDimitry Andric } 20780b57cec5SDimitry Andric } else { 20790b57cec5SDimitry Andric NewTypeSpecs.push_back(std::make_pair(TS, TS)); 20800b57cec5SDimitry Andric } 20810b57cec5SDimitry Andric } 20820b57cec5SDimitry Andric 20830b57cec5SDimitry Andric llvm::sort(NewTypeSpecs); 20840b57cec5SDimitry Andric NewTypeSpecs.erase(std::unique(NewTypeSpecs.begin(), NewTypeSpecs.end()), 20850b57cec5SDimitry Andric NewTypeSpecs.end()); 20860b57cec5SDimitry Andric auto &Entry = IntrinsicMap[Name]; 20870b57cec5SDimitry Andric 20880b57cec5SDimitry Andric for (auto &I : NewTypeSpecs) { 20890b57cec5SDimitry Andric Entry.emplace_back(R, Name, Proto, I.first, I.second, CK, Body, *this, 20900b57cec5SDimitry Andric Guard, IsUnavailable, BigEndianSafe); 20910b57cec5SDimitry Andric Out.push_back(&Entry.back()); 20920b57cec5SDimitry Andric } 20930b57cec5SDimitry Andric 20940b57cec5SDimitry Andric CurrentRecord = nullptr; 20950b57cec5SDimitry Andric } 20960b57cec5SDimitry Andric 20970b57cec5SDimitry Andric /// genBuiltinsDef: Generate the BuiltinsARM.def and BuiltinsAArch64.def 20980b57cec5SDimitry Andric /// declaration of builtins, checking for unique builtin declarations. 20990b57cec5SDimitry Andric void NeonEmitter::genBuiltinsDef(raw_ostream &OS, 21000b57cec5SDimitry Andric SmallVectorImpl<Intrinsic *> &Defs) { 21010b57cec5SDimitry Andric OS << "#ifdef GET_NEON_BUILTINS\n"; 21020b57cec5SDimitry Andric 21030b57cec5SDimitry Andric // We only want to emit a builtin once, and we want to emit them in 21040b57cec5SDimitry Andric // alphabetical order, so use a std::set. 21050b57cec5SDimitry Andric std::set<std::string> Builtins; 21060b57cec5SDimitry Andric 21070b57cec5SDimitry Andric for (auto *Def : Defs) { 21080b57cec5SDimitry Andric if (Def->hasBody()) 21090b57cec5SDimitry Andric continue; 21100b57cec5SDimitry Andric // Functions with 'a' (the splat code) in the type prototype should not get 21110b57cec5SDimitry Andric // their own builtin as they use the non-splat variant. 21120b57cec5SDimitry Andric if (Def->hasSplat()) 21130b57cec5SDimitry Andric continue; 21140b57cec5SDimitry Andric 21150b57cec5SDimitry Andric std::string S = "BUILTIN(__builtin_neon_" + Def->getMangledName() + ", \""; 21160b57cec5SDimitry Andric 21170b57cec5SDimitry Andric S += Def->getBuiltinTypeStr(); 21180b57cec5SDimitry Andric S += "\", \"n\")"; 21190b57cec5SDimitry Andric 21200b57cec5SDimitry Andric Builtins.insert(S); 21210b57cec5SDimitry Andric } 21220b57cec5SDimitry Andric 21230b57cec5SDimitry Andric for (auto &S : Builtins) 21240b57cec5SDimitry Andric OS << S << "\n"; 21250b57cec5SDimitry Andric OS << "#endif\n\n"; 21260b57cec5SDimitry Andric } 21270b57cec5SDimitry Andric 21280b57cec5SDimitry Andric /// Generate the ARM and AArch64 overloaded type checking code for 21290b57cec5SDimitry Andric /// SemaChecking.cpp, checking for unique builtin declarations. 21300b57cec5SDimitry Andric void NeonEmitter::genOverloadTypeCheckCode(raw_ostream &OS, 21310b57cec5SDimitry Andric SmallVectorImpl<Intrinsic *> &Defs) { 21320b57cec5SDimitry Andric OS << "#ifdef GET_NEON_OVERLOAD_CHECK\n"; 21330b57cec5SDimitry Andric 21340b57cec5SDimitry Andric // We record each overload check line before emitting because subsequent Inst 21350b57cec5SDimitry Andric // definitions may extend the number of permitted types (i.e. augment the 21360b57cec5SDimitry Andric // Mask). Use std::map to avoid sorting the table by hash number. 21370b57cec5SDimitry Andric struct OverloadInfo { 21380b57cec5SDimitry Andric uint64_t Mask; 21390b57cec5SDimitry Andric int PtrArgNum; 21400b57cec5SDimitry Andric bool HasConstPtr; 21410b57cec5SDimitry Andric OverloadInfo() : Mask(0ULL), PtrArgNum(0), HasConstPtr(false) {} 21420b57cec5SDimitry Andric }; 21430b57cec5SDimitry Andric std::map<std::string, OverloadInfo> OverloadMap; 21440b57cec5SDimitry Andric 21450b57cec5SDimitry Andric for (auto *Def : Defs) { 21460b57cec5SDimitry Andric // If the def has a body (that is, it has Operation DAGs), it won't call 21470b57cec5SDimitry Andric // __builtin_neon_* so we don't need to generate a definition for it. 21480b57cec5SDimitry Andric if (Def->hasBody()) 21490b57cec5SDimitry Andric continue; 21500b57cec5SDimitry Andric // Functions with 'a' (the splat code) in the type prototype should not get 21510b57cec5SDimitry Andric // their own builtin as they use the non-splat variant. 21520b57cec5SDimitry Andric if (Def->hasSplat()) 21530b57cec5SDimitry Andric continue; 21540b57cec5SDimitry Andric // Functions which have a scalar argument cannot be overloaded, no need to 21550b57cec5SDimitry Andric // check them if we are emitting the type checking code. 21560b57cec5SDimitry Andric if (Def->protoHasScalar()) 21570b57cec5SDimitry Andric continue; 21580b57cec5SDimitry Andric 21590b57cec5SDimitry Andric uint64_t Mask = 0ULL; 21600b57cec5SDimitry Andric Type Ty = Def->getReturnType(); 21610b57cec5SDimitry Andric if (Def->getProto()[0] == 'v' || 21620b57cec5SDimitry Andric isFloatingPointProtoModifier(Def->getProto()[0])) 21630b57cec5SDimitry Andric Ty = Def->getParamType(0); 21640b57cec5SDimitry Andric if (Ty.isPointer()) 21650b57cec5SDimitry Andric Ty = Def->getParamType(1); 21660b57cec5SDimitry Andric 21670b57cec5SDimitry Andric Mask |= 1ULL << Ty.getNeonEnum(); 21680b57cec5SDimitry Andric 21690b57cec5SDimitry Andric // Check if the function has a pointer or const pointer argument. 21700b57cec5SDimitry Andric std::string Proto = Def->getProto(); 21710b57cec5SDimitry Andric int PtrArgNum = -1; 21720b57cec5SDimitry Andric bool HasConstPtr = false; 21730b57cec5SDimitry Andric for (unsigned I = 0; I < Def->getNumParams(); ++I) { 21740b57cec5SDimitry Andric char ArgType = Proto[I + 1]; 21750b57cec5SDimitry Andric if (ArgType == 'c') { 21760b57cec5SDimitry Andric HasConstPtr = true; 21770b57cec5SDimitry Andric PtrArgNum = I; 21780b57cec5SDimitry Andric break; 21790b57cec5SDimitry Andric } 21800b57cec5SDimitry Andric if (ArgType == 'p') { 21810b57cec5SDimitry Andric PtrArgNum = I; 21820b57cec5SDimitry Andric break; 21830b57cec5SDimitry Andric } 21840b57cec5SDimitry Andric } 21850b57cec5SDimitry Andric // For sret builtins, adjust the pointer argument index. 21860b57cec5SDimitry Andric if (PtrArgNum >= 0 && Def->getReturnType().getNumVectors() > 1) 21870b57cec5SDimitry Andric PtrArgNum += 1; 21880b57cec5SDimitry Andric 21890b57cec5SDimitry Andric std::string Name = Def->getName(); 21900b57cec5SDimitry Andric // Omit type checking for the pointer arguments of vld1_lane, vld1_dup, 21910b57cec5SDimitry Andric // and vst1_lane intrinsics. Using a pointer to the vector element 21920b57cec5SDimitry Andric // type with one of those operations causes codegen to select an aligned 21930b57cec5SDimitry Andric // load/store instruction. If you want an unaligned operation, 21940b57cec5SDimitry Andric // the pointer argument needs to have less alignment than element type, 21950b57cec5SDimitry Andric // so just accept any pointer type. 21960b57cec5SDimitry Andric if (Name == "vld1_lane" || Name == "vld1_dup" || Name == "vst1_lane") { 21970b57cec5SDimitry Andric PtrArgNum = -1; 21980b57cec5SDimitry Andric HasConstPtr = false; 21990b57cec5SDimitry Andric } 22000b57cec5SDimitry Andric 22010b57cec5SDimitry Andric if (Mask) { 22020b57cec5SDimitry Andric std::string Name = Def->getMangledName(); 22030b57cec5SDimitry Andric OverloadMap.insert(std::make_pair(Name, OverloadInfo())); 22040b57cec5SDimitry Andric OverloadInfo &OI = OverloadMap[Name]; 22050b57cec5SDimitry Andric OI.Mask |= Mask; 22060b57cec5SDimitry Andric OI.PtrArgNum |= PtrArgNum; 22070b57cec5SDimitry Andric OI.HasConstPtr = HasConstPtr; 22080b57cec5SDimitry Andric } 22090b57cec5SDimitry Andric } 22100b57cec5SDimitry Andric 22110b57cec5SDimitry Andric for (auto &I : OverloadMap) { 22120b57cec5SDimitry Andric OverloadInfo &OI = I.second; 22130b57cec5SDimitry Andric 22140b57cec5SDimitry Andric OS << "case NEON::BI__builtin_neon_" << I.first << ": "; 22150b57cec5SDimitry Andric OS << "mask = 0x" << Twine::utohexstr(OI.Mask) << "ULL"; 22160b57cec5SDimitry Andric if (OI.PtrArgNum >= 0) 22170b57cec5SDimitry Andric OS << "; PtrArgNum = " << OI.PtrArgNum; 22180b57cec5SDimitry Andric if (OI.HasConstPtr) 22190b57cec5SDimitry Andric OS << "; HasConstPtr = true"; 22200b57cec5SDimitry Andric OS << "; break;\n"; 22210b57cec5SDimitry Andric } 22220b57cec5SDimitry Andric OS << "#endif\n\n"; 22230b57cec5SDimitry Andric } 22240b57cec5SDimitry Andric 22250b57cec5SDimitry Andric void NeonEmitter::genIntrinsicRangeCheckCode(raw_ostream &OS, 22260b57cec5SDimitry Andric SmallVectorImpl<Intrinsic *> &Defs) { 22270b57cec5SDimitry Andric OS << "#ifdef GET_NEON_IMMEDIATE_CHECK\n"; 22280b57cec5SDimitry Andric 22290b57cec5SDimitry Andric std::set<std::string> Emitted; 22300b57cec5SDimitry Andric 22310b57cec5SDimitry Andric for (auto *Def : Defs) { 22320b57cec5SDimitry Andric if (Def->hasBody()) 22330b57cec5SDimitry Andric continue; 22340b57cec5SDimitry Andric // Functions with 'a' (the splat code) in the type prototype should not get 22350b57cec5SDimitry Andric // their own builtin as they use the non-splat variant. 22360b57cec5SDimitry Andric if (Def->hasSplat()) 22370b57cec5SDimitry Andric continue; 22380b57cec5SDimitry Andric // Functions which do not have an immediate do not need to have range 22390b57cec5SDimitry Andric // checking code emitted. 22400b57cec5SDimitry Andric if (!Def->hasImmediate()) 22410b57cec5SDimitry Andric continue; 22420b57cec5SDimitry Andric if (Emitted.find(Def->getMangledName()) != Emitted.end()) 22430b57cec5SDimitry Andric continue; 22440b57cec5SDimitry Andric 22450b57cec5SDimitry Andric std::string LowerBound, UpperBound; 22460b57cec5SDimitry Andric 22470b57cec5SDimitry Andric Record *R = Def->getRecord(); 22480b57cec5SDimitry Andric if (R->getValueAsBit("isVCVT_N")) { 22490b57cec5SDimitry Andric // VCVT between floating- and fixed-point values takes an immediate 22500b57cec5SDimitry Andric // in the range [1, 32) for f32 or [1, 64) for f64 or [1, 16) for f16. 22510b57cec5SDimitry Andric LowerBound = "1"; 22520b57cec5SDimitry Andric if (Def->getBaseType().getElementSizeInBits() == 16 || 22530b57cec5SDimitry Andric Def->getName().find('h') != std::string::npos) 22540b57cec5SDimitry Andric // VCVTh operating on FP16 intrinsics in range [1, 16) 22550b57cec5SDimitry Andric UpperBound = "15"; 22560b57cec5SDimitry Andric else if (Def->getBaseType().getElementSizeInBits() == 32) 22570b57cec5SDimitry Andric UpperBound = "31"; 22580b57cec5SDimitry Andric else 22590b57cec5SDimitry Andric UpperBound = "63"; 22600b57cec5SDimitry Andric } else if (R->getValueAsBit("isScalarShift")) { 22610b57cec5SDimitry Andric // Right shifts have an 'r' in the name, left shifts do not. Convert 22620b57cec5SDimitry Andric // instructions have the same bounds and right shifts. 22630b57cec5SDimitry Andric if (Def->getName().find('r') != std::string::npos || 22640b57cec5SDimitry Andric Def->getName().find("cvt") != std::string::npos) 22650b57cec5SDimitry Andric LowerBound = "1"; 22660b57cec5SDimitry Andric 22670b57cec5SDimitry Andric UpperBound = utostr(Def->getReturnType().getElementSizeInBits() - 1); 22680b57cec5SDimitry Andric } else if (R->getValueAsBit("isShift")) { 22690b57cec5SDimitry Andric // Builtins which are overloaded by type will need to have their upper 22700b57cec5SDimitry Andric // bound computed at Sema time based on the type constant. 22710b57cec5SDimitry Andric 22720b57cec5SDimitry Andric // Right shifts have an 'r' in the name, left shifts do not. 22730b57cec5SDimitry Andric if (Def->getName().find('r') != std::string::npos) 22740b57cec5SDimitry Andric LowerBound = "1"; 22750b57cec5SDimitry Andric UpperBound = "RFT(TV, true)"; 22760b57cec5SDimitry Andric } else if (Def->getClassKind(true) == ClassB) { 22770b57cec5SDimitry Andric // ClassB intrinsics have a type (and hence lane number) that is only 22780b57cec5SDimitry Andric // known at runtime. 22790b57cec5SDimitry Andric if (R->getValueAsBit("isLaneQ")) 22800b57cec5SDimitry Andric UpperBound = "RFT(TV, false, true)"; 22810b57cec5SDimitry Andric else 22820b57cec5SDimitry Andric UpperBound = "RFT(TV, false, false)"; 22830b57cec5SDimitry Andric } else { 22840b57cec5SDimitry Andric // The immediate generally refers to a lane in the preceding argument. 22850b57cec5SDimitry Andric assert(Def->getImmediateIdx() > 0); 22860b57cec5SDimitry Andric Type T = Def->getParamType(Def->getImmediateIdx() - 1); 22870b57cec5SDimitry Andric UpperBound = utostr(T.getNumElements() - 1); 22880b57cec5SDimitry Andric } 22890b57cec5SDimitry Andric 22900b57cec5SDimitry Andric // Calculate the index of the immediate that should be range checked. 22910b57cec5SDimitry Andric unsigned Idx = Def->getNumParams(); 22920b57cec5SDimitry Andric if (Def->hasImmediate()) 22930b57cec5SDimitry Andric Idx = Def->getGeneratedParamIdx(Def->getImmediateIdx()); 22940b57cec5SDimitry Andric 22950b57cec5SDimitry Andric OS << "case NEON::BI__builtin_neon_" << Def->getMangledName() << ": " 22960b57cec5SDimitry Andric << "i = " << Idx << ";"; 22970b57cec5SDimitry Andric if (!LowerBound.empty()) 22980b57cec5SDimitry Andric OS << " l = " << LowerBound << ";"; 22990b57cec5SDimitry Andric if (!UpperBound.empty()) 23000b57cec5SDimitry Andric OS << " u = " << UpperBound << ";"; 23010b57cec5SDimitry Andric OS << " break;\n"; 23020b57cec5SDimitry Andric 23030b57cec5SDimitry Andric Emitted.insert(Def->getMangledName()); 23040b57cec5SDimitry Andric } 23050b57cec5SDimitry Andric 23060b57cec5SDimitry Andric OS << "#endif\n\n"; 23070b57cec5SDimitry Andric } 23080b57cec5SDimitry Andric 23090b57cec5SDimitry Andric /// runHeader - Emit a file with sections defining: 23100b57cec5SDimitry Andric /// 1. the NEON section of BuiltinsARM.def and BuiltinsAArch64.def. 23110b57cec5SDimitry Andric /// 2. the SemaChecking code for the type overload checking. 23120b57cec5SDimitry Andric /// 3. the SemaChecking code for validation of intrinsic immediate arguments. 23130b57cec5SDimitry Andric void NeonEmitter::runHeader(raw_ostream &OS) { 23140b57cec5SDimitry Andric std::vector<Record *> RV = Records.getAllDerivedDefinitions("Inst"); 23150b57cec5SDimitry Andric 23160b57cec5SDimitry Andric SmallVector<Intrinsic *, 128> Defs; 23170b57cec5SDimitry Andric for (auto *R : RV) 23180b57cec5SDimitry Andric createIntrinsic(R, Defs); 23190b57cec5SDimitry Andric 23200b57cec5SDimitry Andric // Generate shared BuiltinsXXX.def 23210b57cec5SDimitry Andric genBuiltinsDef(OS, Defs); 23220b57cec5SDimitry Andric 23230b57cec5SDimitry Andric // Generate ARM overloaded type checking code for SemaChecking.cpp 23240b57cec5SDimitry Andric genOverloadTypeCheckCode(OS, Defs); 23250b57cec5SDimitry Andric 23260b57cec5SDimitry Andric // Generate ARM range checking code for shift/lane immediates. 23270b57cec5SDimitry Andric genIntrinsicRangeCheckCode(OS, Defs); 23280b57cec5SDimitry Andric } 23290b57cec5SDimitry Andric 23300b57cec5SDimitry Andric /// run - Read the records in arm_neon.td and output arm_neon.h. arm_neon.h 23310b57cec5SDimitry Andric /// is comprised of type definitions and function declarations. 23320b57cec5SDimitry Andric void NeonEmitter::run(raw_ostream &OS) { 23330b57cec5SDimitry Andric OS << "/*===---- arm_neon.h - ARM Neon intrinsics " 23340b57cec5SDimitry Andric "------------------------------" 23350b57cec5SDimitry Andric "---===\n" 23360b57cec5SDimitry Andric " *\n" 23370b57cec5SDimitry Andric " * Permission is hereby granted, free of charge, to any person " 23380b57cec5SDimitry Andric "obtaining " 23390b57cec5SDimitry Andric "a copy\n" 23400b57cec5SDimitry Andric " * of this software and associated documentation files (the " 23410b57cec5SDimitry Andric "\"Software\")," 23420b57cec5SDimitry Andric " to deal\n" 23430b57cec5SDimitry Andric " * in the Software without restriction, including without limitation " 23440b57cec5SDimitry Andric "the " 23450b57cec5SDimitry Andric "rights\n" 23460b57cec5SDimitry Andric " * to use, copy, modify, merge, publish, distribute, sublicense, " 23470b57cec5SDimitry Andric "and/or sell\n" 23480b57cec5SDimitry Andric " * copies of the Software, and to permit persons to whom the Software " 23490b57cec5SDimitry Andric "is\n" 23500b57cec5SDimitry Andric " * furnished to do so, subject to the following conditions:\n" 23510b57cec5SDimitry Andric " *\n" 23520b57cec5SDimitry Andric " * The above copyright notice and this permission notice shall be " 23530b57cec5SDimitry Andric "included in\n" 23540b57cec5SDimitry Andric " * all copies or substantial portions of the Software.\n" 23550b57cec5SDimitry Andric " *\n" 23560b57cec5SDimitry Andric " * THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, " 23570b57cec5SDimitry Andric "EXPRESS OR\n" 23580b57cec5SDimitry Andric " * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF " 23590b57cec5SDimitry Andric "MERCHANTABILITY,\n" 23600b57cec5SDimitry Andric " * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT " 23610b57cec5SDimitry Andric "SHALL THE\n" 23620b57cec5SDimitry Andric " * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR " 23630b57cec5SDimitry Andric "OTHER\n" 23640b57cec5SDimitry Andric " * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, " 23650b57cec5SDimitry Andric "ARISING FROM,\n" 23660b57cec5SDimitry Andric " * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER " 23670b57cec5SDimitry Andric "DEALINGS IN\n" 23680b57cec5SDimitry Andric " * THE SOFTWARE.\n" 23690b57cec5SDimitry Andric " *\n" 23700b57cec5SDimitry Andric " *===-----------------------------------------------------------------" 23710b57cec5SDimitry Andric "---" 23720b57cec5SDimitry Andric "---===\n" 23730b57cec5SDimitry Andric " */\n\n"; 23740b57cec5SDimitry Andric 23750b57cec5SDimitry Andric OS << "#ifndef __ARM_NEON_H\n"; 23760b57cec5SDimitry Andric OS << "#define __ARM_NEON_H\n\n"; 23770b57cec5SDimitry Andric 23780b57cec5SDimitry Andric OS << "#if !defined(__ARM_NEON)\n"; 23790b57cec5SDimitry Andric OS << "#error \"NEON support not enabled\"\n"; 23800b57cec5SDimitry Andric OS << "#endif\n\n"; 23810b57cec5SDimitry Andric 23820b57cec5SDimitry Andric OS << "#include <stdint.h>\n\n"; 23830b57cec5SDimitry Andric 23840b57cec5SDimitry Andric // Emit NEON-specific scalar typedefs. 23850b57cec5SDimitry Andric OS << "typedef float float32_t;\n"; 23860b57cec5SDimitry Andric OS << "typedef __fp16 float16_t;\n"; 23870b57cec5SDimitry Andric 23880b57cec5SDimitry Andric OS << "#ifdef __aarch64__\n"; 23890b57cec5SDimitry Andric OS << "typedef double float64_t;\n"; 23900b57cec5SDimitry Andric OS << "#endif\n\n"; 23910b57cec5SDimitry Andric 23920b57cec5SDimitry Andric // For now, signedness of polynomial types depends on target 23930b57cec5SDimitry Andric OS << "#ifdef __aarch64__\n"; 23940b57cec5SDimitry Andric OS << "typedef uint8_t poly8_t;\n"; 23950b57cec5SDimitry Andric OS << "typedef uint16_t poly16_t;\n"; 23960b57cec5SDimitry Andric OS << "typedef uint64_t poly64_t;\n"; 23970b57cec5SDimitry Andric OS << "typedef __uint128_t poly128_t;\n"; 23980b57cec5SDimitry Andric OS << "#else\n"; 23990b57cec5SDimitry Andric OS << "typedef int8_t poly8_t;\n"; 24000b57cec5SDimitry Andric OS << "typedef int16_t poly16_t;\n"; 24010b57cec5SDimitry Andric OS << "#endif\n"; 24020b57cec5SDimitry Andric 24030b57cec5SDimitry Andric // Emit Neon vector typedefs. 24040b57cec5SDimitry Andric std::string TypedefTypes( 24050b57cec5SDimitry Andric "cQcsQsiQilQlUcQUcUsQUsUiQUiUlQUlhQhfQfdQdPcQPcPsQPsPlQPl"); 24060b57cec5SDimitry Andric std::vector<TypeSpec> TDTypeVec = TypeSpec::fromTypeSpecs(TypedefTypes); 24070b57cec5SDimitry Andric 24080b57cec5SDimitry Andric // Emit vector typedefs. 24090b57cec5SDimitry Andric bool InIfdef = false; 24100b57cec5SDimitry Andric for (auto &TS : TDTypeVec) { 24110b57cec5SDimitry Andric bool IsA64 = false; 24120b57cec5SDimitry Andric Type T(TS, 'd'); 24130b57cec5SDimitry Andric if (T.isDouble() || (T.isPoly() && T.isLong())) 24140b57cec5SDimitry Andric IsA64 = true; 24150b57cec5SDimitry Andric 24160b57cec5SDimitry Andric if (InIfdef && !IsA64) { 24170b57cec5SDimitry Andric OS << "#endif\n"; 24180b57cec5SDimitry Andric InIfdef = false; 24190b57cec5SDimitry Andric } 24200b57cec5SDimitry Andric if (!InIfdef && IsA64) { 24210b57cec5SDimitry Andric OS << "#ifdef __aarch64__\n"; 24220b57cec5SDimitry Andric InIfdef = true; 24230b57cec5SDimitry Andric } 24240b57cec5SDimitry Andric 24250b57cec5SDimitry Andric if (T.isPoly()) 24260b57cec5SDimitry Andric OS << "typedef __attribute__((neon_polyvector_type("; 24270b57cec5SDimitry Andric else 24280b57cec5SDimitry Andric OS << "typedef __attribute__((neon_vector_type("; 24290b57cec5SDimitry Andric 24300b57cec5SDimitry Andric Type T2 = T; 24310b57cec5SDimitry Andric T2.makeScalar(); 24320b57cec5SDimitry Andric OS << T.getNumElements() << "))) "; 24330b57cec5SDimitry Andric OS << T2.str(); 24340b57cec5SDimitry Andric OS << " " << T.str() << ";\n"; 24350b57cec5SDimitry Andric } 24360b57cec5SDimitry Andric if (InIfdef) 24370b57cec5SDimitry Andric OS << "#endif\n"; 24380b57cec5SDimitry Andric OS << "\n"; 24390b57cec5SDimitry Andric 24400b57cec5SDimitry Andric // Emit struct typedefs. 24410b57cec5SDimitry Andric InIfdef = false; 24420b57cec5SDimitry Andric for (unsigned NumMembers = 2; NumMembers <= 4; ++NumMembers) { 24430b57cec5SDimitry Andric for (auto &TS : TDTypeVec) { 24440b57cec5SDimitry Andric bool IsA64 = false; 24450b57cec5SDimitry Andric Type T(TS, 'd'); 24460b57cec5SDimitry Andric if (T.isDouble() || (T.isPoly() && T.isLong())) 24470b57cec5SDimitry Andric IsA64 = true; 24480b57cec5SDimitry Andric 24490b57cec5SDimitry Andric if (InIfdef && !IsA64) { 24500b57cec5SDimitry Andric OS << "#endif\n"; 24510b57cec5SDimitry Andric InIfdef = false; 24520b57cec5SDimitry Andric } 24530b57cec5SDimitry Andric if (!InIfdef && IsA64) { 24540b57cec5SDimitry Andric OS << "#ifdef __aarch64__\n"; 24550b57cec5SDimitry Andric InIfdef = true; 24560b57cec5SDimitry Andric } 24570b57cec5SDimitry Andric 24580b57cec5SDimitry Andric char M = '2' + (NumMembers - 2); 24590b57cec5SDimitry Andric Type VT(TS, M); 24600b57cec5SDimitry Andric OS << "typedef struct " << VT.str() << " {\n"; 24610b57cec5SDimitry Andric OS << " " << T.str() << " val"; 24620b57cec5SDimitry Andric OS << "[" << NumMembers << "]"; 24630b57cec5SDimitry Andric OS << ";\n} "; 24640b57cec5SDimitry Andric OS << VT.str() << ";\n"; 24650b57cec5SDimitry Andric OS << "\n"; 24660b57cec5SDimitry Andric } 24670b57cec5SDimitry Andric } 24680b57cec5SDimitry Andric if (InIfdef) 24690b57cec5SDimitry Andric OS << "#endif\n"; 24700b57cec5SDimitry Andric OS << "\n"; 24710b57cec5SDimitry Andric 24720b57cec5SDimitry Andric OS << "#define __ai static __inline__ __attribute__((__always_inline__, " 24730b57cec5SDimitry Andric "__nodebug__))\n\n"; 24740b57cec5SDimitry Andric 24750b57cec5SDimitry Andric SmallVector<Intrinsic *, 128> Defs; 24760b57cec5SDimitry Andric std::vector<Record *> RV = Records.getAllDerivedDefinitions("Inst"); 24770b57cec5SDimitry Andric for (auto *R : RV) 24780b57cec5SDimitry Andric createIntrinsic(R, Defs); 24790b57cec5SDimitry Andric 24800b57cec5SDimitry Andric for (auto *I : Defs) 24810b57cec5SDimitry Andric I->indexBody(); 24820b57cec5SDimitry Andric 2483*a7dea167SDimitry Andric llvm::stable_sort(Defs, llvm::deref<std::less<>>()); 24840b57cec5SDimitry Andric 24850b57cec5SDimitry Andric // Only emit a def when its requirements have been met. 24860b57cec5SDimitry Andric // FIXME: This loop could be made faster, but it's fast enough for now. 24870b57cec5SDimitry Andric bool MadeProgress = true; 24880b57cec5SDimitry Andric std::string InGuard; 24890b57cec5SDimitry Andric while (!Defs.empty() && MadeProgress) { 24900b57cec5SDimitry Andric MadeProgress = false; 24910b57cec5SDimitry Andric 24920b57cec5SDimitry Andric for (SmallVector<Intrinsic *, 128>::iterator I = Defs.begin(); 24930b57cec5SDimitry Andric I != Defs.end(); /*No step*/) { 24940b57cec5SDimitry Andric bool DependenciesSatisfied = true; 24950b57cec5SDimitry Andric for (auto *II : (*I)->getDependencies()) { 24960b57cec5SDimitry Andric if (llvm::is_contained(Defs, II)) 24970b57cec5SDimitry Andric DependenciesSatisfied = false; 24980b57cec5SDimitry Andric } 24990b57cec5SDimitry Andric if (!DependenciesSatisfied) { 25000b57cec5SDimitry Andric // Try the next one. 25010b57cec5SDimitry Andric ++I; 25020b57cec5SDimitry Andric continue; 25030b57cec5SDimitry Andric } 25040b57cec5SDimitry Andric 25050b57cec5SDimitry Andric // Emit #endif/#if pair if needed. 25060b57cec5SDimitry Andric if ((*I)->getGuard() != InGuard) { 25070b57cec5SDimitry Andric if (!InGuard.empty()) 25080b57cec5SDimitry Andric OS << "#endif\n"; 25090b57cec5SDimitry Andric InGuard = (*I)->getGuard(); 25100b57cec5SDimitry Andric if (!InGuard.empty()) 25110b57cec5SDimitry Andric OS << "#if " << InGuard << "\n"; 25120b57cec5SDimitry Andric } 25130b57cec5SDimitry Andric 25140b57cec5SDimitry Andric // Actually generate the intrinsic code. 25150b57cec5SDimitry Andric OS << (*I)->generate(); 25160b57cec5SDimitry Andric 25170b57cec5SDimitry Andric MadeProgress = true; 25180b57cec5SDimitry Andric I = Defs.erase(I); 25190b57cec5SDimitry Andric } 25200b57cec5SDimitry Andric } 25210b57cec5SDimitry Andric assert(Defs.empty() && "Some requirements were not satisfied!"); 25220b57cec5SDimitry Andric if (!InGuard.empty()) 25230b57cec5SDimitry Andric OS << "#endif\n"; 25240b57cec5SDimitry Andric 25250b57cec5SDimitry Andric OS << "\n"; 25260b57cec5SDimitry Andric OS << "#undef __ai\n\n"; 25270b57cec5SDimitry Andric OS << "#endif /* __ARM_NEON_H */\n"; 25280b57cec5SDimitry Andric } 25290b57cec5SDimitry Andric 25300b57cec5SDimitry Andric /// run - Read the records in arm_fp16.td and output arm_fp16.h. arm_fp16.h 25310b57cec5SDimitry Andric /// is comprised of type definitions and function declarations. 25320b57cec5SDimitry Andric void NeonEmitter::runFP16(raw_ostream &OS) { 25330b57cec5SDimitry Andric OS << "/*===---- arm_fp16.h - ARM FP16 intrinsics " 25340b57cec5SDimitry Andric "------------------------------" 25350b57cec5SDimitry Andric "---===\n" 25360b57cec5SDimitry Andric " *\n" 25370b57cec5SDimitry Andric " * Permission is hereby granted, free of charge, to any person " 25380b57cec5SDimitry Andric "obtaining a copy\n" 25390b57cec5SDimitry Andric " * of this software and associated documentation files (the " 25400b57cec5SDimitry Andric "\"Software\"), to deal\n" 25410b57cec5SDimitry Andric " * in the Software without restriction, including without limitation " 25420b57cec5SDimitry Andric "the rights\n" 25430b57cec5SDimitry Andric " * to use, copy, modify, merge, publish, distribute, sublicense, " 25440b57cec5SDimitry Andric "and/or sell\n" 25450b57cec5SDimitry Andric " * copies of the Software, and to permit persons to whom the Software " 25460b57cec5SDimitry Andric "is\n" 25470b57cec5SDimitry Andric " * furnished to do so, subject to the following conditions:\n" 25480b57cec5SDimitry Andric " *\n" 25490b57cec5SDimitry Andric " * The above copyright notice and this permission notice shall be " 25500b57cec5SDimitry Andric "included in\n" 25510b57cec5SDimitry Andric " * all copies or substantial portions of the Software.\n" 25520b57cec5SDimitry Andric " *\n" 25530b57cec5SDimitry Andric " * THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, " 25540b57cec5SDimitry Andric "EXPRESS OR\n" 25550b57cec5SDimitry Andric " * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF " 25560b57cec5SDimitry Andric "MERCHANTABILITY,\n" 25570b57cec5SDimitry Andric " * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT " 25580b57cec5SDimitry Andric "SHALL THE\n" 25590b57cec5SDimitry Andric " * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR " 25600b57cec5SDimitry Andric "OTHER\n" 25610b57cec5SDimitry Andric " * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, " 25620b57cec5SDimitry Andric "ARISING FROM,\n" 25630b57cec5SDimitry Andric " * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER " 25640b57cec5SDimitry Andric "DEALINGS IN\n" 25650b57cec5SDimitry Andric " * THE SOFTWARE.\n" 25660b57cec5SDimitry Andric " *\n" 25670b57cec5SDimitry Andric " *===-----------------------------------------------------------------" 25680b57cec5SDimitry Andric "---" 25690b57cec5SDimitry Andric "---===\n" 25700b57cec5SDimitry Andric " */\n\n"; 25710b57cec5SDimitry Andric 25720b57cec5SDimitry Andric OS << "#ifndef __ARM_FP16_H\n"; 25730b57cec5SDimitry Andric OS << "#define __ARM_FP16_H\n\n"; 25740b57cec5SDimitry Andric 25750b57cec5SDimitry Andric OS << "#include <stdint.h>\n\n"; 25760b57cec5SDimitry Andric 25770b57cec5SDimitry Andric OS << "typedef __fp16 float16_t;\n"; 25780b57cec5SDimitry Andric 25790b57cec5SDimitry Andric OS << "#define __ai static __inline__ __attribute__((__always_inline__, " 25800b57cec5SDimitry Andric "__nodebug__))\n\n"; 25810b57cec5SDimitry Andric 25820b57cec5SDimitry Andric SmallVector<Intrinsic *, 128> Defs; 25830b57cec5SDimitry Andric std::vector<Record *> RV = Records.getAllDerivedDefinitions("Inst"); 25840b57cec5SDimitry Andric for (auto *R : RV) 25850b57cec5SDimitry Andric createIntrinsic(R, Defs); 25860b57cec5SDimitry Andric 25870b57cec5SDimitry Andric for (auto *I : Defs) 25880b57cec5SDimitry Andric I->indexBody(); 25890b57cec5SDimitry Andric 2590*a7dea167SDimitry Andric llvm::stable_sort(Defs, llvm::deref<std::less<>>()); 25910b57cec5SDimitry Andric 25920b57cec5SDimitry Andric // Only emit a def when its requirements have been met. 25930b57cec5SDimitry Andric // FIXME: This loop could be made faster, but it's fast enough for now. 25940b57cec5SDimitry Andric bool MadeProgress = true; 25950b57cec5SDimitry Andric std::string InGuard; 25960b57cec5SDimitry Andric while (!Defs.empty() && MadeProgress) { 25970b57cec5SDimitry Andric MadeProgress = false; 25980b57cec5SDimitry Andric 25990b57cec5SDimitry Andric for (SmallVector<Intrinsic *, 128>::iterator I = Defs.begin(); 26000b57cec5SDimitry Andric I != Defs.end(); /*No step*/) { 26010b57cec5SDimitry Andric bool DependenciesSatisfied = true; 26020b57cec5SDimitry Andric for (auto *II : (*I)->getDependencies()) { 26030b57cec5SDimitry Andric if (llvm::is_contained(Defs, II)) 26040b57cec5SDimitry Andric DependenciesSatisfied = false; 26050b57cec5SDimitry Andric } 26060b57cec5SDimitry Andric if (!DependenciesSatisfied) { 26070b57cec5SDimitry Andric // Try the next one. 26080b57cec5SDimitry Andric ++I; 26090b57cec5SDimitry Andric continue; 26100b57cec5SDimitry Andric } 26110b57cec5SDimitry Andric 26120b57cec5SDimitry Andric // Emit #endif/#if pair if needed. 26130b57cec5SDimitry Andric if ((*I)->getGuard() != InGuard) { 26140b57cec5SDimitry Andric if (!InGuard.empty()) 26150b57cec5SDimitry Andric OS << "#endif\n"; 26160b57cec5SDimitry Andric InGuard = (*I)->getGuard(); 26170b57cec5SDimitry Andric if (!InGuard.empty()) 26180b57cec5SDimitry Andric OS << "#if " << InGuard << "\n"; 26190b57cec5SDimitry Andric } 26200b57cec5SDimitry Andric 26210b57cec5SDimitry Andric // Actually generate the intrinsic code. 26220b57cec5SDimitry Andric OS << (*I)->generate(); 26230b57cec5SDimitry Andric 26240b57cec5SDimitry Andric MadeProgress = true; 26250b57cec5SDimitry Andric I = Defs.erase(I); 26260b57cec5SDimitry Andric } 26270b57cec5SDimitry Andric } 26280b57cec5SDimitry Andric assert(Defs.empty() && "Some requirements were not satisfied!"); 26290b57cec5SDimitry Andric if (!InGuard.empty()) 26300b57cec5SDimitry Andric OS << "#endif\n"; 26310b57cec5SDimitry Andric 26320b57cec5SDimitry Andric OS << "\n"; 26330b57cec5SDimitry Andric OS << "#undef __ai\n\n"; 26340b57cec5SDimitry Andric OS << "#endif /* __ARM_FP16_H */\n"; 26350b57cec5SDimitry Andric } 26360b57cec5SDimitry Andric 2637*a7dea167SDimitry Andric void clang::EmitNeon(RecordKeeper &Records, raw_ostream &OS) { 26380b57cec5SDimitry Andric NeonEmitter(Records).run(OS); 26390b57cec5SDimitry Andric } 26400b57cec5SDimitry Andric 2641*a7dea167SDimitry Andric void clang::EmitFP16(RecordKeeper &Records, raw_ostream &OS) { 26420b57cec5SDimitry Andric NeonEmitter(Records).runFP16(OS); 26430b57cec5SDimitry Andric } 26440b57cec5SDimitry Andric 2645*a7dea167SDimitry Andric void clang::EmitNeonSema(RecordKeeper &Records, raw_ostream &OS) { 26460b57cec5SDimitry Andric NeonEmitter(Records).runHeader(OS); 26470b57cec5SDimitry Andric } 26480b57cec5SDimitry Andric 2649*a7dea167SDimitry Andric void clang::EmitNeonTest(RecordKeeper &Records, raw_ostream &OS) { 26500b57cec5SDimitry Andric llvm_unreachable("Neon test generation no longer implemented!"); 26510b57cec5SDimitry Andric } 2652