xref: /freebsd/contrib/llvm-project/clang/utils/TableGen/NeonEmitter.cpp (revision a7dea1671b87c07d2d266f836bfa8b58efc7c134)
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