lib/SPIRVProducerPass.cpp

// Copyright 2017 The Clspv Authors. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#ifdef _MSC_VER
#pragma warning(push, 0)
#endif

#include <cassert>
#include <clspv/Passes.h>

#include <llvm/ADT/StringSwitch.h>
#include <llvm/ADT/UniqueVector.h>
#include <llvm/Analysis/LoopInfo.h>
#include <llvm/IR/Constants.h>
#include <llvm/IR/Dominators.h>
#include <llvm/IR/Instructions.h>
#include <llvm/IR/Metadata.h>
#include <llvm/IR/Module.h>
#include <llvm/Pass.h>
#include <llvm/Support/raw_ostream.h>
#include <llvm/Transforms/Utils/Cloning.h>

#include <spirv/1.0/spirv.hpp>
#include <clspv/AddressSpace.h>
#include <clspv/spirv_c_strings.hpp>
#include <clspv/spirv_glsl.hpp>

#include <list>
#include <iomanip>
#include <sstream>
#include <utility>

#if defined(_MSC_VER)
#pragma warning(pop)
#endif

using namespace llvm;
using namespace clspv;
using namespace mdconst;

namespace {
enum SPIRVOperandType {
  NUMBERID,
  LITERAL_INTEGER,
  LITERAL_STRING,
  LITERAL_FLOAT
};

struct SPIRVOperand {
  explicit SPIRVOperand(SPIRVOperandType Ty, uint32_t Num)
      : Type(Ty), LiteralNum(1, Num) {}
  explicit SPIRVOperand(SPIRVOperandType Ty, const char *Str)
      : Type(Ty), LiteralStr(Str) {}
  explicit SPIRVOperand(SPIRVOperandType Ty, StringRef Str)
      : Type(Ty), LiteralStr(Str) {}
  explicit SPIRVOperand(SPIRVOperandType Ty, ArrayRef<uint32_t> NumVec)
      : Type(Ty), LiteralNum(NumVec.begin(), NumVec.end()) {}

  SPIRVOperandType getType() { return Type; };
  uint32_t getNumID() { return LiteralNum[0]; };
  std::string getLiteralStr() { return LiteralStr; };
  ArrayRef<uint32_t> getLiteralNum() { return LiteralNum; };

private:
  SPIRVOperandType Type;
  std::string LiteralStr;
  SmallVector<uint32_t, 4> LiteralNum;
};

struct SPIRVInstruction {
  explicit SPIRVInstruction(uint16_t WCount, spv::Op Opc, uint32_t ResID,
                            ArrayRef<SPIRVOperand *> Ops)
      : WordCount(WCount), Opcode(static_cast<uint16_t>(Opc)), ResultID(ResID),
        Operands(Ops.begin(), Ops.end()) {}

  uint16_t getWordCount() const { return WordCount; }
  uint16_t getOpcode() const { return Opcode; }
  uint32_t getResultID() const { return ResultID; }
  ArrayRef<SPIRVOperand *> getOperands() const { return Operands; }

private:
  uint16_t WordCount;
  uint16_t Opcode;
  uint32_t ResultID;
  SmallVector<SPIRVOperand *, 4> Operands;
};

struct SPIRVProducerPass final : public ModulePass {
  typedef std::vector<SPIRVOperand *> SPIRVOperandList;
  typedef DenseMap<Type *, uint32_t> TypeMapType;
  typedef UniqueVector<Type *> TypeList;
  typedef DenseMap<Value *, uint32_t> ValueMapType;
  typedef std::vector<Value *> ValueList;
  typedef std::vector<std::pair<Value *, uint32_t>> EntryPointVecType;
  typedef std::list<SPIRVInstruction *> SPIRVInstructionList;
  typedef std::vector<
      std::tuple<Value *, SPIRVInstructionList::iterator, uint32_t>>
      DeferredInstVecType;
  typedef DenseMap<FunctionType *, std::pair<FunctionType *, uint32_t>>
      GlobalConstFuncMapType;

  explicit SPIRVProducerPass(
      raw_pwrite_stream &out, raw_ostream &descriptor_map_out,
      ArrayRef<std::pair<unsigned, std::string>> samplerMap, bool outputAsm,
      bool outputCInitList)
      : ModulePass(ID), samplerMap(samplerMap), out(out),
        binaryTempOut(binaryTempUnderlyingVector), binaryOut(&out),
        descriptorMapOut(descriptor_map_out), outputAsm(outputAsm),
        outputCInitList(outputCInitList), patchBoundOffset(0), nextID(1),
        OpExtInstImportID(0), HasVariablePointers(false), SamplerTy(nullptr) {}

  void getAnalysisUsage(AnalysisUsage &AU) const override {
    AU.addRequired<DominatorTreeWrapperPass>();
    AU.addRequired<LoopInfoWrapperPass>();
  }

  virtual bool runOnModule(Module &module) override;

  // output the SPIR-V header block
  void outputHeader();

  // patch the SPIR-V header block
  void patchHeader();

  uint32_t lookupType(Type *Ty) {
    if (Ty->isPointerTy() &&
        (Ty->getPointerAddressSpace() != AddressSpace::UniformConstant)) {
      auto PointeeTy = Ty->getPointerElementType();
      if (PointeeTy->isStructTy() &&
          dyn_cast<StructType>(PointeeTy)->isOpaque()) {
        Ty = PointeeTy;
      }
    }

    if (0 == TypeMap.count(Ty)) {
      Ty->print(errs());
      llvm_unreachable("Unhandled type!");
    }

    return TypeMap[Ty];
  }
  TypeMapType &getImageTypeMap() { return ImageTypeMap; }
  TypeList &getTypeList() { return Types; };
  ValueList &getConstantList() { return Constants; };
  ValueMapType &getValueMap() { return ValueMap; }
  ValueMapType &getAllocatedValueMap() { return AllocatedValueMap; }
  SPIRVInstructionList &getSPIRVInstList() { return SPIRVInsts; };
  ValueToValueMapTy &getArgumentGVMap() { return ArgumentGVMap; };
  ValueMapType &getArgumentGVIDMap() { return ArgumentGVIDMap; };
  EntryPointVecType &getEntryPointVec() { return EntryPointVec; };
  DeferredInstVecType &getDeferredInstVec() { return DeferredInstVec; };
  ValueList &getEntryPointInterfacesVec() { return EntryPointInterfacesVec; };
  uint32_t &getOpExtInstImportID() { return OpExtInstImportID; };
  std::vector<uint32_t> &getBuiltinDimVec() { return BuiltinDimensionVec; };
  bool hasVariablePointers() { return true; /* We use StorageBuffer everywhere */ };
  void setVariablePointers(bool Val) { HasVariablePointers = Val; };
  ArrayRef<std::pair<unsigned, std::string>> &getSamplerMap() { return samplerMap; }
  GlobalConstFuncMapType &getGlobalConstFuncTypeMap() {
    return GlobalConstFuncTypeMap;
  }
  SmallPtrSet<Value *, 16> &getGlobalConstArgSet() {
    return GlobalConstArgumentSet;
  }
  TypeList &getPointerTypesNeedingArrayStride() {
    return PointerTypesNeedingArrayStride;
  }

  void GenerateLLVMIRInfo(Module &M);
  bool FindExtInst(Module &M);
  void FindTypePerGlobalVar(GlobalVariable &GV);
  void FindTypePerFunc(Function &F);
  void FindType(Type *Ty);
  void FindConstantPerGlobalVar(GlobalVariable &GV);
  void FindConstantPerFunc(Function &F);
  void FindConstant(Value *V);
  void GenerateExtInstImport();
  void GenerateSPIRVTypes(const DataLayout &DL);
  void GenerateSPIRVConstants();
  void GenerateModuleInfo();
  void GenerateGlobalVar(GlobalVariable &GV);
  void GenerateSamplers(Module &M);
  void GenerateFuncPrologue(Function &F);
  void GenerateFuncBody(Function &F);
  void GenerateInstForArg(Function &F);
  spv::Op GetSPIRVCmpOpcode(CmpInst *CmpI);
  spv::Op GetSPIRVCastOpcode(Instruction &I);
  spv::Op GetSPIRVBinaryOpcode(Instruction &I);
  void GenerateInstruction(Instruction &I);
  void GenerateFuncEpilogue();
  void HandleDeferredInstruction();
  void HandleDeferredDecorations(const DataLayout& DL);
  bool is4xi8vec(Type *Ty) const;
  spv::StorageClass GetStorageClass(unsigned AddrSpace) const;
  spv::BuiltIn GetBuiltin(StringRef globalVarName) const;
  glsl::ExtInst getExtInstEnum(StringRef Name);
  void PrintResID(SPIRVInstruction *Inst);
  void PrintOpcode(SPIRVInstruction *Inst);
  void PrintOperand(SPIRVOperand *Op);
  void PrintCapability(SPIRVOperand *Op);
  void PrintExtInst(SPIRVOperand *Op);
  void PrintAddrModel(SPIRVOperand *Op);
  void PrintMemModel(SPIRVOperand *Op);
  void PrintExecModel(SPIRVOperand *Op);
  void PrintExecMode(SPIRVOperand *Op);
  void PrintSourceLanguage(SPIRVOperand *Op);
  void PrintFuncCtrl(SPIRVOperand *Op);
  void PrintStorageClass(SPIRVOperand *Op);
  void PrintDecoration(SPIRVOperand *Op);
  void PrintBuiltIn(SPIRVOperand *Op);
  void PrintSelectionControl(SPIRVOperand *Op);
  void PrintLoopControl(SPIRVOperand *Op);
  void PrintDimensionality(SPIRVOperand *Op);
  void PrintImageFormat(SPIRVOperand *Op);
  void PrintMemoryAccess(SPIRVOperand *Op);
  void PrintImageOperandsType(SPIRVOperand *Op);
  void WriteSPIRVAssembly();
  void WriteOneWord(uint32_t Word);
  void WriteResultID(SPIRVInstruction *Inst);
  void WriteWordCountAndOpcode(SPIRVInstruction *Inst);
  void WriteOperand(SPIRVOperand *Op);
  void WriteSPIRVBinary();

private:
  static char ID;
  ArrayRef<std::pair<unsigned, std::string>> samplerMap;
  raw_pwrite_stream &out;

  // TODO(dneto): Wouldn't it be better to always just emit a binary, and then
  // convert to other formats on demand?

  // When emitting a C initialization list, the WriteSPIRVBinary method
  // will actually write its words to this vector via binaryTempOut.
  SmallVector<char, 100> binaryTempUnderlyingVector;
  raw_svector_ostream binaryTempOut;

  // Binary output writes to this stream, which might be |out| or
  // |binaryTempOut|.  It's the latter when we really want to write a C
  // initializer list.
  raw_pwrite_stream* binaryOut;
  raw_ostream &descriptorMapOut;
  const bool outputAsm;
  const bool outputCInitList; // If true, output look like {0x7023, ... , 5}
  uint64_t patchBoundOffset;
  uint32_t nextID;

  TypeMapType TypeMap;
  TypeMapType ImageTypeMap;
  TypeList Types;
  ValueList Constants;
  ValueMapType ValueMap;
  ValueMapType AllocatedValueMap;
  SPIRVInstructionList SPIRVInsts;
  ValueToValueMapTy ArgumentGVMap;
  ValueMapType ArgumentGVIDMap;
  EntryPointVecType EntryPointVec;
  DeferredInstVecType DeferredInstVec;
  ValueList EntryPointInterfacesVec;
  uint32_t OpExtInstImportID;
  std::vector<uint32_t> BuiltinDimensionVec;
  bool HasVariablePointers;
  Type *SamplerTy;
  GlobalConstFuncMapType GlobalConstFuncTypeMap;
  SmallPtrSet<Value *, 16> GlobalConstArgumentSet;
  // An ordered set of pointer types of Base arguments to OpPtrAccessChain,
  // and which point into transparent memory (StorageBuffer storage class).
  // These will require an ArrayStride decoration.
  // See SPV_KHR_variable_pointers rev 13.
  TypeList PointerTypesNeedingArrayStride;
};

char SPIRVProducerPass::ID;
}

namespace clspv {
ModulePass *
createSPIRVProducerPass(raw_pwrite_stream &out, raw_ostream &descriptor_map_out,
                        ArrayRef<std::pair<unsigned, std::string>> samplerMap,
                        bool outputAsm, bool outputCInitList) {
  return new SPIRVProducerPass(out, descriptor_map_out, samplerMap, outputAsm,
                               outputCInitList);
}
} // namespace clspv

bool SPIRVProducerPass::runOnModule(Module &module) {
  binaryOut = outputCInitList ? &binaryTempOut : &out;

  // SPIR-V always begins with its header information
  outputHeader();

  // Gather information from the LLVM IR that we require.
  GenerateLLVMIRInfo(module);

  // If we are using a sampler map, find the type of the sampler.
  if (0 < getSamplerMap().size()) {
    auto SamplerStructTy = module.getTypeByName("opencl.sampler_t");
    if (!SamplerStructTy) {
      SamplerStructTy =
          StructType::create(module.getContext(), "opencl.sampler_t");
    }

    SamplerTy = SamplerStructTy->getPointerTo(AddressSpace::UniformConstant);

    FindType(SamplerTy);
  }

  // Collect information on global variables too.
  for (GlobalVariable &GV : module.globals()) {
    // If the GV is one of our special __spirv_* variables, remove the
    // initializer as it was only placed there to force LLVM to not throw the
    // value away.
    if (GV.getName().startswith("__spirv_")) {
      GV.setInitializer(nullptr);
    }

    // Collect types' information from global variable.
    FindTypePerGlobalVar(GV);

    // Collect constant information from global variable.
    FindConstantPerGlobalVar(GV);

    // If the variable is an input, entry points need to know about it.
    if (AddressSpace::Input == GV.getType()->getPointerAddressSpace()) {
      getEntryPointInterfacesVec().push_back(&GV);
    }
  }

  // If there are extended instructions, generate OpExtInstImport.
  if (FindExtInst(module)) {
    GenerateExtInstImport();
  }

  // Generate SPIRV instructions for types.
  const DataLayout &DL = module.getDataLayout();
  GenerateSPIRVTypes(DL);

  // Generate SPIRV constants.
  GenerateSPIRVConstants();

  // If we have a sampler map, we might have literal samplers to generate.
  if (0 < getSamplerMap().size()) {
    GenerateSamplers(module);
  }

  // Generate SPIRV variables.
  for (GlobalVariable &GV : module.globals()) {
    GenerateGlobalVar(GV);
  }

  // Generate SPIRV instructions for each function.
  for (Function &F : module) {
    if (F.isDeclaration()) {
      continue;
    }

    // Generate Function Prologue.
    GenerateFuncPrologue(F);

    // Generate SPIRV instructions for function body.
    GenerateFuncBody(F);

    // Generate Function Epilogue.
    GenerateFuncEpilogue();
  }

  HandleDeferredInstruction();
  HandleDeferredDecorations(DL);

  // Generate SPIRV module information.
  GenerateModuleInfo();

  if (outputAsm) {
    WriteSPIRVAssembly();
  } else {
    WriteSPIRVBinary();
  }

  // We need to patch the SPIR-V header to set bound correctly.
  patchHeader();

  if (outputCInitList) {
    bool first = true;
    int word_index = 0;
    std::ostringstream os;

    os << std::hex;
    auto emit_word = [&os, &word_index, &first](uint32_t word) {
      if (!first)
        os << ',';
      if (word_index > 0)
        os << ((word_index & 3) ? ' ' : '\n');
      os << "0x" << word;
      ++word_index;
      first = false;
    };

    os << "{";
    const StringRef str = binaryTempOut.str();
    for (unsigned i = 0; i < str.size() / 4; i++) {
      emit_word(uint32_t(str[4 * i]) | (uint32_t(str[4 * i + 1]) << 8) |
                (uint32_t(str[4 * i + 2]) << 16) |
                (uint32_t(str[4 * i + 3]) << 24));
    }
    os << "}\n";
    out << os.str();
  }

  return false;
}

void SPIRVProducerPass::outputHeader() {
  if (outputAsm) {
    // for ASM output the header goes into 5 comments at the beginning of the
    // file
    out << "; SPIR-V\n";

    // the major version number is in the 2nd highest byte
    const uint32_t major = (spv::Version >> 16) & 0xFF;

    // the minor version number is in the 2nd lowest byte
    const uint32_t minor = (spv::Version >> 8) & 0xFF;
    out << "; Version: " << major << "." << minor << "\n";

    // use Codeplay's vendor ID
    out << "; Generator: Codeplay; 0\n";

    out << "; Bound: ";

    // we record where we need to come back to and patch in the bound value
    patchBoundOffset = out.tell();

    // output one space per digit for the max size of a 32 bit unsigned integer
    // (which is the maximum ID we could possibly be using)
    for (uint32_t i = std::numeric_limits<uint32_t>::max(); 0 != i; i /= 10) {
      out << " ";
    }

    out << "\n";

    out << "; Schema: 0\n";
  } else {
    binaryOut->write(reinterpret_cast<const char *>(&spv::MagicNumber),
              sizeof(spv::MagicNumber));
    binaryOut->write(reinterpret_cast<const char *>(&spv::Version),
              sizeof(spv::Version));

    // use Codeplay's vendor ID
    const uint32_t vendor = 3 << 16;
    binaryOut->write(reinterpret_cast<const char *>(&vendor), sizeof(vendor));

    // we record where we need to come back to and patch in the bound value
    patchBoundOffset = binaryOut->tell();

    // output a bad bound for now
    binaryOut->write(reinterpret_cast<const char *>(&nextID), sizeof(nextID));

    // output the schema (reserved for use and must be 0)
    const uint32_t schema = 0;
    binaryOut->write(reinterpret_cast<const char *>(&schema), sizeof(schema));
  }
}

void SPIRVProducerPass::patchHeader() {
  if (outputAsm) {
    // get the string representation of the max bound used (nextID will be the
    // max ID used)
    auto asString = std::to_string(nextID);
    out.pwrite(asString.c_str(), asString.size(), patchBoundOffset);
  } else {
    // for a binary we just write the value of nextID over bound
    binaryOut->pwrite(reinterpret_cast<char *>(&nextID), sizeof(nextID),
                      patchBoundOffset);
  }
}

void SPIRVProducerPass::GenerateLLVMIRInfo(Module &M) {
  // This function generates LLVM IR for function such as global variable for
  // argument, constant and pointer type for argument access. These information
  // is artificial one because we need Vulkan SPIR-V output. This function is
  // executed ahead of FindType and FindConstant.
  ValueToValueMapTy &ArgGVMap = getArgumentGVMap();
  LLVMContext &Context = M.getContext();

  // Map for avoiding to generate struct type with same fields.
  DenseMap<Type *, Type *> ArgTyMap;

  // Collect global constant variables.
  SmallVector<GlobalVariable *, 8> GVList;
  for (GlobalVariable &GV : M.globals()) {
    if (GV.getType()->getAddressSpace() == AddressSpace::Constant) {
      GVList.push_back(&GV);
    }
  }

  // Change global constant variable's address space to ModuleScopePrivate.
  auto &GlobalConstFuncTyMap = getGlobalConstFuncTypeMap();
  for (auto GV : GVList) {
    // If there is no user of gv, delete gv.
    if (GV->use_empty()) {
      GV->eraseFromParent();
      continue;
    }

    // Create new gv with ModuleScopePrivate address space.
    Type *NewGVTy = GV->getType()->getPointerElementType();
    GlobalVariable *NewGV = new GlobalVariable(
        M, NewGVTy, false, GV->getLinkage(), GV->getInitializer(), "", nullptr,
        GV->getThreadLocalMode(), AddressSpace::ModuleScopePrivate);
    NewGV->takeName(GV);

    const SmallVector<User *, 8> GVUsers(GV->user_begin(), GV->user_end());
    SmallVector<User*, 8> CandidateUsers;

    for (User *GVU : GVUsers) {
      if (CallInst *Call = dyn_cast<CallInst>(GVU)) {
        // Find argument index.
        unsigned GVCstArgIdx = 0;
        for (unsigned i = 0; i < Call->getNumArgOperands(); i++) {
          if (GV == Call->getOperand(i)) {
            GVCstArgIdx = i;
          }
        }

        // Record function with global constant.
        GlobalConstFuncTyMap[Call->getFunctionType()] =
            std::make_pair(Call->getFunctionType(), GVCstArgIdx);
      } else if (GetElementPtrInst *GEP = dyn_cast<GetElementPtrInst>(GVU)) {
        // Check GEP users.
        for (User *GEPU : GEP->users()) {
          if (CallInst *GEPCall = dyn_cast<CallInst>(GEPU)) {
            // Find argument index.
            unsigned GVCstArgIdx = 0;
            for (unsigned i = 0; i < GEPCall->getNumArgOperands(); i++) {
              if (GEP == GEPCall->getOperand(i)) {
                GVCstArgIdx = i;
              }
            }

            // Record function with global constant.
            GlobalConstFuncTyMap[GEPCall->getFunctionType()] =
                std::make_pair(GEPCall->getFunctionType(), GVCstArgIdx);
          }
        }
      }

      CandidateUsers.push_back(GVU);
    }

    for (User *U : CandidateUsers) {
      // Update users of gv with new gv.
      U->replaceUsesOfWith(GV, NewGV);
    }

    // Delete original gv.
    GV->eraseFromParent();
  }

  bool HasWorkGroupBuiltin = false;
  for (GlobalVariable &GV : M.globals()) {
    const spv::BuiltIn BuiltinType = GetBuiltin(GV.getName());
    if (spv::BuiltInWorkgroupSize == BuiltinType) {
      HasWorkGroupBuiltin = true;
    }
  }


  for (Function &F : M) {
    // Handle kernel function first.
    if (F.isDeclaration() || F.getCallingConv() != CallingConv::SPIR_KERNEL) {
      continue;
    }

    for (BasicBlock &BB : F) {
      for (Instruction &I : BB) {
        if (I.getOpcode() == Instruction::ZExt ||
            I.getOpcode() == Instruction::SExt ||
            I.getOpcode() == Instruction::UIToFP) {
          // If there is zext with i1 type, it will be changed to OpSelect. The
          // OpSelect needs constant 0 and 1 so the constants are added here.

          auto OpTy = I.getOperand(0)->getType();

          if (OpTy->isIntegerTy(1) ||
              (OpTy->isVectorTy() &&
               OpTy->getVectorElementType()->isIntegerTy(1))) {
            if (I.getOpcode() == Instruction::ZExt) {
              APInt One(32, 1);
              FindConstant(Constant::getNullValue(I.getType()));
              FindConstant(Constant::getIntegerValue(I.getType(), One));
            } else if (I.getOpcode() == Instruction::SExt) {
              APInt MinusOne(32, UINT64_MAX, true);
              FindConstant(Constant::getNullValue(I.getType()));
              FindConstant(Constant::getIntegerValue(I.getType(), MinusOne));
            } else {
              FindConstant(ConstantFP::get(Context, APFloat(0.0f)));
              FindConstant(ConstantFP::get(Context, APFloat(1.0f)));
            }
          }
        } else if (CallInst *Call = dyn_cast<CallInst>(&I)) {
          Function *Callee = Call->getCalledFunction();

          // Handle image type specially.
          if (Callee->getName().equals(
                  "_Z11read_imagef14ocl_image2d_ro11ocl_samplerDv2_f") ||
              Callee->getName().equals(
                  "_Z11read_imagef14ocl_image3d_ro11ocl_samplerDv4_f")) {
            TypeMapType &OpImageTypeMap = getImageTypeMap();
            Type *ImageTy =
                Call->getArgOperand(0)->getType()->getPointerElementType();
            OpImageTypeMap[ImageTy] = 0;

            FindConstant(ConstantFP::get(Context, APFloat(0.0f)));
          }
        }
      }
    }

    if (M.getTypeByName("opencl.image2d_ro_t") ||
        M.getTypeByName("opencl.image2d_wo_t") ||
        M.getTypeByName("opencl.image3d_ro_t") ||
        M.getTypeByName("opencl.image3d_wo_t")) {
      // Assume Image type's sampled type is float type.
      FindType(Type::getFloatTy(Context));
    }

    if (const MDNode *MD =
            dyn_cast<Function>(&F)->getMetadata("reqd_work_group_size")) {
      // We generate constants if the WorkgroupSize builtin is being used.
      if (HasWorkGroupBuiltin) {
        // Collect constant information for work group size.
        FindConstant(mdconst::extract<ConstantInt>(MD->getOperand(0)));
        FindConstant(mdconst::extract<ConstantInt>(MD->getOperand(1)));
        FindConstant(mdconst::extract<ConstantInt>(MD->getOperand(2)));
      }
    }

    // Wrap up all argument types with struct type and create global variables
    // with them.
    bool HasArgUser = false;
    unsigned Idx = 0;

    for (const Argument &Arg : F.args()) {
      Type *ArgTy = Arg.getType();
      Type *GVTy = nullptr;

      // Check argument type whether it is pointer type or not. If it is
      // pointer type, add its address space to new global variable for
      // argument.
      unsigned AddrSpace = AddressSpace::Global;
      if (PointerType *ArgPTy = dyn_cast<PointerType>(ArgTy)) {
        AddrSpace = ArgPTy->getAddressSpace();
      }

      Type *TmpArgTy = ArgTy;

      // sampler_t and image types have pointer type of struct type with
      // opaque
      // type as field. Extract the struct type. It will be used by global
      // variable for argument.
      bool IsSamplerType = false;
      bool IsImageType = false;
      if (PointerType *TmpArgPTy = dyn_cast<PointerType>(TmpArgTy)) {
        if (StructType *STy =
                dyn_cast<StructType>(TmpArgPTy->getElementType())) {
          if (STy->isOpaque()) {
            if (STy->getName().equals("opencl.sampler_t")) {
              AddrSpace = AddressSpace::UniformConstant;
              IsSamplerType = true;
              TmpArgTy = STy;
            } else if (STy->getName().equals("opencl.image2d_ro_t") ||
                       STy->getName().equals("opencl.image2d_wo_t") ||
                       STy->getName().equals("opencl.image3d_ro_t") ||
                       STy->getName().equals("opencl.image3d_wo_t")) {
              AddrSpace = AddressSpace::UniformConstant;
              IsImageType = true;
              TmpArgTy = STy;
            } else {
              llvm_unreachable("Argument has opaque type unsupported???");
            }
          }
        }
      }

      // LLVM's pointer type is distinguished by address space but we need to
      // regard constant and global address space as same here. If pointer
      // type has constant address space, generate new pointer type
      // temporarily to check previous struct type for argument.
      if (PointerType *TmpArgPTy = dyn_cast<PointerType>(TmpArgTy)) {
        AddrSpace = TmpArgPTy->getAddressSpace();
        if (AddrSpace == AddressSpace::Constant) {
          TmpArgTy = PointerType::get(TmpArgPTy->getElementType(),
                                      AddressSpace::Global);
        }
      }

      if (IsSamplerType || IsImageType) {
        GVTy = TmpArgTy;
      } else if (ArgTyMap.count(TmpArgTy)) {
        // If there are arguments handled previously, use its type.
        GVTy = ArgTyMap[TmpArgTy];
      } else {
        // Wrap up argument type with struct type.
        StructType *STy = StructType::create(Context);

        SmallVector<Type *, 8> EltTys;
        EltTys.push_back(ArgTy);

        STy->setBody(EltTys, false);

        GVTy = STy;
        ArgTyMap[TmpArgTy] = STy;
      }

      // In order to build type map between llvm type and spirv id, LLVM
      // global variable is needed. It has llvm type and other instructions
      // can access it with its type.
      GlobalVariable *NewGV = new GlobalVariable(
          M, GVTy, false, GlobalValue::ExternalLinkage, UndefValue::get(GVTy),
          F.getName() + ".arg." + std::to_string(Idx++), nullptr,
          GlobalValue::ThreadLocalMode::NotThreadLocal, AddrSpace);

      // Generate type info for argument global variable.
      FindType(NewGV->getType());

      ArgGVMap[&Arg] = NewGV;

      // Generate pointer type of argument type for OpAccessChain of argument.
      if (!Arg.use_empty()) {
        if (!isa<PointerType>(ArgTy)) {
          FindType(PointerType::get(ArgTy, AddrSpace));
        }
        HasArgUser = true;
      }
    }

    if (HasArgUser) {
      // Generate constant 0 for OpAccessChain of argument.
      Type *IdxTy = Type::getInt32Ty(Context);
      FindConstant(ConstantInt::get(IdxTy, 0));
      FindType(IdxTy);
    }

    // Collect types' information from function.
    FindTypePerFunc(F);

    // Collect constant information from function.
    FindConstantPerFunc(F);
  }

  for (Function &F : M) {
    // Handle non-kernel functions.
    if (F.isDeclaration() || F.getCallingConv() == CallingConv::SPIR_KERNEL) {
      continue;
    }

    for (BasicBlock &BB : F) {
      for (Instruction &I : BB) {
        if (I.getOpcode() == Instruction::ZExt ||
            I.getOpcode() == Instruction::SExt ||
            I.getOpcode() == Instruction::UIToFP) {
          // If there is zext with i1 type, it will be changed to OpSelect. The
          // OpSelect needs constant 0 and 1 so the constants are added here.

          auto OpTy = I.getOperand(0)->getType();

          if (OpTy->isIntegerTy(1) ||
              (OpTy->isVectorTy() &&
               OpTy->getVectorElementType()->isIntegerTy(1))) {
            if (I.getOpcode() == Instruction::ZExt) {
              APInt One(32, 1);
              FindConstant(Constant::getNullValue(I.getType()));
              FindConstant(Constant::getIntegerValue(I.getType(), One));
            } else if (I.getOpcode() == Instruction::SExt) {
              APInt MinusOne(32, UINT64_MAX, true);
              FindConstant(Constant::getNullValue(I.getType()));
              FindConstant(Constant::getIntegerValue(I.getType(), MinusOne));
            } else {
              FindConstant(ConstantFP::get(Context, APFloat(0.0f)));
              FindConstant(ConstantFP::get(Context, APFloat(1.0f)));
            }
          }
        } else if (CallInst *Call = dyn_cast<CallInst>(&I)) {
          Function *Callee = Call->getCalledFunction();

          // Handle image type specially.
          if (Callee->getName().equals(
                  "_Z11read_imagef14ocl_image2d_ro11ocl_samplerDv2_f") ||
              Callee->getName().equals(
                  "_Z11read_imagef14ocl_image3d_ro11ocl_samplerDv4_f")) {
            TypeMapType &OpImageTypeMap = getImageTypeMap();
            Type *ImageTy =
                Call->getArgOperand(0)->getType()->getPointerElementType();
            OpImageTypeMap[ImageTy] = 0;

            FindConstant(ConstantFP::get(Context, APFloat(0.0f)));
          }
        }
      }
    }

    if (M.getTypeByName("opencl.image2d_ro_t") ||
        M.getTypeByName("opencl.image2d_wo_t") ||
        M.getTypeByName("opencl.image3d_ro_t") ||
        M.getTypeByName("opencl.image3d_wo_t")) {
      // Assume Image type's sampled type is float type.
      FindType(Type::getFloatTy(Context));
    }

    // Collect types' information from function.
    FindTypePerFunc(F);

    // Collect constant information from function.
    FindConstantPerFunc(F);
  }
}

bool SPIRVProducerPass::FindExtInst(Module &M) {
  LLVMContext &Context = M.getContext();
  bool HasExtInst = false;

  for (Function &F : M) {
    for (BasicBlock &BB : F) {
      for (Instruction &I : BB) {
        if (CallInst *Call = dyn_cast<CallInst>(&I)) {
          Function *Callee = Call->getCalledFunction();
          // Check whether this call is for extend instructions.
          glsl::ExtInst EInst = getExtInstEnum(Callee->getName());
          if (EInst) {
            // clz needs OpExtInst and OpISub with constant 31. Add constant 31
            // to constant list here.
            if (EInst == glsl::ExtInstFindUMsb) {
              Type *IdxTy = Type::getInt32Ty(Context);
              FindConstant(ConstantInt::get(IdxTy, 31));
              FindType(IdxTy);
            }

            HasExtInst = true;
          }
        }
      }
    }
  }

  return HasExtInst;
}

void SPIRVProducerPass::FindTypePerGlobalVar(GlobalVariable &GV) {
  // Investigate global variable's type.
  FindType(GV.getType());
}

void SPIRVProducerPass::FindTypePerFunc(Function &F) {
  // Investigate function's type.
  FunctionType *FTy = F.getFunctionType();

  if (F.getCallingConv() != CallingConv::SPIR_KERNEL) {
    auto &GlobalConstFuncTyMap = getGlobalConstFuncTypeMap();
    // Handle function with global constant parameters.
    if (GlobalConstFuncTyMap.count(FTy)) {
      uint32_t GVCstArgIdx = GlobalConstFuncTypeMap[FTy].second;
      SmallVector<Type *, 4> NewFuncParamTys;
      for (unsigned i = 0; i < FTy->getNumParams(); i++) {
        Type *ParamTy = FTy->getParamType(i);
        if (i == GVCstArgIdx) {
          Type *EleTy = ParamTy->getPointerElementType();
          ParamTy = PointerType::get(EleTy, AddressSpace::ModuleScopePrivate);
        }

        NewFuncParamTys.push_back(ParamTy);
      }

      FunctionType *NewFTy =
          FunctionType::get(FTy->getReturnType(), NewFuncParamTys, false);
      GlobalConstFuncTyMap[FTy] = std::make_pair(NewFTy, GVCstArgIdx);
      FTy = NewFTy;
    }

    FindType(FTy);
  } else {
    // As kernel functions do not have parameters, create new function type and
    // add it to type map.
    SmallVector<Type *, 4> NewFuncParamTys;
    FunctionType *NewFTy =
        FunctionType::get(FTy->getReturnType(), NewFuncParamTys, false);
    FindType(NewFTy);
  }

  // Investigate instructions' type in function body.
  for (BasicBlock &BB : F) {
    for (Instruction &I : BB) {
      if (isa<ShuffleVectorInst>(I)) {
        for (unsigned i = 0; i < I.getNumOperands(); i++) {
          // Ignore type for mask of shuffle vector instruction.
          if (i == 2) {
            continue;
          }

          Value *Op = I.getOperand(i);
          if (!isa<MetadataAsValue>(Op)) {
            FindType(Op->getType());
          }
        }

        FindType(I.getType());
        continue;
      }

      // Work through the operands of the instruction.
      for (unsigned i = 0; i < I.getNumOperands(); i++) {
        Value *const Op = I.getOperand(i);
        // If any of the operands is a constant, find the type!
        if (isa<Constant>(Op) && !isa<GlobalValue>(Op)) {
          FindType(Op->getType());
        }
      }

      for (Use &Op : I.operands()) {
        if (CallInst *Call = dyn_cast<CallInst>(&I)) {
          // Avoid to check call instruction's type.
          break;
        }
        if (!isa<MetadataAsValue>(&Op)) {
          FindType(Op->getType());
          continue;
        }
      }

      CallInst *Call = dyn_cast<CallInst>(&I);

      // We don't want to track the type of this call as we are going to replace
      // it.
      if (Call && ("__translate_sampler_initializer" ==
                   Call->getCalledFunction()->getName())) {
        continue;
      }

      if (GetElementPtrInst *GEP = dyn_cast<GetElementPtrInst>(&I)) {
        // If gep's base operand has ModuleScopePrivate address space, make gep
        // return ModuleScopePrivate address space.
        if (GEP->getPointerAddressSpace() == AddressSpace::ModuleScopePrivate) {
          // Add pointer type with private address space for global constant to
          // type list.
          Type *EleTy = I.getType()->getPointerElementType();
          Type *NewPTy =
              PointerType::get(EleTy, AddressSpace::ModuleScopePrivate);

          FindType(NewPTy);
          continue;
        }
      }

      FindType(I.getType());
    }
  }
}

void SPIRVProducerPass::FindType(Type *Ty) {
  TypeList &TyList = getTypeList();

  if (0 != TyList.idFor(Ty)) {
    return;
  }

  if (Ty->isPointerTy()) {
    auto AddrSpace = Ty->getPointerAddressSpace();
    if ((AddressSpace::Constant == AddrSpace) ||
        (AddressSpace::Global == AddrSpace)) {
      auto PointeeTy = Ty->getPointerElementType();

      if (PointeeTy->isStructTy() &&
          dyn_cast<StructType>(PointeeTy)->isOpaque()) {
        FindType(PointeeTy);
        auto ActualPointerTy =
            PointeeTy->getPointerTo(AddressSpace::UniformConstant);
        FindType(ActualPointerTy);
        return;
      }
    }
  }

  // OpTypeArray has constant and we need to support type of the constant.
  if (isa<ArrayType>(Ty)) {
    LLVMContext &Context = Ty->getContext();
    FindType(Type::getInt32Ty(Context));
  }

  for (Type *SubTy : Ty->subtypes()) {
    FindType(SubTy);
  }

  TyList.insert(Ty);
}

void SPIRVProducerPass::FindConstantPerGlobalVar(GlobalVariable &GV) {
  // If the global variable has a (non undef) initializer.
  if (GV.hasInitializer() && !isa<UndefValue>(GV.getInitializer())) {
    FindConstant(GV.getInitializer());
  }
}

void SPIRVProducerPass::FindConstantPerFunc(Function &F) {
  // Investigate constants in function body.
  for (BasicBlock &BB : F) {
    for (Instruction &I : BB) {
      CallInst *Call = dyn_cast<CallInst>(&I);

      if (Call && ("__translate_sampler_initializer" ==
                   Call->getCalledFunction()->getName())) {
        // We've handled these constants elsewhere, so skip it.
        continue;
      }

      if (isa<AllocaInst>(I)) {
        // Alloca instruction has constant for the number of element. Ignore it.
        continue;
      } else if (isa<ShuffleVectorInst>(I)) {
        for (unsigned i = 0; i < I.getNumOperands(); i++) {
          // Ignore constant for mask of shuffle vector instruction.
          if (i == 2) {
            continue;
          }

          if (isa<Constant>(I.getOperand(i)) &&
              !isa<GlobalValue>(I.getOperand(i))) {
            FindConstant(I.getOperand(i));
          }
        }

        continue;
      } else if (isa<InsertElementInst>(I)) {
        // Handle InsertElement with <4 x i8> specially.
        Type *CompositeTy = I.getOperand(0)->getType();
        if (is4xi8vec(CompositeTy)) {
          LLVMContext &Context = CompositeTy->getContext();
          if (isa<Constant>(I.getOperand(0))) {
            FindConstant(I.getOperand(0));
          }

          if (isa<Constant>(I.getOperand(1))) {
            FindConstant(I.getOperand(1));
          }

          // Add mask constant 0xFF.
          Constant *CstFF = ConstantInt::get(Type::getInt32Ty(Context), 0xFF);
          FindConstant(CstFF);

          // Add shift amount constant.
          if (ConstantInt *CI = dyn_cast<ConstantInt>(I.getOperand(2))) {
            uint64_t Idx = CI->getZExtValue();
            Constant *CstShiftAmount =
                ConstantInt::get(Type::getInt32Ty(Context), Idx * 8);
            FindConstant(CstShiftAmount);
          }

          continue;
        }

        for (unsigned i = 0; i < I.getNumOperands(); i++) {
          // Ignore constant for index of InsertElement instruction.
          if (i == 2) {
            continue;
          }

          if (isa<Constant>(I.getOperand(i)) &&
              !isa<GlobalValue>(I.getOperand(i))) {
            FindConstant(I.getOperand(i));
          }
        }

        continue;
      } else if (isa<ExtractElementInst>(I)) {
        // Handle ExtractElement with <4 x i8> specially.
        Type *CompositeTy = I.getOperand(0)->getType();
        if (is4xi8vec(CompositeTy)) {
          LLVMContext &Context = CompositeTy->getContext();
          if (isa<Constant>(I.getOperand(0))) {
            FindConstant(I.getOperand(0));
          }

          // Add mask constant 0xFF.
          Constant *CstFF = ConstantInt::get(Type::getInt32Ty(Context), 0xFF);
          FindConstant(CstFF);

          // Add shift amount constant.
          if (ConstantInt *CI = dyn_cast<ConstantInt>(I.getOperand(1))) {
            uint64_t Idx = CI->getZExtValue();
            Constant *CstShiftAmount =
                ConstantInt::get(Type::getInt32Ty(Context), Idx * 8);
            FindConstant(CstShiftAmount);
          } else {
            ConstantInt *Cst8 = ConstantInt::get(Type::getInt32Ty(Context), 8);
            FindConstant(Cst8);
          }

          continue;
        }

        for (unsigned i = 0; i < I.getNumOperands(); i++) {
          // Ignore constant for index of ExtractElement instruction.
          if (i == 1) {
            continue;
          }

          if (isa<Constant>(I.getOperand(i)) &&
              !isa<GlobalValue>(I.getOperand(i))) {
            FindConstant(I.getOperand(i));
          }
        }

        continue;
      } else if ((Instruction::Xor == I.getOpcode()) && I.getType()->isIntegerTy(1)) {
        // We special case for Xor where the type is i1 and one of the arguments is a constant 1 (true), this is an OpLogicalNot in SPIR-V, and we don't need the constant
        bool foundConstantTrue = false;
        for (Use &Op : I.operands()) {
          if (isa<Constant>(Op) && !isa<GlobalValue>(Op)) {
            auto CI = cast<ConstantInt>(Op);

            if (CI->isZero() || foundConstantTrue) {
              // If we already found the true constant, we might (probably only on -O0) have an OpLogicalNot which is taking a constant argument, so discover it anyway.
              FindConstant(Op);
            } else {
              foundConstantTrue = true;
            }
          }
        }

        continue;
      }

      for (Use &Op : I.operands()) {
        if (isa<Constant>(Op) && !isa<GlobalValue>(Op)) {
          FindConstant(Op);
        }
      }
    }
  }
}

void SPIRVProducerPass::FindConstant(Value *V) {
  ValueMapType &VMap = getValueMap();
  ValueList &CstList = getConstantList();

  // If V is already in VMap, ignore it.
  if (VMap.find_as(V) != VMap.end()) {
    return;
  }

  Constant *Cst = cast<Constant>(V);

  // Handle constant with <4 x i8> type specially.
  Type *CstTy = Cst->getType();
  if (is4xi8vec(CstTy)) {
    if (!isa<GlobalValue>(V)) {
      CstList.push_back(V);
      VMap[V] = static_cast<uint32_t>(CstList.size());
    }
  }

  if (Cst->getNumOperands()) {
    for (User::const_op_iterator I = Cst->op_begin(), E = Cst->op_end(); I != E;
         ++I) {
      FindConstant(*I);
    }

    CstList.push_back(Cst);
    VMap[V] = static_cast<uint32_t>(CstList.size());
    return;
  } else if (const ConstantDataSequential *CDS =
                 dyn_cast<ConstantDataSequential>(Cst)) {
    // Add constants for each element to constant list.
    for (unsigned i = 0; i < CDS->getNumElements(); i++) {
      Constant *EleCst = CDS->getElementAsConstant(i);
      FindConstant(EleCst);
    }
  }

  if (!isa<GlobalValue>(V)) {
    CstList.push_back(V);
    VMap[V] = static_cast<uint32_t>(CstList.size());
  }
}

spv::StorageClass SPIRVProducerPass::GetStorageClass(unsigned AddrSpace) const {
  switch (AddrSpace) {
  default:
    llvm_unreachable("Unsupported OpenCL address space");
  case AddressSpace::Private:
    return spv::StorageClassFunction;
  case AddressSpace::Global:
  case AddressSpace::Constant:
    return spv::StorageClassStorageBuffer;
  case AddressSpace::Input:
    return spv::StorageClassInput;
  case AddressSpace::Local:
    return spv::StorageClassWorkgroup;
  case AddressSpace::UniformConstant:
    return spv::StorageClassUniformConstant;
  case AddressSpace::ModuleScopePrivate:
    return spv::StorageClassPrivate;
  }
}

spv::BuiltIn SPIRVProducerPass::GetBuiltin(StringRef Name) const {
  return StringSwitch<spv::BuiltIn>(Name)
      .Case("__spirv_GlobalInvocationId", spv::BuiltInGlobalInvocationId)
      .Case("__spirv_LocalInvocationId", spv::BuiltInLocalInvocationId)
      .Case("__spirv_WorkgroupSize", spv::BuiltInWorkgroupSize)
      .Case("__spirv_NumWorkgroups", spv::BuiltInNumWorkgroups)
      .Case("__spirv_WorkgroupId", spv::BuiltInWorkgroupId)
      .Default(spv::BuiltInMax);
}

void SPIRVProducerPass::GenerateExtInstImport() {
  SPIRVInstructionList &SPIRVInstList = getSPIRVInstList();
  uint32_t &ExtInstImportID = getOpExtInstImportID();

  //
  // Generate OpExtInstImport.
  //
  // Ops[0] ... Ops[n] = Name (Literal String)
  SPIRVOperandList Ops;

  SPIRVOperand *Name =
      new SPIRVOperand(SPIRVOperandType::LITERAL_STRING, "GLSL.std.450");
  Ops.push_back(Name);

  size_t NameWordSize = (Name->getLiteralStr().size() + 1) / 4;
  assert(NameWordSize < (UINT16_MAX - 2));
  if ((Name->getLiteralStr().size() + 1) % 4) {
    NameWordSize += 1;
  }

  uint16_t WordCount = static_cast<uint16_t>(2 + NameWordSize);
  ExtInstImportID = nextID;

  SPIRVInstruction *Inst =
      new SPIRVInstruction(WordCount, spv::OpExtInstImport, nextID++, Ops);
  SPIRVInstList.push_back(Inst);
}

void SPIRVProducerPass::GenerateSPIRVTypes(const DataLayout &DL) {
  SPIRVInstructionList &SPIRVInstList = getSPIRVInstList();
  ValueMapType &VMap = getValueMap();
  ValueMapType &AllocatedVMap = getAllocatedValueMap();
  ValueToValueMapTy &ArgGVMap = getArgumentGVMap();

  // Map for OpTypeRuntimeArray. If argument has pointer type, 2 spirv type
  // instructions are generated. They are OpTypePointer and OpTypeRuntimeArray.
  DenseMap<Type *, uint32_t> OpRuntimeTyMap;

  for (Type *Ty : getTypeList()) {
    // Update TypeMap with nextID for reference later.
    TypeMap[Ty] = nextID;

    switch (Ty->getTypeID()) {
    default: {
      Ty->print(errs());
      llvm_unreachable("Unsupported type???");
      break;
    }
    case Type::MetadataTyID:
    case Type::LabelTyID: {
      // Ignore these types.
      break;
    }
    case Type::PointerTyID: {
      PointerType *PTy = cast<PointerType>(Ty);
      unsigned AddrSpace = PTy->getAddressSpace();

      // For the purposes of our Vulkan SPIR-V type system, constant and global
      // are conflated.
      bool UseExistingOpTypePointer = false;
      if (AddressSpace::Constant == AddrSpace) {
        AddrSpace = AddressSpace::Global;

        // Check to see if we already created this type (for instance, if we had
        // a constant <type>* and a global <type>*, the type would be created by
        // one of these types, and shared by both).
        auto GlobalTy = PTy->getPointerElementType()->getPointerTo(AddrSpace);
        if (0 < TypeMap.count(GlobalTy)) {
          TypeMap[PTy] = TypeMap[GlobalTy];
          break;
        }
      } else if (AddressSpace::Global == AddrSpace) {
        AddrSpace = AddressSpace::Constant;

        // Check to see if we already created this type (for instance, if we had
        // a constant <type>* and a global <type>*, the type would be created by
        // one of these types, and shared by both).
        auto ConstantTy = PTy->getPointerElementType()->getPointerTo(AddrSpace);
        if (0 < TypeMap.count(ConstantTy)) {
          TypeMap[PTy] = TypeMap[ConstantTy];
          UseExistingOpTypePointer = true;
        }
      }

      bool IsOpTypeRuntimeArray = false;
      bool HasArgUser = false;

      for (auto ArgGV : ArgGVMap) {
        auto Arg = ArgGV.first;

        Type *ArgTy = Arg->getType();
        if (ArgTy == PTy) {
          if (AddrSpace != AddressSpace::UniformConstant) {
            IsOpTypeRuntimeArray = true;
          }

          for (auto U : Arg->users()) {
            if (!isa<GetElementPtrInst>(U) || (U->getType() == PTy)) {
              HasArgUser = true;
              break;
            }
          }
        }
      }

      if ((!IsOpTypeRuntimeArray || HasArgUser) && !UseExistingOpTypePointer) {
        //
        // Generate OpTypePointer.
        //

        // OpTypePointer
        // Ops[0] = Storage Class
        // Ops[1] = Element Type ID
        SPIRVOperandList Ops;

        spv::StorageClass StorageClass = GetStorageClass(AddrSpace);

        SPIRVOperand *StorageClassOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, StorageClass);
        Ops.push_back(StorageClassOp);

        uint32_t EleTyID = lookupType(PTy->getElementType());
        SPIRVOperand *EleTyOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, EleTyID);
        Ops.push_back(EleTyOp);

        spv::Op Opcode = spv::OpTypePointer;
        uint16_t WordCount = 4;

        SPIRVInstruction *Inst =
            new SPIRVInstruction(WordCount, Opcode, nextID++, Ops);
        SPIRVInstList.push_back(Inst);
      }

      if (IsOpTypeRuntimeArray) {
        //
        // Generate OpTypeRuntimeArray.
        //

        // OpTypeRuntimeArray
        // Ops[0] = Element Type ID
        SPIRVOperandList Ops;

        uint32_t EleTyID = lookupType(PTy->getElementType());
        SPIRVOperand *EleTyOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, EleTyID);
        Ops.push_back(EleTyOp);

        spv::Op Opcode = spv::OpTypeRuntimeArray;
        uint16_t WordCount = 3;

        uint32_t OpTypeRuntimeArrayID = nextID;
        assert(0 == OpRuntimeTyMap.count(Ty));
        OpRuntimeTyMap[Ty] = nextID;

        SPIRVInstruction *Inst =
            new SPIRVInstruction(WordCount, Opcode, nextID++, Ops);
        SPIRVInstList.push_back(Inst);

        // Generate OpDecorate.
        auto DecoInsertPoint =
            std::find_if(SPIRVInstList.begin(), SPIRVInstList.end(),
                         [](SPIRVInstruction *Inst) -> bool {
                           return Inst->getOpcode() != spv::OpDecorate &&
                                  Inst->getOpcode() != spv::OpMemberDecorate &&
                                  Inst->getOpcode() != spv::OpExtInstImport;
                         });

        // Ops[0] = Target ID
        // Ops[1] = Decoration (ArrayStride)
        // Ops[2] = Stride Number(Literal Number)
        Ops.clear();

        SPIRVOperand *PTyIDOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, OpTypeRuntimeArrayID);
        Ops.push_back(PTyIDOp);

        SPIRVOperand *DecoOp = new SPIRVOperand(SPIRVOperandType::NUMBERID,
                                                spv::DecorationArrayStride);
        Ops.push_back(DecoOp);

        std::vector<uint32_t> LiteralNum;
        Type *EleTy = PTy->getElementType();
        const unsigned ArrayStride = DL.getTypeAllocSize(EleTy);
        LiteralNum.push_back(ArrayStride);
        SPIRVOperand *ArrayStrideOp =
            new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, LiteralNum);
        Ops.push_back(ArrayStrideOp);

        SPIRVInstruction *DecoInst =
            new SPIRVInstruction(4, spv::OpDecorate, 0 /* No id */, Ops);
        SPIRVInstList.insert(DecoInsertPoint, DecoInst);
      }
      break;
    }
    case Type::StructTyID: {
      LLVMContext &Context = Ty->getContext();

      StructType *STy = cast<StructType>(Ty);

      // Handle sampler type.
      if (STy->isOpaque()) {
        if (STy->getName().equals("opencl.sampler_t")) {
          //
          // Generate OpTypeSampler
          //
          // Empty Ops.
          SPIRVOperandList Ops;

          SPIRVInstruction *Inst =
              new SPIRVInstruction(2, spv::OpTypeSampler, nextID++, Ops);
          SPIRVInstList.push_back(Inst);
          break;
        } else if (STy->getName().equals("opencl.image2d_ro_t") ||
                   STy->getName().equals("opencl.image2d_wo_t") ||
                   STy->getName().equals("opencl.image3d_ro_t") ||
                   STy->getName().equals("opencl.image3d_wo_t")) {
          //
          // Generate OpTypeImage
          //
          // Ops[0] = Sampled Type ID
          // Ops[1] = Dim ID
          // Ops[2] = Depth (Literal Number)
          // Ops[3] = Arrayed (Literal Number)
          // Ops[4] = MS (Literal Number)
          // Ops[5] = Sampled (Literal Number)
          // Ops[6] = Image Format ID
          //
          SPIRVOperandList Ops;

          // TODO: Changed Sampled Type according to situations.
          uint32_t SampledTyID = lookupType(Type::getFloatTy(Context));
          SPIRVOperand *SampledTyIDOp =
              new SPIRVOperand(SPIRVOperandType::NUMBERID, SampledTyID);
          Ops.push_back(SampledTyIDOp);

          spv::Dim DimID = spv::Dim2D;
          if (STy->getName().equals("opencl.image3d_ro_t") ||
              STy->getName().equals("opencl.image3d_wo_t")) {
            DimID = spv::Dim3D;
          }
          SPIRVOperand *DimIDOp =
              new SPIRVOperand(SPIRVOperandType::NUMBERID, DimID);
          Ops.push_back(DimIDOp);

          // TODO: Set up Depth.
          std::vector<uint32_t> LiteralNum;
          LiteralNum.push_back(0);
          SPIRVOperand *DepthOp =
              new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, LiteralNum);
          Ops.push_back(DepthOp);

          // TODO: Set up Arrayed.
          LiteralNum.clear();
          LiteralNum.push_back(0);
          SPIRVOperand *ArrayedOp =
              new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, LiteralNum);
          Ops.push_back(ArrayedOp);

          // TODO: Set up MS.
          LiteralNum.clear();
          LiteralNum.push_back(0);
          SPIRVOperand *MSOp =
              new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, LiteralNum);
          Ops.push_back(MSOp);

          // TODO: Set up Sampled.
          //
          // From Spec
          //
          // 0 indicates this is only known at run time, not at compile time
          // 1 indicates will be used with sampler
          // 2 indicates will be used without a sampler (a storage image)
          uint32_t Sampled = 1;
          if (STy->getName().equals("opencl.image2d_wo_t") ||
              STy->getName().equals("opencl.image3d_wo_t")) {
            Sampled = 2;
          }
          LiteralNum.clear();
          LiteralNum.push_back(Sampled);
          SPIRVOperand *SampledOp =
              new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, LiteralNum);
          Ops.push_back(SampledOp);

          // TODO: Set up Image Format.
          SPIRVOperand *ImageFormatOp = new SPIRVOperand(
              SPIRVOperandType::NUMBERID, spv::ImageFormatUnknown);
          Ops.push_back(ImageFormatOp);

          SPIRVInstruction *Inst =
              new SPIRVInstruction(9, spv::OpTypeImage, nextID++, Ops);
          SPIRVInstList.push_back(Inst);
          break;
        }
      }

      //
      // Generate OpTypeStruct
      //
      // Ops[0] ... Ops[n] = Member IDs
      SPIRVOperandList Ops;

      for (auto *EleTy : STy->elements()) {
        uint32_t EleTyID = lookupType(EleTy);

        // Check OpTypeRuntimeArray.
        if (isa<PointerType>(EleTy)) {
          for (auto ArgGV : ArgGVMap) {
            Type *ArgTy = ArgGV.first->getType();
            if (ArgTy == EleTy) {
              assert(0 != OpRuntimeTyMap.count(EleTy));
              EleTyID = OpRuntimeTyMap[EleTy];
            }
          }
        }

        SPIRVOperand *EleTyOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, EleTyID);
        Ops.push_back(EleTyOp);
      }

      uint16_t WordCount = static_cast<uint16_t>(2 + Ops.size());
      uint32_t STyID = nextID;

      SPIRVInstruction *Inst =
          new SPIRVInstruction(WordCount, spv::OpTypeStruct, nextID++, Ops);
      SPIRVInstList.push_back(Inst);

      // Generate OpMemberDecorate.
      auto DecoInsertPoint =
          std::find_if(SPIRVInstList.begin(), SPIRVInstList.end(),
                       [](SPIRVInstruction *Inst) -> bool {
                         return Inst->getOpcode() != spv::OpDecorate &&
                                Inst->getOpcode() != spv::OpMemberDecorate &&
                                Inst->getOpcode() != spv::OpExtInstImport;
                       });

      uint32_t ByteOffset = 0;
      for (unsigned MemberIdx = 0; MemberIdx < STy->getNumElements();
           MemberIdx++) {
        // Ops[0] = Structure Type ID
        // Ops[1] = Member Index(Literal Number)
        // Ops[2] = Decoration (Offset)
        // Ops[3] = Byte Offset (Literal Number)
        Ops.clear();

        SPIRVOperand *STyIDOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, STyID);
        Ops.push_back(STyIDOp);

        std::vector<uint32_t> LiteralNum;
        LiteralNum.push_back(MemberIdx);
        SPIRVOperand *MemberIdxOp =
            new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, LiteralNum);
        Ops.push_back(MemberIdxOp);

        SPIRVOperand *DecoOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, spv::DecorationOffset);
        Ops.push_back(DecoOp);

        LiteralNum.clear();
        LiteralNum.push_back(ByteOffset);
        SPIRVOperand *ByteOffsetOp =
            new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, LiteralNum);
        Ops.push_back(ByteOffsetOp);

        SPIRVInstruction *DecoInst =
            new SPIRVInstruction(5, spv::OpMemberDecorate, 0 /* No id */, Ops);
        SPIRVInstList.insert(DecoInsertPoint, DecoInst);

        // Update ByteOffset.
        Type *EleTy = STy->getElementType(MemberIdx);
        ByteOffset += static_cast<uint32_t>(DL.getTypeSizeInBits(EleTy) / 8);
      }

      // Generate OpDecorate.
      for (auto ArgGV : ArgGVMap) {
        Type *ArgGVTy = ArgGV.second->getType();
        PointerType *PTy = cast<PointerType>(ArgGVTy);
        Type *ArgTy = PTy->getElementType();

        // Struct type from argument is already distinguished with the other
        // struct types on llvm types. As a result, if current processing struct
        // type is same with argument type, we can generate OpDecorate with
        // Block or BufferBlock.
        if (ArgTy == STy) {
          // Ops[0] = Target ID
          // Ops[1] = Decoration (Block or BufferBlock)
          Ops.clear();

          SPIRVOperand *STyIDOp =
              new SPIRVOperand(SPIRVOperandType::NUMBERID, STyID);
          Ops.push_back(STyIDOp);

          // Use Block decorations with StorageBuffer storage class.
          const spv::Decoration Deco = spv::DecorationBlock;

          SPIRVOperand *DecoOp =
              new SPIRVOperand(SPIRVOperandType::NUMBERID, Deco);
          Ops.push_back(DecoOp);

          SPIRVInstruction *DecoInst =
              new SPIRVInstruction(3, spv::OpDecorate, 0 /* No id */, Ops);
          SPIRVInstList.insert(DecoInsertPoint, DecoInst);
          break;
        }
      }
      break;
    }
    case Type::IntegerTyID: {
      unsigned BitWidth = Ty->getPrimitiveSizeInBits();

      if (BitWidth == 1) {
        SPIRVInstruction *Inst =
            new SPIRVInstruction(2, spv::OpTypeBool, nextID++, {});
        SPIRVInstList.push_back(Inst);
      } else {
        // i8 is added to TypeMap as i32.
        if (BitWidth == 8) {
          BitWidth = 32;
        }

        SPIRVOperand *Ops[2] = {
            new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, BitWidth),
            new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, 0u)};

        SPIRVInstList.push_back(
            new SPIRVInstruction(4, spv::OpTypeInt, nextID++, Ops));
      }
      break;
    }
    case Type::HalfTyID:
    case Type::FloatTyID:
    case Type::DoubleTyID: {
      SPIRVOperand *WidthOp = new SPIRVOperand(
          SPIRVOperandType::LITERAL_INTEGER, Ty->getPrimitiveSizeInBits());

      SPIRVInstList.push_back(
          new SPIRVInstruction(3, spv::OpTypeFloat, nextID++, WidthOp));
      break;
    }
    case Type::ArrayTyID: {
      LLVMContext &Context = Ty->getContext();
      ArrayType *ArrTy = cast<ArrayType>(Ty);
      //
      // Generate OpConstant and OpTypeArray.
      //

      //
      // Generate OpConstant for array length.
      //
      // Ops[0] = Result Type ID
      // Ops[1] .. Ops[n] = Values LiteralNumber
      SPIRVOperandList Ops;

      Type *LengthTy = Type::getInt32Ty(Context);
      uint32_t ResTyID = lookupType(LengthTy);
      SPIRVOperand *ResTyOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, ResTyID);
      Ops.push_back(ResTyOp);

      uint64_t Length = ArrTy->getArrayNumElements();
      assert(Length < UINT32_MAX);
      std::vector<uint32_t> LiteralNum;
      LiteralNum.push_back(static_cast<uint32_t>(Length));
      SPIRVOperand *ValOp =
          new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, LiteralNum);
      Ops.push_back(ValOp);

      // Add constant for length to constant list.
      Constant *CstLength = ConstantInt::get(LengthTy, Length);
      AllocatedVMap[CstLength] = nextID;
      VMap[CstLength] = nextID;
      uint32_t LengthID = nextID;

      SPIRVInstruction *CstInst =
          new SPIRVInstruction(4, spv::OpConstant, nextID++, Ops);
      SPIRVInstList.push_back(CstInst);

      //
      // Generate OpTypeArray.
      //
      // Ops[0] = Element Type ID
      // Ops[1] = Array Length Constant ID
      Ops.clear();

      uint32_t EleTyID = lookupType(ArrTy->getElementType());
      SPIRVOperand *EleTyOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, EleTyID);
      Ops.push_back(EleTyOp);

      SPIRVOperand *LengthOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, LengthID);
      Ops.push_back(LengthOp);

      // Update TypeMap with nextID.
      TypeMap[Ty] = nextID;

      SPIRVInstruction *ArrayInst =
          new SPIRVInstruction(4, spv::OpTypeArray, nextID++, Ops);
      SPIRVInstList.push_back(ArrayInst);
      break;
    }
    case Type::VectorTyID: {
      // <4 x i8> is changed to i32.
      LLVMContext &Context = Ty->getContext();
      if (Ty->getVectorElementType() == Type::getInt8Ty(Context)) {
        if (Ty->getVectorNumElements() == 4) {
          TypeMap[Ty] = lookupType(Ty->getVectorElementType());
          break;
        } else {
          Ty->print(errs());
          llvm_unreachable("Support above i8 vector type");
        }
      }

      // Ops[0] = Component Type ID
      // Ops[1] = Component Count (Literal Number)
      SPIRVOperand *Ops[2] = {
          new SPIRVOperand(SPIRVOperandType::NUMBERID,
                           lookupType(Ty->getVectorElementType())),
          new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER,
                           Ty->getVectorNumElements())};

      SPIRVInstList.push_back(
          new SPIRVInstruction(4, spv::OpTypeVector, nextID++, Ops));
      break;
    }
    case Type::VoidTyID: {
      SPIRVInstruction *Inst =
          new SPIRVInstruction(2, spv::OpTypeVoid, nextID++, {});
      SPIRVInstList.push_back(Inst);
      break;
    }
    case Type::FunctionTyID: {
      // Generate SPIRV instruction for function type.
      FunctionType *FTy = cast<FunctionType>(Ty);

      // Ops[0] = Return Type ID
      // Ops[1] ... Ops[n] = Parameter Type IDs
      SPIRVOperandList Ops;

      // Find SPIRV instruction for return type
      uint32_t RetTyID = lookupType(FTy->getReturnType());

      SPIRVOperand *RetTyOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, RetTyID);
      Ops.push_back(RetTyOp);

      // Find SPIRV instructions for parameter types
      for (unsigned k = 0; k < FTy->getNumParams(); k++) {
        // Find SPIRV instruction for parameter type.
        auto ParamTy = FTy->getParamType(k);
        if (ParamTy->isPointerTy()) {
          auto PointeeTy = ParamTy->getPointerElementType();
          if (PointeeTy->isStructTy() &&
              dyn_cast<StructType>(PointeeTy)->isOpaque()) {
            ParamTy = PointeeTy;
          }
        }

        uint32_t ParamTyID = lookupType(ParamTy);
        SPIRVOperand *ParamTyOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, ParamTyID);
        Ops.push_back(ParamTyOp);
      }

      // Return type id is included in operand list.
      uint16_t WordCount = static_cast<uint16_t>(2 + Ops.size());

      SPIRVInstruction *Inst =
          new SPIRVInstruction(WordCount, spv::OpTypeFunction, nextID++, Ops);
      SPIRVInstList.push_back(Inst);
      break;
    }
    }
  }

  // Generate OpTypeSampledImage.
  TypeMapType &OpImageTypeMap = getImageTypeMap();
  for (auto &ImageType : OpImageTypeMap) {
    //
    // Generate OpTypeSampledImage.
    //
    // Ops[0] = Image Type ID
    //
    SPIRVOperandList Ops;

    Type *ImgTy = ImageType.first;
    uint32_t ImgTyID = TypeMap[ImgTy];
    SPIRVOperand *ImgTyOp =
        new SPIRVOperand(SPIRVOperandType::NUMBERID, ImgTyID);
    Ops.push_back(ImgTyOp);

    // Update OpImageTypeMap.
    ImageType.second = nextID;

    SPIRVInstruction *Inst =
        new SPIRVInstruction(3, spv::OpTypeSampledImage, nextID++, Ops);
    SPIRVInstList.push_back(Inst);
  }
}

void SPIRVProducerPass::GenerateSPIRVConstants() {
  SPIRVInstructionList &SPIRVInstList = getSPIRVInstList();
  ValueMapType &VMap = getValueMap();
  ValueMapType &AllocatedVMap = getAllocatedValueMap();
  ValueList &CstList = getConstantList();

  for (uint32_t i = 0; i < CstList.size(); i++) {
    Constant *Cst = cast<Constant>(CstList[i]);

    // OpTypeArray's constant was already generated.
    if (AllocatedVMap[Cst]) {
      continue;
    }

    // Update TypeMap with nextID for reference later.
    VMap[Cst] = nextID;

    //
    // Generate OpConstant.
    //

    // Ops[0] = Result Type ID
    // Ops[1] .. Ops[n] = Values LiteralNumber
    SPIRVOperandList Ops;

    uint32_t ResTyID = lookupType(Cst->getType());
    SPIRVOperand *ResTyIDOp =
        new SPIRVOperand(SPIRVOperandType::NUMBERID, ResTyID);
    Ops.push_back(ResTyIDOp);

    std::vector<uint32_t> LiteralNum;
    uint16_t WordCount = 0;
    spv::Op Opcode = spv::OpNop;

    if (isa<UndefValue>(Cst)) {
      // Ops[0] = Result Type ID
      Opcode = spv::OpUndef;
      WordCount = 3;
    } else if (const ConstantInt *CI = dyn_cast<ConstantInt>(Cst)) {
      unsigned BitWidth = CI->getBitWidth();
      if (BitWidth == 1) {
        // If the bitwidth of constant is 1, generate OpConstantTrue or
        // OpConstantFalse.
        if (CI->getZExtValue()) {
          // Ops[0] = Result Type ID
          Opcode = spv::OpConstantTrue;
        } else {
          // Ops[0] = Result Type ID
          Opcode = spv::OpConstantFalse;
        }
        WordCount = 3;
      } else {
        auto V = CI->getZExtValue();
        LiteralNum.push_back(V & 0xFFFFFFFF);

        if (BitWidth > 32) {
          LiteralNum.push_back(V >> 32);
        }

        Opcode = spv::OpConstant;
        WordCount = static_cast<uint16_t>(3 + LiteralNum.size());

        SPIRVOperand *CstValue =
            new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, LiteralNum);
        Ops.push_back(CstValue);
      }
    } else if (const ConstantFP *CFP = dyn_cast<ConstantFP>(Cst)) {
      uint64_t FPVal = CFP->getValueAPF().bitcastToAPInt().getZExtValue();
      Type *CFPTy = CFP->getType();
      if (CFPTy->isFloatTy()) {
        LiteralNum.push_back(FPVal & 0xFFFFFFFF);
      } else {
        CFPTy->print(errs());
        llvm_unreachable("Implement this ConstantFP Type");
      }

      Opcode = spv::OpConstant;
      WordCount = static_cast<uint16_t>(3 + LiteralNum.size());

      SPIRVOperand *CstValue =
          new SPIRVOperand(SPIRVOperandType::LITERAL_FLOAT, LiteralNum);
      Ops.push_back(CstValue);
    } else if (isa<ConstantDataSequential>(Cst) &&
               cast<ConstantDataSequential>(Cst)->isString()) {
      Cst->print(errs());
      llvm_unreachable("Implement this Constant");

    } else if (const ConstantDataSequential *CDS =
                   dyn_cast<ConstantDataSequential>(Cst)) {
      for (unsigned k = 0; k < CDS->getNumElements(); k++) {
        Constant *EleCst = CDS->getElementAsConstant(k);
        uint32_t EleCstID = VMap[EleCst];
        SPIRVOperand *EleCstIDOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, EleCstID);
        Ops.push_back(EleCstIDOp);
      }

      Opcode = spv::OpConstantComposite;
      WordCount = static_cast<uint16_t>(3 + CDS->getNumElements());
    } else if (const ConstantAggregate *CA = dyn_cast<ConstantAggregate>(Cst)) {
      // Let's convert <4 x i8> constant to int constant specially.
      Type *CstTy = Cst->getType();
      if (is4xi8vec(CstTy)) {
        LLVMContext &Context = CstTy->getContext();

        //
        // Generate OpConstant with OpTypeInt 32 0.
        //
        uint64_t IntValue = 0;
        uint32_t Idx = 0;
        for (User::const_op_iterator I = Cst->op_begin(), E = Cst->op_end();
             I != E; ++I) {
          uint64_t Val = 0;
          if (ConstantInt *CI2 = dyn_cast<ConstantInt>(I)) {
            Val = CI2->getZExtValue();
          }
          IntValue = (IntValue << Idx) | Val;
        }

        ConstantInt *CstInt =
            ConstantInt::get(Type::getInt32Ty(Context), IntValue);
        // If this constant is already registered on VMap, use it.
        if (VMap.count(CstInt)) {
          uint32_t CstID = VMap[CstInt];
          VMap[Cst] = CstID;
          return;
        }

        LiteralNum.push_back(IntValue & 0xFFFFFFFF);
        SPIRVOperand *CstValue =
            new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, LiteralNum);
        Ops.push_back(CstValue);

        SPIRVInstruction *CstInst =
            new SPIRVInstruction(4, spv::OpConstant, nextID++, Ops);
        SPIRVInstList.push_back(CstInst);

        return;
      }

      // We use a constant composite in SPIR-V for our constant aggregate in
      // LLVM.
      Opcode = spv::OpConstantComposite;
      WordCount = static_cast<uint16_t>(3 + CA->getNumOperands());

      for (unsigned k = 0; k < CA->getNumOperands(); k++) {
        // Look up the ID of the element of this aggregate (which we will
        // previously have created a constant for).
        uint32_t ElementConstantID = VMap[CA->getAggregateElement(k)];

        // And add an operand to the composite we are constructing
        Ops.push_back(
            new SPIRVOperand(SPIRVOperandType::NUMBERID, ElementConstantID));
      }
    } else if (Cst->isNullValue()) {
      Opcode = spv::OpConstantNull;
      WordCount = 3;
    } else {
      Cst->print(errs());
      llvm_unreachable("Unsupported Constant???");
    }

    SPIRVInstruction *CstInst =
        new SPIRVInstruction(WordCount, Opcode, nextID++, Ops);
    SPIRVInstList.push_back(CstInst);
  }
}

void SPIRVProducerPass::GenerateSamplers(Module &M) {
  SPIRVInstructionList &SPIRVInstList = getSPIRVInstList();
  ValueMapType &VMap = getValueMap();

  DenseMap<unsigned, unsigned> SamplerLiteralToIDMap;

  unsigned BindingIdx = 0;

  // Generate the sampler map.
  for (auto SamplerLiteral : getSamplerMap()) {
    // Generate OpVariable.
    //
    // GIDOps[0] : Result Type ID
    // GIDOps[1] : Storage Class
    SPIRVOperandList Ops;

    Ops.push_back(
        new SPIRVOperand(SPIRVOperandType::NUMBERID, lookupType(SamplerTy)));

    spv::StorageClass StorageClass = spv::StorageClassUniformConstant;
    Ops.push_back(new SPIRVOperand(SPIRVOperandType::NUMBERID, StorageClass));

    SPIRVInstruction *Inst = new SPIRVInstruction(
        static_cast<uint16_t>(2 + Ops.size()), spv::OpVariable, nextID, Ops);
    SPIRVInstList.push_back(Inst);

    SamplerLiteralToIDMap[SamplerLiteral.first] = nextID++;

    // Find Insert Point for OpDecorate.
    auto DecoInsertPoint =
        std::find_if(SPIRVInstList.begin(), SPIRVInstList.end(),
                     [](SPIRVInstruction *Inst) -> bool {
                       return Inst->getOpcode() != spv::OpDecorate &&
                              Inst->getOpcode() != spv::OpMemberDecorate &&
                              Inst->getOpcode() != spv::OpExtInstImport;
                     });

    // Ops[0] = Target ID
    // Ops[1] = Decoration (DescriptorSet)
    // Ops[2] = LiteralNumber according to Decoration
    Ops.clear();

    SPIRVOperand *ArgIDOp =
        new SPIRVOperand(SPIRVOperandType::NUMBERID,
                         SamplerLiteralToIDMap[SamplerLiteral.first]);
    Ops.push_back(ArgIDOp);

    spv::Decoration Deco = spv::DecorationDescriptorSet;
    SPIRVOperand *DecoOp = new SPIRVOperand(SPIRVOperandType::NUMBERID, Deco);
    Ops.push_back(DecoOp);

    descriptorMapOut << "sampler," << SamplerLiteral.first << ",samplerExpr,\""
                     << SamplerLiteral.second << "\",descriptorSet,0,binding,"
                     << BindingIdx << "\n";

    std::vector<uint32_t> LiteralNum;
    LiteralNum.push_back(0);
    SPIRVOperand *DescSet =
        new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, LiteralNum);
    Ops.push_back(DescSet);

    SPIRVInstruction *DescDecoInst =
        new SPIRVInstruction(4, spv::OpDecorate, 0 /* No id */, Ops);
    SPIRVInstList.insert(DecoInsertPoint, DescDecoInst);

    // Ops[0] = Target ID
    // Ops[1] = Decoration (Binding)
    // Ops[2] = LiteralNumber according to Decoration
    Ops.clear();

    Ops.push_back(ArgIDOp);

    Deco = spv::DecorationBinding;
    DecoOp = new SPIRVOperand(SPIRVOperandType::NUMBERID, Deco);
    Ops.push_back(DecoOp);

    LiteralNum.clear();
    LiteralNum.push_back(BindingIdx++);
    SPIRVOperand *Binding =
        new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, LiteralNum);
    Ops.push_back(Binding);

    SPIRVInstruction *BindDecoInst =
        new SPIRVInstruction(4, spv::OpDecorate, 0 /* No id */, Ops);
    SPIRVInstList.insert(DecoInsertPoint, BindDecoInst);
  }

  const char *TranslateSamplerFunctionName = "__translate_sampler_initializer";

  auto SamplerFunction = M.getFunction(TranslateSamplerFunctionName);

  // If there are no uses of the sampler function, no work to do!
  if (!SamplerFunction) {
    return;
  }

  // Iterate through the users of the sampler function.
  for (auto User : SamplerFunction->users()) {
    if (auto CI = dyn_cast<CallInst>(User)) {
      // Get the literal used to initialize the sampler.
      auto Constant = dyn_cast<ConstantInt>(CI->getArgOperand(0));

      if (!Constant) {
        CI->getArgOperand(0)->print(errs());
        llvm_unreachable("Argument of sampler initializer was non-constant!");
      }

      auto SamplerLiteral = static_cast<unsigned>(Constant->getZExtValue());

      if (0 == SamplerLiteralToIDMap.count(SamplerLiteral)) {
        Constant->print(errs());
        llvm_unreachable("Sampler literal was not found in sampler map!");
      }

      // Calls to the sampler literal function to initialize a sampler are
      // re-routed to the global variables declared for the sampler.
      VMap[CI] = SamplerLiteralToIDMap[SamplerLiteral];
    }
  }
}

void SPIRVProducerPass::GenerateGlobalVar(GlobalVariable &GV) {
  SPIRVInstructionList &SPIRVInstList = getSPIRVInstList();
  ValueMapType &VMap = getValueMap();
  std::vector<uint32_t> &BuiltinDimVec = getBuiltinDimVec();

  const spv::BuiltIn BuiltinType = GetBuiltin(GV.getName());
  Type *Ty = GV.getType();
  PointerType *PTy = cast<PointerType>(Ty);

  uint32_t InitializerID = 0;

  // Workgroup size is handled differently (it goes into a constant)
  if (spv::BuiltInWorkgroupSize == BuiltinType) {
    std::vector<bool> HasMDVec;
    uint32_t PrevXDimCst = 0xFFFFFFFF;
    uint32_t PrevYDimCst = 0xFFFFFFFF;
    uint32_t PrevZDimCst = 0xFFFFFFFF;
    for (Function &Func : *GV.getParent()) {
      if (Func.isDeclaration()) {
        continue;
      }

      // We only need to check kernels.
      if (Func.getCallingConv() != CallingConv::SPIR_KERNEL) {
        continue;
      }

      if (const MDNode *MD =
              dyn_cast<Function>(&Func)->getMetadata("reqd_work_group_size")) {
        uint32_t CurXDimCst = static_cast<uint32_t>(
            mdconst::extract<ConstantInt>(MD->getOperand(0))->getZExtValue());
        uint32_t CurYDimCst = static_cast<uint32_t>(
            mdconst::extract<ConstantInt>(MD->getOperand(1))->getZExtValue());
        uint32_t CurZDimCst = static_cast<uint32_t>(
            mdconst::extract<ConstantInt>(MD->getOperand(2))->getZExtValue());

        if (PrevXDimCst == 0xFFFFFFFF && PrevYDimCst == 0xFFFFFFFF &&
            PrevZDimCst == 0xFFFFFFFF) {
          PrevXDimCst = CurXDimCst;
          PrevYDimCst = CurYDimCst;
          PrevZDimCst = CurZDimCst;
        } else if (CurXDimCst != PrevXDimCst || CurYDimCst != PrevYDimCst ||
                   CurZDimCst != PrevZDimCst) {
          llvm_unreachable(
              "reqd_work_group_size must be the same across all kernels");
        } else {
          continue;
        }

        //
        // Generate OpConstantComposite.
        //
        // Ops[0] : Result Type ID
        // Ops[1] : Constant size for x dimension.
        // Ops[2] : Constant size for y dimension.
        // Ops[3] : Constant size for z dimension.
        SPIRVOperandList Ops;

        uint32_t XDimCstID =
            VMap[mdconst::extract<ConstantInt>(MD->getOperand(0))];
        uint32_t YDimCstID =
            VMap[mdconst::extract<ConstantInt>(MD->getOperand(1))];
        uint32_t ZDimCstID =
            VMap[mdconst::extract<ConstantInt>(MD->getOperand(2))];

        InitializerID = nextID;

        Ops.push_back(
            new SPIRVOperand(SPIRVOperandType::NUMBERID,
                             lookupType(Ty->getPointerElementType())));

        Ops.push_back(new SPIRVOperand(SPIRVOperandType::NUMBERID, XDimCstID));
        Ops.push_back(new SPIRVOperand(SPIRVOperandType::NUMBERID, YDimCstID));
        Ops.push_back(new SPIRVOperand(SPIRVOperandType::NUMBERID, ZDimCstID));

        SPIRVInstruction *Inst =
            new SPIRVInstruction(6, spv::OpConstantComposite, nextID++, Ops);
        SPIRVInstList.push_back(Inst);

        HasMDVec.push_back(true);
      } else {
        HasMDVec.push_back(false);
      }
    }

    // Check all kernels have same definitions for work_group_size.
    bool HasMD = false;
    if (!HasMDVec.empty()) {
      HasMD = HasMDVec[0];
      for (uint32_t i = 1; i < HasMDVec.size(); i++) {
        if (HasMD != HasMDVec[i]) {
          llvm_unreachable(
              "Kernels should have consistent work group size definition");
        }
      }
    }

    // If all kernels do not have metadata for reqd_work_group_size, generate
    // OpSpecConstants for x/y/z dimension.
    if (!HasMD) {
      //
      // Generate OpSpecConstants for x/y/z dimension.
      //
      // Ops[0] : Result Type ID
      // Ops[1] : Constant size for x/y/z dimension (Literal Number).
      uint32_t XDimCstID = 0;
      uint32_t YDimCstID = 0;
      uint32_t ZDimCstID = 0;

      // X Dimension
      SPIRVOperandList Ops;

      Ops.push_back(new SPIRVOperand(
          SPIRVOperandType::NUMBERID,
          lookupType(Ty->getPointerElementType()->getSequentialElementType())));

      std::vector<uint32_t> LiteralNum;
      LiteralNum.push_back(1);
      SPIRVOperand *XDim =
          new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, LiteralNum);
      Ops.push_back(XDim);

      XDimCstID = nextID;
      BuiltinDimVec.push_back(XDimCstID);

      SPIRVInstruction *XDimCstInst =
          new SPIRVInstruction(4, spv::OpSpecConstant, nextID++, Ops);
      SPIRVInstList.push_back(XDimCstInst);

      // Y Dimension
      Ops.clear();

      Ops.push_back(new SPIRVOperand(
          SPIRVOperandType::NUMBERID,
          lookupType(Ty->getPointerElementType()->getSequentialElementType())));

      LiteralNum.clear();
      LiteralNum.push_back(1);
      SPIRVOperand *YDim =
          new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, LiteralNum);
      Ops.push_back(YDim);

      YDimCstID = nextID;
      BuiltinDimVec.push_back(YDimCstID);

      SPIRVInstruction *YDimCstInst =
          new SPIRVInstruction(4, spv::OpSpecConstant, nextID++, Ops);
      SPIRVInstList.push_back(YDimCstInst);

      // Z Dimension
      Ops.clear();

      Ops.push_back(new SPIRVOperand(
          SPIRVOperandType::NUMBERID,
          lookupType(Ty->getPointerElementType()->getSequentialElementType())));

      LiteralNum.clear();
      LiteralNum.push_back(1);
      SPIRVOperand *ZDim =
          new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, LiteralNum);
      Ops.push_back(ZDim);

      ZDimCstID = nextID;
      BuiltinDimVec.push_back(ZDimCstID);

      SPIRVInstruction *ZDimCstInst =
          new SPIRVInstruction(4, spv::OpSpecConstant, nextID++, Ops);
      SPIRVInstList.push_back(ZDimCstInst);

      //
      // Generate OpSpecConstantComposite.
      //
      // Ops[0] : Result Type ID
      // Ops[1] : Constant size for x dimension.
      // Ops[2] : Constant size for y dimension.
      // Ops[3] : Constant size for z dimension.
      InitializerID = nextID;

      Ops.clear();

      Ops.push_back(new SPIRVOperand(SPIRVOperandType::NUMBERID,
                                     lookupType(Ty->getPointerElementType())));

      Ops.push_back(new SPIRVOperand(SPIRVOperandType::NUMBERID, XDimCstID));
      Ops.push_back(new SPIRVOperand(SPIRVOperandType::NUMBERID, YDimCstID));
      Ops.push_back(new SPIRVOperand(SPIRVOperandType::NUMBERID, ZDimCstID));

      SPIRVInstruction *Inst =
          new SPIRVInstruction(6, spv::OpSpecConstantComposite, nextID++, Ops);
      SPIRVInstList.push_back(Inst);
    }
  }

  if (GV.hasInitializer()) {
    InitializerID = VMap[GV.getInitializer()];
  }

  VMap[&GV] = nextID;

  //
  // Generate OpVariable.
  //
  // GIDOps[0] : Result Type ID
  // GIDOps[1] : Storage Class
  SPIRVOperandList Ops;

  Ops.push_back(new SPIRVOperand(SPIRVOperandType::NUMBERID, lookupType(Ty)));

  spv::StorageClass StorageClass = GetStorageClass(PTy->getAddressSpace());
  Ops.push_back(new SPIRVOperand(SPIRVOperandType::NUMBERID, StorageClass));

  if (0 != InitializerID) {
    Ops.push_back(new SPIRVOperand(SPIRVOperandType::NUMBERID, InitializerID));
  }

  SPIRVInstruction *Inst = new SPIRVInstruction(
      static_cast<uint16_t>(2 + Ops.size()), spv::OpVariable, nextID++, Ops);
  SPIRVInstList.push_back(Inst);

  // If we have a builtin.
  if (spv::BuiltInMax != BuiltinType) {
    // Find Insert Point for OpDecorate.
    auto DecoInsertPoint =
        std::find_if(SPIRVInstList.begin(), SPIRVInstList.end(),
                     [](SPIRVInstruction *Inst) -> bool {
                       return Inst->getOpcode() != spv::OpDecorate &&
                              Inst->getOpcode() != spv::OpMemberDecorate &&
                              Inst->getOpcode() != spv::OpExtInstImport;
                     });
    //
    // Generate OpDecorate.
    //
    // DOps[0] = Target ID
    // DOps[1] = Decoration (Builtin)
    // DOps[2] = BuiltIn ID
    uint32_t ResultID;

    // WorkgroupSize is different, we decorate the constant composite that has
    // its value, rather than the variable that we use to access the value.
    if (spv::BuiltInWorkgroupSize == BuiltinType) {
      ResultID = InitializerID;
    } else {
      ResultID = VMap[&GV];
    }

    SPIRVOperandList DOps;
    SPIRVOperand *ResultIDOp =
        new SPIRVOperand(SPIRVOperandType::NUMBERID, ResultID);
    DOps.push_back(ResultIDOp);

    spv::Decoration Deco = spv::DecorationBuiltIn;
    SPIRVOperand *DecoOp = new SPIRVOperand(SPIRVOperandType::NUMBERID, Deco);
    DOps.push_back(DecoOp);

    SPIRVOperand *Builtin =
        new SPIRVOperand(SPIRVOperandType::NUMBERID, BuiltinType);
    DOps.push_back(Builtin);

    SPIRVInstruction *DescDecoInst =
        new SPIRVInstruction(4, spv::OpDecorate, 0 /* No id */, DOps);
    SPIRVInstList.insert(DecoInsertPoint, DescDecoInst);
  }
}

void SPIRVProducerPass::GenerateFuncPrologue(Function &F) {
  SPIRVInstructionList &SPIRVInstList = getSPIRVInstList();
  ValueMapType &VMap = getValueMap();
  EntryPointVecType &EntryPoints = getEntryPointVec();
  ValueToValueMapTy &ArgGVMap = getArgumentGVMap();
  ValueMapType &ArgGVIDMap = getArgumentGVIDMap();
  auto &GlobalConstFuncTyMap = getGlobalConstFuncTypeMap();
  auto &GlobalConstArgSet = getGlobalConstArgSet();
  const DataLayout& dataLayout(F.getParent()->getDataLayout());

  FunctionType *FTy = F.getFunctionType();

  //
  // Generate OpVariable and OpDecorate for kernel function with arguments.
  //
  if (F.getCallingConv() == CallingConv::SPIR_KERNEL) {

    // Find Insert Point for OpDecorate.
    auto DecoInsertPoint =
        std::find_if(SPIRVInstList.begin(), SPIRVInstList.end(),
                     [](SPIRVInstruction *Inst) -> bool {
                       return Inst->getOpcode() != spv::OpDecorate &&
                              Inst->getOpcode() != spv::OpMemberDecorate &&
                              Inst->getOpcode() != spv::OpExtInstImport;
                     });

    uint32_t DescriptorSetIdx = (0 < getSamplerMap().size()) ? 1u : 0u;
    for (Function &Func : *F.getParent()) {
      if (Func.isDeclaration()) {
        continue;
      }

      if (Func.getCallingConv() == CallingConv::SPIR_KERNEL) {
        if (&Func == &F) {
          break;
        }
        DescriptorSetIdx++;
      }
    }

    const auto *ArgMap = F.getMetadata("kernel_arg_map");
    // Emit descriptor map entries, if there was explicit metadata
    // attached.
    if (ArgMap) {
      for (const auto &arg : ArgMap->operands()) {
        const MDNode *arg_node = dyn_cast<MDNode>(arg.get());
        assert(arg_node->getNumOperands() == 4);
        const auto name =
            dyn_cast<MDString>(arg_node->getOperand(0))->getString();
        const auto old_index =
            dyn_extract<ConstantInt>(arg_node->getOperand(1))->getZExtValue();
        const auto new_index =
            dyn_extract<ConstantInt>(arg_node->getOperand(2))->getZExtValue();
        const auto offset =
            dyn_extract<ConstantInt>(arg_node->getOperand(3))->getZExtValue();
        descriptorMapOut << "kernel," << F.getName() << ",arg," << name
                         << ",argOrdinal," << old_index << ",descriptorSet,"
                         << DescriptorSetIdx << ",binding," << new_index
                         << ",offset," << offset << "\n";
      }
    }

    uint32_t BindingIdx = 0;
    for (auto &Arg : F.args()) {
      Value *NewGV = ArgGVMap[&Arg];
      VMap[&Arg] = VMap[NewGV];
      ArgGVIDMap[&Arg] = VMap[&Arg];

      // Emit a descriptor map entry for this arg, in case there was no explicit
      // kernel arg mapping metadata.
      if (!ArgMap) {
        descriptorMapOut << "kernel," << F.getName() << ",arg," << Arg.getName()
                         << ",argOrdinal," << BindingIdx << ",descriptorSet,"
                         << DescriptorSetIdx << ",binding," << BindingIdx
                         << ",offset,0\n";
      }

      // Ops[0] = Target ID
      // Ops[1] = Decoration (DescriptorSet)
      // Ops[2] = LiteralNumber according to Decoration
      SPIRVOperandList Ops;

      SPIRVOperand *ArgIDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, VMap[&Arg]);
      Ops.push_back(ArgIDOp);

      spv::Decoration Deco = spv::DecorationDescriptorSet;
      SPIRVOperand *DecoOp = new SPIRVOperand(SPIRVOperandType::NUMBERID, Deco);
      Ops.push_back(DecoOp);

      std::vector<uint32_t> LiteralNum;
      LiteralNum.push_back(DescriptorSetIdx);
      SPIRVOperand *DescSet =
          new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, LiteralNum);
      Ops.push_back(DescSet);

      SPIRVInstruction *DescDecoInst =
          new SPIRVInstruction(4, spv::OpDecorate, 0 /* No id */, Ops);
      SPIRVInstList.insert(DecoInsertPoint, DescDecoInst);

      // Ops[0] = Target ID
      // Ops[1] = Decoration (Binding)
      // Ops[2] = LiteralNumber according to Decoration
      Ops.clear();

      Ops.push_back(ArgIDOp);

      Deco = spv::DecorationBinding;
      DecoOp = new SPIRVOperand(SPIRVOperandType::NUMBERID, Deco);
      Ops.push_back(DecoOp);

      LiteralNum.clear();
      LiteralNum.push_back(BindingIdx++);
      SPIRVOperand *Binding =
          new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, LiteralNum);
      Ops.push_back(Binding);

      SPIRVInstruction *BindDecoInst =
          new SPIRVInstruction(4, spv::OpDecorate, 0 /* No id */, Ops);
      SPIRVInstList.insert(DecoInsertPoint, BindDecoInst);

      // Handle image type argument.
      bool HasReadOnlyImageType = false;
      bool HasWriteOnlyImageType = false;
      if (PointerType *ArgPTy = dyn_cast<PointerType>(Arg.getType())) {
        if (StructType *STy = dyn_cast<StructType>(ArgPTy->getElementType())) {
          if (STy->isOpaque()) {
            if (STy->getName().equals("opencl.image2d_ro_t") ||
                STy->getName().equals("opencl.image3d_ro_t")) {
              HasReadOnlyImageType = true;
            } else if (STy->getName().equals("opencl.image2d_wo_t") ||
                       STy->getName().equals("opencl.image3d_wo_t")) {
              HasWriteOnlyImageType = true;
            }
          }
        }
      }

      if (HasReadOnlyImageType || HasWriteOnlyImageType) {
        // Ops[0] = Target ID
        // Ops[1] = Decoration (NonReadable or NonWritable)
        Ops.clear();

        ArgIDOp = new SPIRVOperand(SPIRVOperandType::NUMBERID, VMap[&Arg]);
        Ops.push_back(ArgIDOp);

        Deco = spv::DecorationNonReadable;
        if (HasReadOnlyImageType) {
          Deco = spv::DecorationNonWritable;
        }
        DecoOp = new SPIRVOperand(SPIRVOperandType::NUMBERID, Deco);
        Ops.push_back(DecoOp);

        DescDecoInst =
            new SPIRVInstruction(3, spv::OpDecorate, 0 /* No id */, Ops);
        SPIRVInstList.insert(DecoInsertPoint, DescDecoInst);
      }

      // Handle const address space.
      if (NewGV->getType()->getPointerAddressSpace() ==
          AddressSpace::Constant) {
        // Ops[0] = Target ID
        // Ops[1] = Decoration (NonWriteable)
        Ops.clear();

        Ops.push_back(ArgIDOp);

        Deco = spv::DecorationNonWritable;
        DecoOp = new SPIRVOperand(SPIRVOperandType::NUMBERID, Deco);
        Ops.push_back(DecoOp);

        BindDecoInst =
            new SPIRVInstruction(3, spv::OpDecorate, 0 /* No id */, Ops);
        SPIRVInstList.insert(DecoInsertPoint, BindDecoInst);
      }
    }
  }

  //
  // Generate OPFunction.
  //

  // FOps[0] : Result Type ID
  // FOps[1] : Function Control
  // FOps[2] : Function Type ID
  SPIRVOperandList FOps;

  // Find SPIRV instruction for return type.
  uint32_t RetTyID = lookupType(FTy->getReturnType());
  SPIRVOperand *RetTyOp = new SPIRVOperand(SPIRVOperandType::NUMBERID, RetTyID);
  FOps.push_back(RetTyOp);

  // Check function attributes for SPIRV Function Control.
  uint32_t FuncControl = spv::FunctionControlMaskNone;
  if (F.hasFnAttribute(Attribute::AlwaysInline)) {
    FuncControl |= spv::FunctionControlInlineMask;
  }
  if (F.hasFnAttribute(Attribute::NoInline)) {
    FuncControl |= spv::FunctionControlDontInlineMask;
  }
  // TODO: Check llvm attribute for Function Control Pure.
  if (F.hasFnAttribute(Attribute::ReadOnly)) {
    FuncControl |= spv::FunctionControlPureMask;
  }
  // TODO: Check llvm attribute for Function Control Const.
  if (F.hasFnAttribute(Attribute::ReadNone)) {
    FuncControl |= spv::FunctionControlConstMask;
  }

  SPIRVOperand *FunctionControlOp =
      new SPIRVOperand(SPIRVOperandType::NUMBERID, FuncControl);
  FOps.push_back(FunctionControlOp);

  uint32_t FTyID;
  if (F.getCallingConv() == CallingConv::SPIR_KERNEL) {
    SmallVector<Type *, 4> NewFuncParamTys;
    FunctionType *NewFTy =
        FunctionType::get(FTy->getReturnType(), NewFuncParamTys, false);
    FTyID = lookupType(NewFTy);
  } else {
    // Handle function with global constant parameters.
    if (GlobalConstFuncTyMap.count(FTy)) {
      FTyID = lookupType(GlobalConstFuncTyMap[FTy].first);
    } else {
      FTyID = lookupType(FTy);
    }
  }

  SPIRVOperand *FTyOp = new SPIRVOperand(SPIRVOperandType::NUMBERID, FTyID);
  FOps.push_back(FTyOp);

  if (F.getCallingConv() == CallingConv::SPIR_KERNEL) {
    EntryPoints.push_back(std::make_pair(&F, nextID));
  }

  VMap[&F] = nextID;

  // Generate SPIRV instruction for function.
  SPIRVInstruction *FuncInst =
      new SPIRVInstruction(5, spv::OpFunction, nextID++, FOps);
  SPIRVInstList.push_back(FuncInst);

  //
  // Generate OpFunctionParameter for Normal function.
  //

  if (F.getCallingConv() != CallingConv::SPIR_KERNEL) {
    // Iterate Argument for name instead of param type from function type.
    unsigned ArgIdx = 0;
    for (Argument &Arg : F.args()) {
      VMap[&Arg] = nextID;

      // ParamOps[0] : Result Type ID
      SPIRVOperandList ParamOps;

      // Find SPIRV instruction for parameter type.
      uint32_t ParamTyID = lookupType(Arg.getType());
      if (PointerType *PTy = dyn_cast<PointerType>(Arg.getType())) {
        if (GlobalConstFuncTyMap.count(FTy)) {
          if (ArgIdx == GlobalConstFuncTyMap[FTy].second) {
            Type *EleTy = PTy->getPointerElementType();
            Type *ArgTy =
                PointerType::get(EleTy, AddressSpace::ModuleScopePrivate);
            ParamTyID = lookupType(ArgTy);
            GlobalConstArgSet.insert(&Arg);
          }
        }
      }
      SPIRVOperand *ParamTyOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, ParamTyID);
      ParamOps.push_back(ParamTyOp);

      // Generate SPIRV instruction for parameter.
      SPIRVInstruction *ParamInst =
          new SPIRVInstruction(3, spv::OpFunctionParameter, nextID++, ParamOps);
      SPIRVInstList.push_back(ParamInst);

      ArgIdx++;
    }
  }
}

void SPIRVProducerPass::GenerateModuleInfo() {
  SPIRVInstructionList &SPIRVInstList = getSPIRVInstList();
  EntryPointVecType &EntryPoints = getEntryPointVec();
  ValueMapType &VMap = getValueMap();
  ValueList &EntryPointInterfaces = getEntryPointInterfacesVec();
  uint32_t &ExtInstImportID = getOpExtInstImportID();
  std::vector<uint32_t> &BuiltinDimVec = getBuiltinDimVec();

  // Set up insert point.
  auto InsertPoint = SPIRVInstList.begin();

  //
  // Generate OpCapability
  //
  // TODO: Which llvm information is mapped to SPIRV Capapbility?

  // Ops[0] = Capability
  SPIRVOperandList Ops;

  SPIRVInstruction *CapInst = new SPIRVInstruction(
      2, spv::OpCapability, 0 /* No id */,
      new SPIRVOperand(SPIRVOperandType::NUMBERID, spv::CapabilityShader));
  SPIRVInstList.insert(InsertPoint, CapInst);

  for (Type *Ty : getTypeList()) {
    // Find the i16 type.
    if (Ty->isIntegerTy(16)) {
      // Generate OpCapability for i16 type.
      SPIRVInstList.insert(
          InsertPoint,
          new SPIRVInstruction(2, spv::OpCapability, 0 /* No id */,
                               new SPIRVOperand(SPIRVOperandType::NUMBERID,
                                                spv::CapabilityInt16)));
    } else if (Ty->isIntegerTy(64)) {
      // Generate OpCapability for i64 type.
      SPIRVInstList.insert(
          InsertPoint,
          new SPIRVInstruction(2, spv::OpCapability, 0 /* No id */,
                               new SPIRVOperand(SPIRVOperandType::NUMBERID,
                                                spv::CapabilityInt64)));
    } else if (Ty->isHalfTy()) {
      // Generate OpCapability for half type.
      SPIRVInstList.insert(
          InsertPoint,
          new SPIRVInstruction(2, spv::OpCapability, 0 /* No id */,
                               new SPIRVOperand(SPIRVOperandType::NUMBERID,
                                                spv::CapabilityFloat16)));
    } else if (Ty->isDoubleTy()) {
      // Generate OpCapability for double type.
      SPIRVInstList.insert(
          InsertPoint,
          new SPIRVInstruction(2, spv::OpCapability, 0 /* No id */,
                               new SPIRVOperand(SPIRVOperandType::NUMBERID,
                                                spv::CapabilityFloat64)));
    } else if (auto *STy = dyn_cast<StructType>(Ty)) {
      if (STy->isOpaque()) {
        if (STy->getName().equals("opencl.image2d_ro_t") ||
            STy->getName().equals("opencl.image3d_ro_t")) {
          // Generate OpCapability for read only image type.
          SPIRVInstList.insert(
              InsertPoint,
              new SPIRVInstruction(
                  2, spv::OpCapability, 0 /* No id */,
                  new SPIRVOperand(
                      SPIRVOperandType::NUMBERID,
                      spv::CapabilityStorageImageReadWithoutFormat)));
        } else if (STy->getName().equals("opencl.image2d_wo_t") ||
                   STy->getName().equals("opencl.image3d_wo_t")) {
          // Generate OpCapability for write only image type.
          SPIRVInstList.insert(
              InsertPoint,
              new SPIRVInstruction(
                  2, spv::OpCapability, 0 /* No id */,
                  new SPIRVOperand(
                      SPIRVOperandType::NUMBERID,
                      spv::CapabilityStorageImageWriteWithoutFormat)));
        }
      }
    }
  }

  if (hasVariablePointers()) {
    //
    // Generate OpCapability and OpExtension
    //

    //
    // Generate OpCapability.
    //
    // Ops[0] = Capability
    //
    Ops.clear();

    Ops.push_back(new SPIRVOperand(SPIRVOperandType::NUMBERID,
                                   spv::CapabilityVariablePointers));

    SPIRVInstList.insert(InsertPoint, new SPIRVInstruction(2, spv::OpCapability,
                                                           0 /* No id */, Ops));

    //
    // Generate OpExtension.
    //
    // Ops[0] = Name (Literal String)
    //
    for (auto extension : {"SPV_KHR_storage_buffer_storage_class",
                           "SPV_KHR_variable_pointers"}) {
      Ops.clear();

      SPIRVOperand *Name =
          new SPIRVOperand(SPIRVOperandType::LITERAL_STRING, extension);
      Ops.push_back(Name);

      size_t NameWordSize = (Name->getLiteralStr().size() + 1) / 4;
      if ((Name->getLiteralStr().size() + 1) % 4) {
        NameWordSize += 1;
      }

      assert((NameWordSize + 1) < UINT16_MAX);
      uint16_t WordCount = static_cast<uint16_t>(1 + NameWordSize);

      SPIRVInstruction *ExtensionInst =
          new SPIRVInstruction(WordCount, spv::OpExtension, 0 /* No id */, Ops);
      SPIRVInstList.insert(InsertPoint, ExtensionInst);
    }
  }

  if (ExtInstImportID) {
    ++InsertPoint;
  }

  //
  // Generate OpMemoryModel
  //
  // Memory model for Vulkan will always be GLSL450.

  // Ops[0] = Addressing Model
  // Ops[1] = Memory Model
  Ops.clear();
  SPIRVOperand *AddrModel =
      new SPIRVOperand(SPIRVOperandType::NUMBERID, spv::AddressingModelLogical);
  Ops.push_back(AddrModel);

  SPIRVOperand *MemModel =
      new SPIRVOperand(SPIRVOperandType::NUMBERID, spv::MemoryModelGLSL450);
  Ops.push_back(MemModel);

  SPIRVInstruction *MemModelInst =
      new SPIRVInstruction(3, spv::OpMemoryModel, 0 /* No id */, Ops);
  SPIRVInstList.insert(InsertPoint, MemModelInst);

  //
  // Generate OpEntryPoint
  //
  for (auto EntryPoint : EntryPoints) {
    // Ops[0] = Execution Model
    // Ops[1] = EntryPoint ID
    // Ops[2] = Name (Literal String)
    // ...
    //
    // TODO: Do we need to consider Interface ID for forward references???
    Ops.clear();
    SPIRVOperand *ExecModel = new SPIRVOperand(SPIRVOperandType::NUMBERID,
                                               spv::ExecutionModelGLCompute);
    Ops.push_back(ExecModel);

    SPIRVOperand *EntryPointID =
        new SPIRVOperand(SPIRVOperandType::NUMBERID, EntryPoint.second);
    Ops.push_back(EntryPointID);

    SPIRVOperand *Name = new SPIRVOperand(SPIRVOperandType::LITERAL_STRING,
                                          EntryPoint.first->getName());
    Ops.push_back(Name);

    size_t NameWordSize = (Name->getLiteralStr().size() + 1) / 4;
    if ((Name->getLiteralStr().size() + 1) % 4) {
      NameWordSize += 1;
    }

    assert((3 + NameWordSize) < UINT16_MAX);
    uint16_t WordCount = static_cast<uint16_t>(3 + NameWordSize);

    for (Value *Interface : EntryPointInterfaces) {
      SPIRVOperand *GIDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, VMap[Interface]);
      Ops.push_back(GIDOp);
      WordCount++;
    }

    SPIRVInstruction *EntryPointInst =
        new SPIRVInstruction(WordCount, spv::OpEntryPoint, 0 /* No id */, Ops);
    SPIRVInstList.insert(InsertPoint, EntryPointInst);
  }

  for (auto EntryPoint : EntryPoints) {
    if (const MDNode *MD = dyn_cast<Function>(EntryPoint.first)
                               ->getMetadata("reqd_work_group_size")) {

      if (!BuiltinDimVec.empty()) {
        llvm_unreachable(
            "Kernels should have consistent work group size definition");
      }

      //
      // Generate OpExecutionMode
      //

      // Ops[0] = Entry Point ID
      // Ops[1] = Execution Mode
      // Ops[2] ... Ops[n] = Optional literals according to Execution Mode
      Ops.clear();
      SPIRVOperand *EntryPointID =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, EntryPoint.second);
      Ops.push_back(EntryPointID);

      SPIRVOperand *ExecMode = new SPIRVOperand(SPIRVOperandType::NUMBERID,
                                                spv::ExecutionModeLocalSize);
      Ops.push_back(ExecMode);

      uint32_t XDim = static_cast<uint32_t>(
          mdconst::extract<ConstantInt>(MD->getOperand(0))->getZExtValue());
      uint32_t YDim = static_cast<uint32_t>(
          mdconst::extract<ConstantInt>(MD->getOperand(1))->getZExtValue());
      uint32_t ZDim = static_cast<uint32_t>(
          mdconst::extract<ConstantInt>(MD->getOperand(2))->getZExtValue());

      std::vector<uint32_t> LiteralNum;
      LiteralNum.push_back(XDim);
      SPIRVOperand *XDimOp =
          new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, LiteralNum);
      Ops.push_back(XDimOp);

      LiteralNum.clear();
      LiteralNum.push_back(YDim);
      SPIRVOperand *YDimOp =
          new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, LiteralNum);
      Ops.push_back(YDimOp);

      LiteralNum.clear();
      LiteralNum.push_back(ZDim);
      SPIRVOperand *ZDimOp =
          new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, LiteralNum);
      Ops.push_back(ZDimOp);

      SPIRVInstruction *ExecModeInst =
          new SPIRVInstruction(static_cast<uint16_t>(1 + Ops.size()),
                               spv::OpExecutionMode, 0 /* No id */, Ops);
      SPIRVInstList.insert(InsertPoint, ExecModeInst);
    }
  }

  //
  // Generate OpSource.
  //
  // Ops[0] = SourceLanguage ID
  // Ops[1] = Version (LiteralNum)
  //
  Ops.clear();
  SPIRVOperand *SourceLanguage =
      new SPIRVOperand(SPIRVOperandType::NUMBERID, spv::SourceLanguageOpenCL_C);
  Ops.push_back(SourceLanguage);

  std::vector<uint32_t> LiteralNum;
  LiteralNum.push_back(120);
  SPIRVOperand *Version =
      new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, LiteralNum);
  Ops.push_back(Version);

  SPIRVInstruction *OpenSourceInst =
      new SPIRVInstruction(3, spv::OpSource, 0 /* No id */, Ops);
  SPIRVInstList.insert(InsertPoint, OpenSourceInst);

  if (!BuiltinDimVec.empty()) {
    //
    // Generate OpDecorates for x/y/z dimension.
    //
    // Ops[0] = Target ID
    // Ops[1] = Decoration (SpecId)
    // Ops[2] = Specialization Cosntant ID (Literal Number)

    // X Dimension
    Ops.clear();

    SPIRVOperand *TargetID =
        new SPIRVOperand(SPIRVOperandType::NUMBERID, BuiltinDimVec[0]);
    Ops.push_back(TargetID);

    SPIRVOperand *DecoOp =
        new SPIRVOperand(SPIRVOperandType::NUMBERID, spv::DecorationSpecId);
    Ops.push_back(DecoOp);

    LiteralNum.clear();
    LiteralNum.push_back(0);
    SPIRVOperand *XDim =
        new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, LiteralNum);
    Ops.push_back(XDim);

    SPIRVInstruction *XDimDecoInst =
        new SPIRVInstruction(4, spv::OpDecorate, 0 /* No id */, Ops);
    SPIRVInstList.insert(InsertPoint, XDimDecoInst);

    // Y Dimension
    Ops.clear();

    TargetID = new SPIRVOperand(SPIRVOperandType::NUMBERID, BuiltinDimVec[1]);
    Ops.push_back(TargetID);

    DecoOp =
        new SPIRVOperand(SPIRVOperandType::NUMBERID, spv::DecorationSpecId);
    Ops.push_back(DecoOp);

    LiteralNum.clear();
    LiteralNum.push_back(1);
    SPIRVOperand *YDim =
        new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, LiteralNum);
    Ops.push_back(YDim);

    SPIRVInstruction *YDimDecoInst =
        new SPIRVInstruction(4, spv::OpDecorate, 0 /* No id */, Ops);
    SPIRVInstList.insert(InsertPoint, YDimDecoInst);

    // Z Dimension
    Ops.clear();

    TargetID = new SPIRVOperand(SPIRVOperandType::NUMBERID, BuiltinDimVec[2]);
    Ops.push_back(TargetID);

    DecoOp =
        new SPIRVOperand(SPIRVOperandType::NUMBERID, spv::DecorationSpecId);
    Ops.push_back(DecoOp);

    LiteralNum.clear();
    LiteralNum.push_back(2);
    SPIRVOperand *ZDim =
        new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, LiteralNum);
    Ops.push_back(ZDim);

    SPIRVInstruction *ZDimDecoInst =
        new SPIRVInstruction(4, spv::OpDecorate, 0 /* No id */, Ops);
    SPIRVInstList.insert(InsertPoint, ZDimDecoInst);
  }
}

void SPIRVProducerPass::GenerateInstForArg(Function &F) {
  SPIRVInstructionList &SPIRVInstList = getSPIRVInstList();
  ValueMapType &VMap = getValueMap();
  Module *Module = F.getParent();
  LLVMContext &Context = Module->getContext();
  ValueToValueMapTy &ArgGVMap = getArgumentGVMap();

  for (Argument &Arg : F.args()) {
    if (Arg.use_empty()) {
      continue;
    }

    // Check the type of users of arguments.
    bool HasOnlyGEPUse = true;
    for (auto *U : Arg.users()) {
      if (!isa<GetElementPtrInst>(U) && isa<Instruction>(U)) {
        HasOnlyGEPUse = false;
        break;
      }
    }

    Type *ArgTy = Arg.getType();

    if (PointerType *PTy = dyn_cast<PointerType>(ArgTy)) {
      if (StructType *STy = dyn_cast<StructType>(PTy->getElementType())) {
        if (STy->isOpaque()) {
          // Generate OpLoad for sampler and image types.
          if (STy->getName().equals("opencl.sampler_t") ||
              STy->getName().equals("opencl.image2d_ro_t") ||
              STy->getName().equals("opencl.image2d_wo_t") ||
              STy->getName().equals("opencl.image3d_ro_t") ||
              STy->getName().equals("opencl.image3d_wo_t")) {
            //
            // Generate OpLoad.
            //
            // Ops[0] = Result Type ID
            // Ops[1] = Pointer ID
            // Ops[2] ... Ops[n] = Optional Memory Access
            //
            // TODO: Do we need to implement Optional Memory Access???
            SPIRVOperandList Ops;

            // Use type with address space modified.
            ArgTy = ArgGVMap[&Arg]->getType()->getPointerElementType();

            uint32_t ResTyID = lookupType(ArgTy);
            SPIRVOperand *ResTyIDOp =
                new SPIRVOperand(SPIRVOperandType::NUMBERID, ResTyID);
            Ops.push_back(ResTyIDOp);

            uint32_t PointerID = VMap[&Arg];
            SPIRVOperand *PointerIDOp =
                new SPIRVOperand(SPIRVOperandType::NUMBERID, PointerID);
            Ops.push_back(PointerIDOp);

            VMap[&Arg] = nextID;
            SPIRVInstruction *Inst =
                new SPIRVInstruction(4, spv::OpLoad, nextID++, Ops);
            SPIRVInstList.push_back(Inst);
            continue;
          }
        }
      }

      if (!HasOnlyGEPUse) {
        //
        // Generate OpAccessChain.
        //
        // Ops[0] = Result Type ID
        // Ops[1] = Base ID
        // Ops[2] ... Ops[n] = Indexes ID
        SPIRVOperandList Ops;

        uint32_t ResTyID = lookupType(ArgTy);
        if (!isa<PointerType>(ArgTy)) {
          ResTyID = lookupType(PointerType::get(ArgTy, AddressSpace::Global));
        }
        SPIRVOperand *ResTyOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, ResTyID);
        Ops.push_back(ResTyOp);

        uint32_t BaseID = VMap[&Arg];
        SPIRVOperand *BaseOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, BaseID);
        Ops.push_back(BaseOp);

        Type *IdxTy = Type::getInt32Ty(Context);
        uint32_t IndexID = VMap[ConstantInt::get(IdxTy, 0)];
        SPIRVOperand *IndexIDOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, IndexID);
        Ops.push_back(IndexIDOp);
        Ops.push_back(IndexIDOp);

        // Generate SPIRV instruction for argument.
        VMap[&Arg] = nextID;
        SPIRVInstruction *ArgInst =
            new SPIRVInstruction(6, spv::OpAccessChain, nextID++, Ops);
        SPIRVInstList.push_back(ArgInst);
      } else {
        // For GEP uses, generate OpAccessChain with folding GEP ahead of GEP.
        // Nothing to do here.
      }
    } else {
      //
      // Generate OpAccessChain and OpLoad for non-pointer type argument.
      //

      //
      // Generate OpAccessChain.
      //
      // Ops[0] = Result Type ID
      // Ops[1] = Base ID
      // Ops[2] ... Ops[n] = Indexes ID
      SPIRVOperandList Ops;

      uint32_t ResTyID = lookupType(ArgTy);
      if (!isa<PointerType>(ArgTy)) {
        ResTyID = lookupType(PointerType::get(ArgTy, AddressSpace::Global));
      }
      SPIRVOperand *ResTyIDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, ResTyID);
      Ops.push_back(ResTyIDOp);

      uint32_t BaseID = VMap[&Arg];
      SPIRVOperand *BaseOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, BaseID);
      Ops.push_back(BaseOp);

      Type *IdxTy = Type::getInt32Ty(Context);
      uint32_t IndexID = VMap[ConstantInt::get(IdxTy, 0)];
      SPIRVOperand *IndexIDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, IndexID);
      Ops.push_back(IndexIDOp);

      // Generate SPIRV instruction for argument.
      uint32_t PointerID = nextID;
      VMap[&Arg] = nextID;
      SPIRVInstruction *ArgInst =
          new SPIRVInstruction(5, spv::OpAccessChain, nextID++, Ops);
      SPIRVInstList.push_back(ArgInst);

      //
      // Generate OpLoad.
      //

      // Ops[0] = Result Type ID
      // Ops[1] = Pointer ID
      // Ops[2] ... Ops[n] = Optional Memory Access
      //
      // TODO: Do we need to implement Optional Memory Access???
      Ops.clear();

      ResTyID = lookupType(ArgTy);
      ResTyIDOp = new SPIRVOperand(SPIRVOperandType::NUMBERID, ResTyID);
      Ops.push_back(ResTyIDOp);

      SPIRVOperand *PointerIDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, PointerID);
      Ops.push_back(PointerIDOp);

      VMap[&Arg] = nextID;
      SPIRVInstruction *Inst =
          new SPIRVInstruction(4, spv::OpLoad, nextID++, Ops);
      SPIRVInstList.push_back(Inst);
    }
  }
}

void SPIRVProducerPass::GenerateFuncBody(Function &F) {
  SPIRVInstructionList &SPIRVInstList = getSPIRVInstList();
  ValueMapType &VMap = getValueMap();

  bool IsKernel = false;
  if (F.getCallingConv() == CallingConv::SPIR_KERNEL) {
    IsKernel = true;
  }

  for (BasicBlock &BB : F) {
    // Register BasicBlock to ValueMap.
    VMap[&BB] = nextID;

    //
    // Generate OpLabel for Basic Block.
    //
    SPIRVOperandList Ops;
    SPIRVInstruction *Inst =
        new SPIRVInstruction(2, spv::OpLabel, nextID++, Ops);
    SPIRVInstList.push_back(Inst);

    // OpVariable instructions must come first.
    for (Instruction &I : BB) {
      if (isa<AllocaInst>(I)) {
        GenerateInstruction(I);
      }
    }

    if (&BB == &F.getEntryBlock() && IsKernel) {
      GenerateInstForArg(F);
    }

    for (Instruction &I : BB) {
      if (!isa<AllocaInst>(I)) {
        GenerateInstruction(I);
      }
    }
  }
}

spv::Op SPIRVProducerPass::GetSPIRVCmpOpcode(CmpInst *I) {
  const std::map<CmpInst::Predicate, spv::Op> Map = {
      {CmpInst::ICMP_EQ, spv::OpIEqual},
      {CmpInst::ICMP_NE, spv::OpINotEqual},
      {CmpInst::ICMP_UGT, spv::OpUGreaterThan},
      {CmpInst::ICMP_UGE, spv::OpUGreaterThanEqual},
      {CmpInst::ICMP_ULT, spv::OpULessThan},
      {CmpInst::ICMP_ULE, spv::OpULessThanEqual},
      {CmpInst::ICMP_SGT, spv::OpSGreaterThan},
      {CmpInst::ICMP_SGE, spv::OpSGreaterThanEqual},
      {CmpInst::ICMP_SLT, spv::OpSLessThan},
      {CmpInst::ICMP_SLE, spv::OpSLessThanEqual},
      {CmpInst::FCMP_OEQ, spv::OpFOrdEqual},
      {CmpInst::FCMP_OGT, spv::OpFOrdGreaterThan},
      {CmpInst::FCMP_OGE, spv::OpFOrdGreaterThanEqual},
      {CmpInst::FCMP_OLT, spv::OpFOrdLessThan},
      {CmpInst::FCMP_OLE, spv::OpFOrdLessThanEqual},
      {CmpInst::FCMP_ONE, spv::OpFOrdNotEqual},
      {CmpInst::FCMP_UEQ, spv::OpFUnordEqual},
      {CmpInst::FCMP_UGT, spv::OpFUnordGreaterThan},
      {CmpInst::FCMP_UGE, spv::OpFUnordGreaterThanEqual},
      {CmpInst::FCMP_ULT, spv::OpFUnordLessThan},
      {CmpInst::FCMP_ULE, spv::OpFUnordLessThanEqual},
      {CmpInst::FCMP_UNE, spv::OpFUnordNotEqual}};

  assert(0 != Map.count(I->getPredicate()));

  return Map.at(I->getPredicate());
}

spv::Op SPIRVProducerPass::GetSPIRVCastOpcode(Instruction &I) {
  const std::map<unsigned, spv::Op> Map{
      {Instruction::Trunc, spv::OpUConvert},
      {Instruction::ZExt, spv::OpUConvert},
      {Instruction::SExt, spv::OpSConvert},
      {Instruction::FPToUI, spv::OpConvertFToU},
      {Instruction::FPToSI, spv::OpConvertFToS},
      {Instruction::UIToFP, spv::OpConvertUToF},
      {Instruction::SIToFP, spv::OpConvertSToF},
      {Instruction::FPTrunc, spv::OpFConvert},
      {Instruction::FPExt, spv::OpFConvert},
      {Instruction::BitCast, spv::OpBitcast}};

  assert(0 != Map.count(I.getOpcode()));

  return Map.at(I.getOpcode());
}

spv::Op SPIRVProducerPass::GetSPIRVBinaryOpcode(Instruction &I) {
  if (I.getType()->isIntegerTy(1)) {
    switch (I.getOpcode()) {
    default:
      break;
    case Instruction::Or:
      return spv::OpLogicalOr;
    case Instruction::And:
      return spv::OpLogicalAnd;
    case Instruction::Xor:
      return spv::OpLogicalNotEqual;
    }
  }

  const std::map<unsigned, spv::Op> Map {
      {Instruction::Add, spv::OpIAdd},
      {Instruction::FAdd, spv::OpFAdd},
      {Instruction::Sub, spv::OpISub},
      {Instruction::FSub, spv::OpFSub},
      {Instruction::Mul, spv::OpIMul},
      {Instruction::FMul, spv::OpFMul},
      {Instruction::UDiv, spv::OpUDiv},
      {Instruction::SDiv, spv::OpSDiv},
      {Instruction::FDiv, spv::OpFDiv},
      {Instruction::URem, spv::OpUMod},
      {Instruction::SRem, spv::OpSRem},
      {Instruction::FRem, spv::OpFRem},
      {Instruction::Or, spv::OpBitwiseOr},
      {Instruction::Xor, spv::OpBitwiseXor},
      {Instruction::And, spv::OpBitwiseAnd},
      {Instruction::Shl, spv::OpShiftLeftLogical},
      {Instruction::LShr, spv::OpShiftRightLogical},
      {Instruction::AShr, spv::OpShiftRightArithmetic}};

  assert(0 != Map.count(I.getOpcode()));

  return Map.at(I.getOpcode());
}

void SPIRVProducerPass::GenerateInstruction(Instruction &I) {
  SPIRVInstructionList &SPIRVInstList = getSPIRVInstList();
  ValueMapType &VMap = getValueMap();
  ValueToValueMapTy &ArgGVMap = getArgumentGVMap();
  ValueMapType &ArgGVIDMap = getArgumentGVIDMap();
  DeferredInstVecType &DeferredInsts = getDeferredInstVec();
  LLVMContext &Context = I.getParent()->getParent()->getParent()->getContext();

  // Register Instruction to ValueMap.
  if (0 == VMap[&I]) {
    VMap[&I] = nextID;
  }

  switch (I.getOpcode()) {
  default: {
    if (Instruction::isCast(I.getOpcode())) {
      //
      // Generate SPIRV instructions for cast operators.
      //

      auto OpTy = I.getOperand(0)->getType();
      // Handle zext, sext and uitofp with i1 type specially.
      if ((I.getOpcode() == Instruction::ZExt ||
           I.getOpcode() == Instruction::SExt ||
           I.getOpcode() == Instruction::UIToFP) &&
          (OpTy->isIntegerTy(1) ||
           (OpTy->isVectorTy() &&
            OpTy->getVectorElementType()->isIntegerTy(1)))) {
        //
        // Generate OpSelect.
        //

        // Ops[0] = Result Type ID
        // Ops[1] = Condition ID
        // Ops[2] = True Constant ID
        // Ops[3] = False Constant ID
        SPIRVOperandList Ops;

        uint32_t ResTyID = lookupType(I.getType());
        SPIRVOperand *ResTyIDOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, ResTyID);
        Ops.push_back(ResTyIDOp);

        // TODO: zext's first operand should be compare instructions???
        uint32_t CondID = VMap[I.getOperand(0)];
        SPIRVOperand *CondIDOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, CondID);
        Ops.push_back(CondIDOp);

        uint32_t TrueID = 0;
        if (I.getOpcode() == Instruction::ZExt) {
          APInt One(32, 1);
          TrueID = VMap[Constant::getIntegerValue(I.getType(), One)];
        } else if (I.getOpcode() == Instruction::SExt) {
          APInt MinusOne(32, UINT64_MAX, true);
          TrueID = VMap[Constant::getIntegerValue(I.getType(), MinusOne)];
        } else {
          TrueID = VMap[ConstantFP::get(Context, APFloat(1.0f))];
        }
        SPIRVOperand *TrueIDOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, TrueID);
        Ops.push_back(TrueIDOp);

        uint32_t FalseID = 0;
        if (I.getOpcode() == Instruction::ZExt) {
          FalseID = VMap[Constant::getNullValue(I.getType())];
        } else if (I.getOpcode() == Instruction::SExt) {
          FalseID = VMap[Constant::getNullValue(I.getType())];
        } else {
          FalseID = VMap[ConstantFP::get(Context, APFloat(0.0f))];
        }
        SPIRVOperand *FalseIDOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, FalseID);
        Ops.push_back(FalseIDOp);

        SPIRVInstruction *Inst =
            new SPIRVInstruction(6, spv::OpSelect, nextID++, Ops);
        SPIRVInstList.push_back(Inst);
      } else {
        // Ops[0] = Result Type ID
        // Ops[1] = Source Value ID
        SPIRVOperandList Ops;

        uint32_t ResTyID = lookupType(I.getType());
        SPIRVOperand *ResTyIDOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, ResTyID);
        Ops.push_back(ResTyIDOp);

        uint32_t SrcID = VMap[I.getOperand(0)];
        SPIRVOperand *SrcIDOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, SrcID);
        Ops.push_back(SrcIDOp);

        SPIRVInstruction *Inst =
            new SPIRVInstruction(4, GetSPIRVCastOpcode(I), nextID++, Ops);
        SPIRVInstList.push_back(Inst);
      }
    } else if (isa<BinaryOperator>(I)) {
      //
      // Generate SPIRV instructions for binary operators.
      //

      // Handle xor with i1 type specially.
      if (I.getOpcode() == Instruction::Xor &&
          I.getType() == Type::getInt1Ty(Context) &&
          (isa<Constant>(I.getOperand(0)) || isa<Constant>(I.getOperand(1)))) {
        //
        // Generate OpLogicalNot.
        //
        // Ops[0] = Result Type ID
        // Ops[1] = Operand
        SPIRVOperandList Ops;

        uint32_t ResTyID = lookupType(I.getType());
        SPIRVOperand *ResTyIDOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, ResTyID);
        Ops.push_back(ResTyIDOp);

        Value *CondV = I.getOperand(0);
        if (isa<Constant>(I.getOperand(0))) {
          CondV = I.getOperand(1);
        }
        uint32_t CondID = VMap[CondV];
        SPIRVOperand *CondIDOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, CondID);
        Ops.push_back(CondIDOp);

        SPIRVInstruction *Inst =
            new SPIRVInstruction(4, spv::OpLogicalNot, nextID++, Ops);
        SPIRVInstList.push_back(Inst);
      } else {
        // Ops[0] = Result Type ID
        // Ops[1] = Operand 0
        // Ops[2] = Operand 1
        SPIRVOperandList Ops;

        uint32_t ResTyID = lookupType(I.getType());
        SPIRVOperand *ResTyIDOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, ResTyID);
        Ops.push_back(ResTyIDOp);

        uint32_t Op0ID = VMap[I.getOperand(0)];
        SPIRVOperand *Op0IDOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, Op0ID);
        Ops.push_back(Op0IDOp);

        uint32_t Op1ID = VMap[I.getOperand(1)];
        SPIRVOperand *Op1IDOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, Op1ID);
        Ops.push_back(Op1IDOp);

        SPIRVInstruction *Inst =
            new SPIRVInstruction(5, GetSPIRVBinaryOpcode(I), nextID++, Ops);
        SPIRVInstList.push_back(Inst);
      }
    } else {
      I.print(errs());
      llvm_unreachable("Unsupported instruction???");
    }
    break;
  }
  case Instruction::GetElementPtr: {
    auto &GlobalConstArgSet = getGlobalConstArgSet();

    //
    // Generate OpAccessChain.
    //
    GetElementPtrInst *GEP = cast<GetElementPtrInst>(&I);

    //
    // Generate OpAccessChain.
    //

    // Ops[0] = Result Type ID
    // Ops[1] = Base ID
    // Ops[2] ... Ops[n] = Indexes ID
    SPIRVOperandList Ops;

    PointerType* ResultType = cast<PointerType>(GEP->getType());
    if (GEP->getPointerAddressSpace() == AddressSpace::ModuleScopePrivate ||
        GlobalConstArgSet.count(GEP->getPointerOperand())) {
      // Use pointer type with private address space for global constant.
      Type *EleTy = I.getType()->getPointerElementType();
      ResultType = PointerType::get(EleTy, AddressSpace::ModuleScopePrivate);
    }
    const uint32_t ResTyID = lookupType(ResultType);
    SPIRVOperand *ResTyIDOp =
        new SPIRVOperand(SPIRVOperandType::NUMBERID, ResTyID);
    Ops.push_back(ResTyIDOp);

    // Check whether GEP's pointer operand is pointer argument.
    bool HasArgBasePointer = false;
    for (auto ArgGV : ArgGVMap) {
      if (ArgGV.first == GEP->getPointerOperand()) {
        if (isa<PointerType>(ArgGV.first->getType())) {
          HasArgBasePointer = true;
        } else {
          llvm_unreachable(
              "GEP's pointer operand is argument of non-poninter type???");
        }
      }
    }

    uint32_t BaseID;
    if (HasArgBasePointer) {
      // Point to global variable for argument directly.
      BaseID = ArgGVIDMap[GEP->getPointerOperand()];
    } else {
      BaseID = VMap[GEP->getPointerOperand()];
    }

    SPIRVOperand *BaseIDOp =
        new SPIRVOperand(SPIRVOperandType::NUMBERID, BaseID);
    Ops.push_back(BaseIDOp);

    uint16_t WordCount = 4;

    if (HasArgBasePointer) {
      // If GEP's pointer operand is argument, add one more index for struct
      // type to wrap up argument type.
      Type *IdxTy = Type::getInt32Ty(Context);
      uint32_t IndexID = VMap[ConstantInt::get(IdxTy, 0)];
      SPIRVOperand *IndexIDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, IndexID);
      Ops.push_back(IndexIDOp);

      WordCount++;
    }

    //
    // Follows below rules for gep.
    //
    // 1. If gep's first index is 0 and gep's base is not kernel function's
    //    argument, generate OpAccessChain and ignore gep's first index.
    // 2. If gep's first index is not 0, generate OpPtrAccessChain and use gep's
    //    first index.
    // 3. If gep's first index is not constant, generate OpPtrAccessChain and
    //    use gep's first index.
    // 4. If it is not above case 1, 2 and 3, generate OpAccessChain and use
    //    gep's first index.
    //
    spv::Op Opcode = spv::OpAccessChain;
    unsigned offset = 0;
    if (ConstantInt *CstInt = dyn_cast<ConstantInt>(GEP->getOperand(1))) {
      if (CstInt->getZExtValue() == 0 && !HasArgBasePointer) {
        offset = 1;
      } else if (CstInt->getZExtValue() != 0 && !HasArgBasePointer) {
        Opcode = spv::OpPtrAccessChain;
      }
    } else if (!HasArgBasePointer) {
      Opcode = spv::OpPtrAccessChain;
    }

    if (Opcode == spv::OpPtrAccessChain) {
      setVariablePointers(true);
      // Do we need to generate ArrayStride?  Check against the GEP result type
      // rather than the pointer type of the base because when indexing into
      // an OpenCL program-scope constant, we'll swap out the LLVM base pointer
      // for something else in the SPIR-V.
      // E.g. see test/PointerAccessChain/pointer_index_is_constant_1.cl
      if (GetStorageClass(ResultType->getAddressSpace()) ==
          spv::StorageClassStorageBuffer) {
        // Save the need to generate an ArrayStride decoration.  But defer
        // generation until later, so we only make one decoration.
        getPointerTypesNeedingArrayStride().insert(ResultType);
      }
    }

    for (auto II = GEP->idx_begin() + offset; II != GEP->idx_end(); II++) {
      uint32_t IndexID = VMap[*II];
      SPIRVOperand *IndexIDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, IndexID);
      Ops.push_back(IndexIDOp);

      WordCount++;
    }

    SPIRVInstruction *Inst =
        new SPIRVInstruction(WordCount, Opcode, nextID++, Ops);
    SPIRVInstList.push_back(Inst);
    break;
  }
  case Instruction::ExtractValue: {
    ExtractValueInst *EVI = cast<ExtractValueInst>(&I);
    // Ops[0] = Result Type ID
    // Ops[1] = Composite ID
    // Ops[2] ... Ops[n] = Indexes (Literal Number)
    SPIRVOperandList Ops;

    uint32_t ResTyID = lookupType(I.getType());
    SPIRVOperand *ResTyIDOp =
        new SPIRVOperand(SPIRVOperandType::NUMBERID, ResTyID);
    Ops.push_back(ResTyIDOp);

    uint32_t CompositeID = VMap[EVI->getAggregateOperand()];
    SPIRVOperand *CompositeIDOp =
        new SPIRVOperand(SPIRVOperandType::NUMBERID, CompositeID);
    Ops.push_back(CompositeIDOp);

    for (auto &Index : EVI->indices()) {
      Ops.push_back(new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, Index));
    }

    uint16_t WordCount = static_cast<uint16_t>(2 + Ops.size());
    SPIRVInstruction *Inst =
        new SPIRVInstruction(WordCount, spv::OpCompositeExtract, nextID++, Ops);
    SPIRVInstList.push_back(Inst);
    break;
  }
  case Instruction::InsertValue: {
    InsertValueInst *IVI = cast<InsertValueInst>(&I);
    // Ops[0] = Result Type ID
    // Ops[1] = Object ID
    // Ops[2] = Composite ID
    // Ops[3] ... Ops[n] = Indexes (Literal Number)
    SPIRVOperandList Ops;

    uint32_t ResTyID = lookupType(I.getType());
    Ops.push_back(new SPIRVOperand(SPIRVOperandType::NUMBERID, ResTyID));

    uint32_t ObjectID = VMap[IVI->getInsertedValueOperand()];
    Ops.push_back(new SPIRVOperand(SPIRVOperandType::NUMBERID, ObjectID));

    uint32_t CompositeID = VMap[IVI->getAggregateOperand()];
    Ops.push_back(new SPIRVOperand(SPIRVOperandType::NUMBERID, CompositeID));

    for (auto &Index : IVI->indices()) {
      Ops.push_back(new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, Index));
    }

    uint16_t WordCount = static_cast<uint16_t>(2 + Ops.size());
    SPIRVInstruction *Inst =
        new SPIRVInstruction(WordCount, spv::OpCompositeInsert, nextID++, Ops);
    SPIRVInstList.push_back(Inst);
    break;
  }
  case Instruction::Select: {
    //
    // Generate OpSelect.
    //

    // Ops[0] = Result Type ID
    // Ops[1] = Condition ID
    // Ops[2] = True Constant ID
    // Ops[3] = False Constant ID
    SPIRVOperandList Ops;

    // Find SPIRV instruction for parameter type.
    auto Ty = I.getType();
    if (Ty->isPointerTy()) {
      auto PointeeTy = Ty->getPointerElementType();
      if (PointeeTy->isStructTy() &&
          dyn_cast<StructType>(PointeeTy)->isOpaque()) {
        Ty = PointeeTy;
      }
    }

    uint32_t ResTyID = lookupType(Ty);
    SPIRVOperand *ResTyIDOp =
        new SPIRVOperand(SPIRVOperandType::NUMBERID, ResTyID);
    Ops.push_back(ResTyIDOp);

    uint32_t CondID = VMap[I.getOperand(0)];
    SPIRVOperand *CondIDOp =
        new SPIRVOperand(SPIRVOperandType::NUMBERID, CondID);
    Ops.push_back(CondIDOp);

    uint32_t TrueID = VMap[I.getOperand(1)];
    SPIRVOperand *TrueIDOp =
        new SPIRVOperand(SPIRVOperandType::NUMBERID, TrueID);
    Ops.push_back(TrueIDOp);

    uint32_t FalseID = VMap[I.getOperand(2)];
    SPIRVOperand *FalseIDOp =
        new SPIRVOperand(SPIRVOperandType::NUMBERID, FalseID);
    Ops.push_back(FalseIDOp);

    SPIRVInstruction *Inst =
        new SPIRVInstruction(6, spv::OpSelect, nextID++, Ops);
    SPIRVInstList.push_back(Inst);
    break;
  }
  case Instruction::ExtractElement: {
    // Handle <4 x i8> type manually.
    Type *CompositeTy = I.getOperand(0)->getType();
    if (is4xi8vec(CompositeTy)) {
      //
      // Generate OpShiftRightLogical and OpBitwiseAnd for extractelement with
      // <4 x i8>.
      //

      //
      // Generate OpShiftRightLogical
      //
      // Ops[0] = Result Type ID
      // Ops[1] = Operand 0
      // Ops[2] = Operand 1
      //
      SPIRVOperandList Ops;

      uint32_t ResTyID = lookupType(CompositeTy);
      SPIRVOperand *ResTyIDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, ResTyID);
      Ops.push_back(ResTyIDOp);

      uint32_t Op0ID = VMap[I.getOperand(0)];
      SPIRVOperand *Op0IDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, Op0ID);
      Ops.push_back(Op0IDOp);

      uint32_t Op1ID = 0;
      if (ConstantInt *CI = dyn_cast<ConstantInt>(I.getOperand(1))) {
        // Handle constant index.
        uint64_t Idx = CI->getZExtValue();
        Value *ShiftAmount =
            ConstantInt::get(Type::getInt32Ty(Context), Idx * 8);
        Op1ID = VMap[ShiftAmount];
      } else {
        // Handle variable index.
        SPIRVOperandList TmpOps;

        uint32_t TmpResTyID = lookupType(Type::getInt32Ty(Context));
        SPIRVOperand *TmpResTyIDOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, TmpResTyID);
        TmpOps.push_back(TmpResTyIDOp);

        uint32_t IdxID = VMap[I.getOperand(1)];
        SPIRVOperand *TmpOp0IDOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, IdxID);
        TmpOps.push_back(TmpOp0IDOp);

        ConstantInt *Cst8 = ConstantInt::get(Type::getInt32Ty(Context), 8);
        uint32_t Cst8ID = VMap[Cst8];
        SPIRVOperand *TmpOp1IDOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, Cst8ID);
        TmpOps.push_back(TmpOp1IDOp);

        Op1ID = nextID;

        SPIRVInstruction *TmpInst =
            new SPIRVInstruction(5, spv::OpIMul, nextID++, TmpOps);
        SPIRVInstList.push_back(TmpInst);
      }
      SPIRVOperand *Op1IDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, Op1ID);
      Ops.push_back(Op1IDOp);

      uint32_t ShiftID = nextID;

      SPIRVInstruction *Inst =
          new SPIRVInstruction(5, spv::OpShiftRightLogical, nextID++, Ops);
      SPIRVInstList.push_back(Inst);

      //
      // Generate OpBitwiseAnd
      //
      // Ops[0] = Result Type ID
      // Ops[1] = Operand 0
      // Ops[2] = Operand 1
      //
      Ops.clear();

      ResTyID = lookupType(CompositeTy);
      ResTyIDOp = new SPIRVOperand(SPIRVOperandType::NUMBERID, ResTyID);
      Ops.push_back(ResTyIDOp);

      Op0ID = ShiftID;
      Op0IDOp = new SPIRVOperand(SPIRVOperandType::NUMBERID, Op0ID);
      Ops.push_back(Op0IDOp);

      Constant *CstFF = ConstantInt::get(Type::getInt32Ty(Context), 0xFF);
      Op1ID = VMap[CstFF];
      Op1IDOp = new SPIRVOperand(SPIRVOperandType::NUMBERID, Op1ID);
      Ops.push_back(Op1IDOp);

      Inst = new SPIRVInstruction(5, spv::OpBitwiseAnd, nextID++, Ops);
      SPIRVInstList.push_back(Inst);
      break;
    }

    // Ops[0] = Result Type ID
    // Ops[1] = Composite ID
    // Ops[2] ... Ops[n] = Indexes (Literal Number)
    SPIRVOperandList Ops;

    uint32_t ResTyID = lookupType(I.getType());
    SPIRVOperand *ResTyIDOp =
        new SPIRVOperand(SPIRVOperandType::NUMBERID, ResTyID);
    Ops.push_back(ResTyIDOp);

    uint32_t CompositeID = VMap[I.getOperand(0)];
    SPIRVOperand *CompositeIDOp =
        new SPIRVOperand(SPIRVOperandType::NUMBERID, CompositeID);
    Ops.push_back(CompositeIDOp);

    spv::Op Opcode = spv::OpCompositeExtract;
    if (const ConstantInt *CI = dyn_cast<ConstantInt>(I.getOperand(1))) {
      std::vector<uint32_t> LiteralNum;
      assert(CI->getZExtValue() < UINT32_MAX);
      LiteralNum.push_back(static_cast<uint32_t>(CI->getZExtValue()));
      SPIRVOperand *Indexes =
          new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, LiteralNum);
      Ops.push_back(Indexes);
    } else {
      uint32_t IndexID = VMap[I.getOperand(1)];
      SPIRVOperand *IndexIDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, IndexID);
      Ops.push_back(IndexIDOp);
      Opcode = spv::OpVectorExtractDynamic;
    }

    uint16_t WordCount = 5;
    SPIRVInstruction *Inst =
        new SPIRVInstruction(WordCount, Opcode, nextID++, Ops);
    SPIRVInstList.push_back(Inst);
    break;
  }
  case Instruction::InsertElement: {
    // Handle <4 x i8> type manually.
    Type *CompositeTy = I.getOperand(0)->getType();
    if (is4xi8vec(CompositeTy)) {
      Constant *CstFF = ConstantInt::get(Type::getInt32Ty(Context), 0xFF);
      uint32_t CstFFID = VMap[CstFF];

      uint32_t ShiftAmountID = 0;
      if (ConstantInt *CI = dyn_cast<ConstantInt>(I.getOperand(2))) {
        // Handle constant index.
        uint64_t Idx = CI->getZExtValue();
        Value *ShiftAmount =
            ConstantInt::get(Type::getInt32Ty(Context), Idx * 8);
        ShiftAmountID = VMap[ShiftAmount];
      } else {
        // Handle variable index.
        SPIRVOperandList TmpOps;

        uint32_t TmpResTyID = lookupType(Type::getInt32Ty(Context));
        SPIRVOperand *TmpResTyIDOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, TmpResTyID);
        TmpOps.push_back(TmpResTyIDOp);

        uint32_t IdxID = VMap[I.getOperand(2)];
        SPIRVOperand *TmpOp0IDOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, IdxID);
        TmpOps.push_back(TmpOp0IDOp);

        ConstantInt *Cst8 = ConstantInt::get(Type::getInt32Ty(Context), 8);
        uint32_t Cst8ID = VMap[Cst8];
        SPIRVOperand *TmpOp1IDOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, Cst8ID);
        TmpOps.push_back(TmpOp1IDOp);

        ShiftAmountID = nextID;

        SPIRVInstruction *TmpInst =
            new SPIRVInstruction(5, spv::OpIMul, nextID++, TmpOps);
        SPIRVInstList.push_back(TmpInst);
      }

      //
      // Generate mask operations.
      //

      // ShiftLeft mask according to index of insertelement.
      SPIRVOperandList Ops;

      uint32_t ResTyID = lookupType(CompositeTy);
      SPIRVOperand *ResTyIDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, ResTyID);
      Ops.push_back(ResTyIDOp);

      uint32_t Op0ID = CstFFID;
      SPIRVOperand *Op0IDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, Op0ID);
      Ops.push_back(Op0IDOp);

      uint32_t Op1ID = ShiftAmountID;
      SPIRVOperand *Op1IDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, Op1ID);
      Ops.push_back(Op1IDOp);

      uint32_t MaskID = nextID;

      SPIRVInstruction *Inst =
          new SPIRVInstruction(5, spv::OpShiftLeftLogical, nextID++, Ops);
      SPIRVInstList.push_back(Inst);

      // Inverse mask.
      Ops.clear();

      Ops.push_back(ResTyIDOp);

      Op0ID = MaskID;
      Op0IDOp = new SPIRVOperand(SPIRVOperandType::NUMBERID, Op0ID);
      Ops.push_back(Op0IDOp);

      uint32_t InvMaskID = nextID;

      Inst = new SPIRVInstruction(4, spv::OpLogicalNot, nextID++, Ops);
      SPIRVInstList.push_back(Inst);

      // Apply mask.
      Ops.clear();

      Ops.push_back(ResTyIDOp);

      Op0ID = VMap[I.getOperand(0)];
      Op0IDOp = new SPIRVOperand(SPIRVOperandType::NUMBERID, Op0ID);
      Ops.push_back(Op0IDOp);

      Op1ID = InvMaskID;
      Op1IDOp = new SPIRVOperand(SPIRVOperandType::NUMBERID, Op1ID);
      Ops.push_back(Op1IDOp);

      uint32_t OrgValID = nextID;

      Inst = new SPIRVInstruction(5, spv::OpBitwiseAnd, nextID++, Ops);
      SPIRVInstList.push_back(Inst);

      // Create correct value according to index of insertelement.
      Ops.clear();

      Ops.push_back(ResTyIDOp);

      Op0ID = VMap[I.getOperand(1)];
      Op0IDOp = new SPIRVOperand(SPIRVOperandType::NUMBERID, Op0ID);
      Ops.push_back(Op0IDOp);

      Op1ID = ShiftAmountID;
      Op1IDOp = new SPIRVOperand(SPIRVOperandType::NUMBERID, Op1ID);
      Ops.push_back(Op1IDOp);

      uint32_t InsertValID = nextID;

      Inst = new SPIRVInstruction(5, spv::OpShiftLeftLogical, nextID++, Ops);
      SPIRVInstList.push_back(Inst);

      // Insert value to original value.
      Ops.clear();

      Ops.push_back(ResTyIDOp);

      Op0ID = OrgValID;
      Op0IDOp = new SPIRVOperand(SPIRVOperandType::NUMBERID, Op0ID);
      Ops.push_back(Op0IDOp);

      Op1ID = InsertValID;
      Op1IDOp = new SPIRVOperand(SPIRVOperandType::NUMBERID, Op1ID);
      Ops.push_back(Op1IDOp);

      Inst = new SPIRVInstruction(5, spv::OpBitwiseOr, nextID++, Ops);
      SPIRVInstList.push_back(Inst);

      break;
    }

    // Ops[0] = Result Type ID
    // Ops[1] = Object ID
    // Ops[2] = Composite ID
    // Ops[3] ... Ops[n] = Indexes (Literal Number)
    SPIRVOperandList Ops;

    uint32_t ResTyID = lookupType(I.getType());
    SPIRVOperand *ResTyIDOp =
        new SPIRVOperand(SPIRVOperandType::NUMBERID, ResTyID);
    Ops.push_back(ResTyIDOp);

    uint32_t ObjectID = VMap[I.getOperand(1)];
    SPIRVOperand *ObjectIDOp =
        new SPIRVOperand(SPIRVOperandType::NUMBERID, ObjectID);
    Ops.push_back(ObjectIDOp);

    uint32_t CompositeID = VMap[I.getOperand(0)];
    SPIRVOperand *CompositeIDOp =
        new SPIRVOperand(SPIRVOperandType::NUMBERID, CompositeID);
    Ops.push_back(CompositeIDOp);

    spv::Op Opcode = spv::OpCompositeInsert;
    if (const ConstantInt *CI = dyn_cast<ConstantInt>(I.getOperand(2))) {
      std::vector<uint32_t> LiteralNum;
      assert(CI->getZExtValue() < UINT32_MAX);
      LiteralNum.push_back(static_cast<uint32_t>(CI->getZExtValue()));
      SPIRVOperand *Indexes =
          new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, LiteralNum);
      Ops.push_back(Indexes);
    } else {
      uint32_t IndexID = VMap[I.getOperand(1)];
      SPIRVOperand *IndexIDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, IndexID);
      Ops.push_back(IndexIDOp);
      Opcode = spv::OpVectorInsertDynamic;
    }

    uint16_t WordCount = 6;
    SPIRVInstruction *Inst =
        new SPIRVInstruction(WordCount, Opcode, nextID++, Ops);
    SPIRVInstList.push_back(Inst);
    break;
  }
  case Instruction::ShuffleVector: {
    // Ops[0] = Result Type ID
    // Ops[1] = Vector 1 ID
    // Ops[2] = Vector 2 ID
    // Ops[3] ... Ops[n] = Components (Literal Number)
    SPIRVOperandList Ops;

    uint32_t ResTyID = lookupType(I.getType());
    SPIRVOperand *ResTyIDOp =
        new SPIRVOperand(SPIRVOperandType::NUMBERID, ResTyID);
    Ops.push_back(ResTyIDOp);

    uint32_t Vec1ID = VMap[I.getOperand(0)];
    SPIRVOperand *Vec1IDOp =
        new SPIRVOperand(SPIRVOperandType::NUMBERID, Vec1ID);
    Ops.push_back(Vec1IDOp);

    uint32_t Vec2ID = VMap[I.getOperand(1)];
    SPIRVOperand *Vec2IDOp =
        new SPIRVOperand(SPIRVOperandType::NUMBERID, Vec2ID);
    Ops.push_back(Vec2IDOp);

    uint64_t NumElements = 0;
    if (Constant *Cst = dyn_cast<Constant>(I.getOperand(2))) {
      NumElements = cast<VectorType>(Cst->getType())->getNumElements();

      if (Cst->isNullValue()) {
        for (unsigned i = 0; i < NumElements; i++) {
          std::vector<uint32_t> LiteralNum;
          LiteralNum.push_back(0);
          SPIRVOperand *Component =
              new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, LiteralNum);
          Ops.push_back(Component);
        }
      } else if (const ConstantDataSequential *CDS =
                     dyn_cast<ConstantDataSequential>(Cst)) {
        for (unsigned i = 0; i < CDS->getNumElements(); i++) {
          std::vector<uint32_t> LiteralNum;
          assert(CDS->getElementAsInteger(i) < UINT32_MAX);
          LiteralNum.push_back(
              static_cast<uint32_t>(CDS->getElementAsInteger(i)));
          SPIRVOperand *Component =
              new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, LiteralNum);
          Ops.push_back(Component);
        }
      } else if (const ConstantVector *CV = dyn_cast<ConstantVector>(Cst)) {
        for (unsigned i = 0; i < CV->getNumOperands(); i++) {
          auto Op = CV->getOperand(i);

          uint32_t literal = 0;

          if (auto CI = dyn_cast<ConstantInt>(Op)) {
            literal = static_cast<uint32_t>(CI->getZExtValue());
          } else if (auto UI = dyn_cast<UndefValue>(Op)) {
            literal = 0xFFFFFFFFu;
          } else {
            Op->print(errs());
            llvm_unreachable("Unsupported element in ConstantVector!");
          }

          std::vector<uint32_t> LiteralNum;
          LiteralNum.push_back(literal);
          SPIRVOperand *Component =
              new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, LiteralNum);
          Ops.push_back(Component);
        }
      } else {
        Cst->print(errs());
        llvm_unreachable("Unsupported constant mask in ShuffleVector!");
      }
    }

    uint16_t WordCount = static_cast<uint16_t>(5 + NumElements);
    SPIRVInstruction *Inst =
        new SPIRVInstruction(WordCount, spv::OpVectorShuffle, nextID++, Ops);
    SPIRVInstList.push_back(Inst);
    break;
  }
  case Instruction::ICmp:
  case Instruction::FCmp: {
    CmpInst *CmpI = cast<CmpInst>(&I);

    // Ops[0] = Result Type ID
    // Ops[1] = Operand 1 ID
    // Ops[2] = Operand 2 ID
    SPIRVOperandList Ops;

    uint32_t ResTyID = lookupType(CmpI->getType());
    SPIRVOperand *ResTyIDOp =
        new SPIRVOperand(SPIRVOperandType::NUMBERID, ResTyID);
    Ops.push_back(ResTyIDOp);

    // Pointer equality is invalid.
    Type* ArgTy = CmpI->getOperand(0)->getType();
    if (isa<PointerType>(ArgTy)) {
      CmpI->print(errs());
      std::string name = I.getParent()->getParent()->getName();
      errs()
          << "\nPointer equality test is not supported by SPIR-V for Vulkan, "
          << "in function " << name << "\n";
      llvm_unreachable("Pointer equality check is invalid");
      break;
    }

    uint32_t Op1ID = VMap[CmpI->getOperand(0)];
    SPIRVOperand *Op1IDOp = new SPIRVOperand(SPIRVOperandType::NUMBERID, Op1ID);
    Ops.push_back(Op1IDOp);

    uint32_t Op2ID = VMap[CmpI->getOperand(1)];
    SPIRVOperand *Op2IDOp = new SPIRVOperand(SPIRVOperandType::NUMBERID, Op2ID);
    Ops.push_back(Op2IDOp);

    spv::Op Opcode = GetSPIRVCmpOpcode(CmpI);
    SPIRVInstruction *Inst = new SPIRVInstruction(5, Opcode, nextID++, Ops);
    SPIRVInstList.push_back(Inst);
    break;
  }
  case Instruction::Br: {
    // Branch instrucion is deferred because it needs label's ID. Record slot's
    // location on SPIRVInstructionList.
    DeferredInsts.push_back(
        std::make_tuple(&I, --SPIRVInstList.end(), 0 /* No id */));
    break;
  }
  case Instruction::Switch: {
    I.print(errs());
    llvm_unreachable("Unsupported instruction???");
    break;
  }
  case Instruction::IndirectBr: {
    I.print(errs());
    llvm_unreachable("Unsupported instruction???");
    break;
  }
  case Instruction::PHI: {
    // Branch instrucion is deferred because it needs label's ID. Record slot's
    // location on SPIRVInstructionList.
    DeferredInsts.push_back(
        std::make_tuple(&I, --SPIRVInstList.end(), nextID++));
    break;
  }
  case Instruction::Alloca: {
    //
    // Generate OpVariable.
    //
    // Ops[0] : Result Type ID
    // Ops[1] : Storage Class
    SPIRVOperandList Ops;

    Type *ResTy = I.getType();
    uint32_t ResTyID = lookupType(ResTy);
    SPIRVOperand *ResTyOp =
        new SPIRVOperand(SPIRVOperandType::NUMBERID, ResTyID);
    Ops.push_back(ResTyOp);

    spv::StorageClass StorageClass = spv::StorageClassFunction;
    SPIRVOperand *StorageClassOp =
        new SPIRVOperand(SPIRVOperandType::NUMBERID, StorageClass);
    Ops.push_back(StorageClassOp);

    SPIRVInstruction *Inst =
        new SPIRVInstruction(4, spv::OpVariable, nextID++, Ops);
    SPIRVInstList.push_back(Inst);
    break;
  }
  case Instruction::Load: {
    LoadInst *LD = cast<LoadInst>(&I);
    //
    // Generate OpLoad.
    //

    // Ops[0] = Result Type ID
    // Ops[1] = Pointer ID
    // Ops[2] ... Ops[n] = Optional Memory Access
    //
    // TODO: Do we need to implement Optional Memory Access???
    SPIRVOperandList Ops;

    uint32_t ResTyID = lookupType(LD->getType());
    SPIRVOperand *ResTyIDOp =
        new SPIRVOperand(SPIRVOperandType::NUMBERID, ResTyID);
    Ops.push_back(ResTyIDOp);

    uint32_t PointerID = VMap[LD->getPointerOperand()];
    SPIRVOperand *PointerIDOp =
        new SPIRVOperand(SPIRVOperandType::NUMBERID, PointerID);
    Ops.push_back(PointerIDOp);

    SPIRVInstruction *Inst =
        new SPIRVInstruction(4, spv::OpLoad, nextID++, Ops);
    SPIRVInstList.push_back(Inst);
    break;
  }
  case Instruction::Store: {
    StoreInst *ST = cast<StoreInst>(&I);
    //
    // Generate OpStore.
    //

    // Ops[0] = Pointer ID
    // Ops[1] = Object ID
    // Ops[2] ... Ops[n] = Optional Memory Access (later???)
    //
    // TODO: Do we need to implement Optional Memory Access???
    SPIRVOperand *Ops[2] = {new SPIRVOperand(SPIRVOperandType::NUMBERID,
                                             VMap[ST->getPointerOperand()]),
                            new SPIRVOperand(SPIRVOperandType::NUMBERID,
                                             VMap[ST->getValueOperand()])};

    SPIRVInstruction *Inst =
        new SPIRVInstruction(3, spv::OpStore, 0 /* No id */, Ops);
    SPIRVInstList.push_back(Inst);
    break;
  }
  case Instruction::AtomicCmpXchg: {
    I.print(errs());
    llvm_unreachable("Unsupported instruction???");
    break;
  }
  case Instruction::AtomicRMW: {
    I.print(errs());
    llvm_unreachable("Unsupported instruction???");
    break;
  }
  case Instruction::Fence: {
    I.print(errs());
    llvm_unreachable("Unsupported instruction???");
    break;
  }
  case Instruction::Call: {
    CallInst *Call = dyn_cast<CallInst>(&I);
    Function *Callee = Call->getCalledFunction();

    // Sampler initializers become a load of the corresponding sampler.
    if (Callee->getName().equals("__translate_sampler_initializer")) {
      // Check that the sampler map was definitely used though.
      if (0 == getSamplerMap().size()) {
        llvm_unreachable("Sampler literal in source without sampler map!");
      }

      SPIRVOperandList Ops;

      uint32_t ResTyID = lookupType(SamplerTy->getPointerElementType());
      SPIRVOperand *ResTyIDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, ResTyID);
      Ops.push_back(ResTyIDOp);

      uint32_t PointerID = VMap[Call];
      SPIRVOperand *PointerIDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, PointerID);
      Ops.push_back(PointerIDOp);

      VMap[Call] = nextID;
      SPIRVInstruction *Inst =
          new SPIRVInstruction(4, spv::OpLoad, nextID++, Ops);
      SPIRVInstList.push_back(Inst);

      break;
    }

    if (Callee->getName().startswith("spirv.atomic")) {
      spv::Op opcode = StringSwitch<spv::Op>(Callee->getName())
                           .Case("spirv.atomic_add", spv::OpAtomicIAdd)
                           .Case("spirv.atomic_sub", spv::OpAtomicISub)
                           .Case("spirv.atomic_exchange", spv::OpAtomicExchange)
                           .Case("spirv.atomic_inc", spv::OpAtomicIIncrement)
                           .Case("spirv.atomic_dec", spv::OpAtomicIDecrement)
                           .Case("spirv.atomic_compare_exchange",
                                 spv::OpAtomicCompareExchange)
                           .Case("spirv.atomic_umin", spv::OpAtomicUMin)
                           .Case("spirv.atomic_smin", spv::OpAtomicSMin)
                           .Case("spirv.atomic_umax", spv::OpAtomicUMax)
                           .Case("spirv.atomic_smax", spv::OpAtomicSMax)
                           .Case("spirv.atomic_and", spv::OpAtomicAnd)
                           .Case("spirv.atomic_or", spv::OpAtomicOr)
                           .Case("spirv.atomic_xor", spv::OpAtomicXor)
                           .Default(spv::OpNop);

      //
      // Generate OpAtomic*.
      //
      SPIRVOperandList Ops;

      uint32_t TyID = lookupType(I.getType());
      Ops.push_back(new SPIRVOperand(SPIRVOperandType::NUMBERID, TyID));

      for (unsigned i = 0; i < Call->getNumArgOperands(); i++) {
        Ops.push_back(new SPIRVOperand(SPIRVOperandType::NUMBERID,
                                       VMap[Call->getArgOperand(i)]));
      }

      VMap[&I] = nextID;

      SPIRVInstruction *Inst = new SPIRVInstruction(
          static_cast<uint16_t>(2 + Ops.size()), opcode, nextID++, Ops);
      SPIRVInstList.push_back(Inst);
      break;
    }

    if (Callee->getName().startswith("_Z3dot")) {
      // If the argument is a vector type, generate OpDot
      if (Call->getArgOperand(0)->getType()->isVectorTy()) {
        //
        // Generate OpDot.
        //
        SPIRVOperandList Ops;

        uint32_t TyID = lookupType(I.getType());
        Ops.push_back(new SPIRVOperand(SPIRVOperandType::NUMBERID, TyID));

        for (unsigned i = 0; i < Call->getNumArgOperands(); i++) {
          Ops.push_back(new SPIRVOperand(SPIRVOperandType::NUMBERID,
                                         VMap[Call->getArgOperand(i)]));
        }

        VMap[&I] = nextID;

        SPIRVInstruction *Inst = new SPIRVInstruction(
            static_cast<uint16_t>(2 + Ops.size()), spv::OpDot, nextID++, Ops);
        SPIRVInstList.push_back(Inst);
      } else {
        //
        // Generate OpFMul.
        //
        SPIRVOperandList Ops;

        uint32_t TyID = lookupType(I.getType());
        Ops.push_back(new SPIRVOperand(SPIRVOperandType::NUMBERID, TyID));

        for (unsigned i = 0; i < Call->getNumArgOperands(); i++) {
          Ops.push_back(new SPIRVOperand(SPIRVOperandType::NUMBERID,
                                         VMap[Call->getArgOperand(i)]));
        }

        VMap[&I] = nextID;

        SPIRVInstruction *Inst = new SPIRVInstruction(
            static_cast<uint16_t>(2 + Ops.size()), spv::OpFMul, nextID++, Ops);
        SPIRVInstList.push_back(Inst);
      }
      break;
    }

    // spirv.store_null.* intrinsics become OpStore's.
    if (Callee->getName().startswith("spirv.store_null")) {
      //
      // Generate OpStore.
      //

      // Ops[0] = Pointer ID
      // Ops[1] = Object ID
      // Ops[2] ... Ops[n]
      SPIRVOperandList Ops;

      uint32_t PointerID = VMap[Call->getArgOperand(0)];
      SPIRVOperand *PointerIDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, PointerID);
      Ops.push_back(PointerIDOp);

      uint32_t ObjectID = VMap[Call->getArgOperand(1)];
      SPIRVOperand *ObjectIDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, ObjectID);
      Ops.push_back(ObjectIDOp);

      SPIRVInstruction *Inst =
          new SPIRVInstruction(3, spv::OpStore, 0 /* No id */, Ops);
      SPIRVInstList.push_back(Inst);

      break;
    }

    // spirv.copy_memory.* intrinsics become OpMemoryMemory's.
    if (Callee->getName().startswith("spirv.copy_memory")) {
      //
      // Generate OpCopyMemory.
      //

      // Ops[0] = Dst ID
      // Ops[1] = Src ID
      // Ops[2] = Memory Access
      // Ops[3] = Alignment

      auto IsVolatile =
          dyn_cast<ConstantInt>(Call->getArgOperand(3))->getZExtValue() != 0;

      auto VolatileMemoryAccess = (IsVolatile) ? spv::MemoryAccessVolatileMask
                                               : spv::MemoryAccessMaskNone;

      auto MemoryAccess = VolatileMemoryAccess | spv::MemoryAccessAlignedMask;

      auto Alignment =
          dyn_cast<ConstantInt>(Call->getArgOperand(2))->getZExtValue();

      SPIRVOperand *Ops[4] = {
          new SPIRVOperand(SPIRVOperandType::NUMBERID,
                           VMap[Call->getArgOperand(0)]),
          new SPIRVOperand(SPIRVOperandType::NUMBERID,
                           VMap[Call->getArgOperand(1)]),
          new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, MemoryAccess),
          new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER,
                           static_cast<uint32_t>(Alignment))};

      SPIRVInstruction *Inst =
          new SPIRVInstruction(5, spv::OpCopyMemory, 0 /* No id */, Ops);

      SPIRVInstList.push_back(Inst);

      break;
    }

    // Nothing to do for abs with uint. Map abs's operand ID to VMap for abs
    // with unit.
    if (Callee->getName().equals("_Z3absj") ||
        Callee->getName().equals("_Z3absDv2_j") ||
        Callee->getName().equals("_Z3absDv3_j") ||
        Callee->getName().equals("_Z3absDv4_j")) {
      VMap[&I] = VMap[Call->getOperand(0)];
      break;
    }

    // barrier is converted to OpControlBarrier
    if (Callee->getName().equals("__spirv_control_barrier")) {
      //
      // Generate OpControlBarrier.
      //
      // Ops[0] = Execution Scope ID
      // Ops[1] = Memory Scope ID
      // Ops[2] = Memory Semantics ID
      //
      Value *ExecutionScope = Call->getArgOperand(0);
      Value *MemoryScope = Call->getArgOperand(1);
      Value *MemorySemantics = Call->getArgOperand(2);

      SPIRVOperand *Ops[3] = {
          new SPIRVOperand(SPIRVOperandType::NUMBERID, VMap[ExecutionScope]),
          new SPIRVOperand(SPIRVOperandType::NUMBERID, VMap[MemoryScope]),
          new SPIRVOperand(SPIRVOperandType::NUMBERID, VMap[MemorySemantics])};

      SPIRVInstList.push_back(
          new SPIRVInstruction(4, spv::OpControlBarrier, 0 /* No id */, Ops));
      break;
    }

    // memory barrier is converted to OpMemoryBarrier
    if (Callee->getName().equals("__spirv_memory_barrier")) {
      //
      // Generate OpMemoryBarrier.
      //
      // Ops[0] = Memory Scope ID
      // Ops[1] = Memory Semantics ID
      //
      SPIRVOperandList Ops;

      Value *MemoryScope = Call->getArgOperand(0);
      Value *MemorySemantics = Call->getArgOperand(1);

      uint32_t MemoryScopeID = VMap[MemoryScope];
      Ops.push_back(
          new SPIRVOperand(SPIRVOperandType::NUMBERID, MemoryScopeID));

      uint32_t MemorySemanticsID = VMap[MemorySemantics];
      Ops.push_back(
          new SPIRVOperand(SPIRVOperandType::NUMBERID, MemorySemanticsID));

      SPIRVInstruction *Inst =
          new SPIRVInstruction(3, spv::OpMemoryBarrier, 0 /* No id */, Ops);
      SPIRVInstList.push_back(Inst);
      break;
    }

    // isinf is converted to OpIsInf
    if (Callee->getName().equals("__spirv_isinff") ||
        Callee->getName().equals("__spirv_isinfDv2_f") ||
        Callee->getName().equals("__spirv_isinfDv3_f") ||
        Callee->getName().equals("__spirv_isinfDv4_f")) {
      //
      // Generate OpIsInf.
      //
      // Ops[0] = Result Type ID
      // Ops[1] = X ID
      //
      SPIRVOperandList Ops;

      uint32_t TyID = lookupType(I.getType());
      SPIRVOperand *ResTyIDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, TyID);
      Ops.push_back(ResTyIDOp);

      uint32_t XID = VMap[Call->getArgOperand(0)];
      Ops.push_back(new SPIRVOperand(SPIRVOperandType::NUMBERID, XID));

      VMap[&I] = nextID;

      SPIRVInstruction *Inst =
          new SPIRVInstruction(4, spv::OpIsInf, nextID++, Ops);
      SPIRVInstList.push_back(Inst);
      break;
    }

    // isnan is converted to OpIsNan
    if (Callee->getName().equals("__spirv_isnanf") ||
        Callee->getName().equals("__spirv_isnanDv2_f") ||
        Callee->getName().equals("__spirv_isnanDv3_f") ||
        Callee->getName().equals("__spirv_isnanDv4_f")) {
      //
      // Generate OpIsInf.
      //
      // Ops[0] = Result Type ID
      // Ops[1] = X ID
      //
      SPIRVOperandList Ops;

      uint32_t TyID = lookupType(I.getType());
      SPIRVOperand *ResTyIDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, TyID);
      Ops.push_back(ResTyIDOp);

      uint32_t XID = VMap[Call->getArgOperand(0)];
      Ops.push_back(new SPIRVOperand(SPIRVOperandType::NUMBERID, XID));

      VMap[&I] = nextID;

      SPIRVInstruction *Inst =
          new SPIRVInstruction(4, spv::OpIsNan, nextID++, Ops);
      SPIRVInstList.push_back(Inst);
      break;
    }

    // all is converted to OpAll
    if (Callee->getName().equals("__spirv_allDv2_i") ||
        Callee->getName().equals("__spirv_allDv3_i") ||
        Callee->getName().equals("__spirv_allDv4_i")) {
      //
      // Generate OpAll.
      //
      // Ops[0] = Result Type ID
      // Ops[1] = Vector ID
      //
      SPIRVOperandList Ops;

      uint32_t TyID = lookupType(I.getType());
      SPIRVOperand *ResTyIDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, TyID);
      Ops.push_back(ResTyIDOp);

      uint32_t VectorID = VMap[Call->getArgOperand(0)];
      Ops.push_back(new SPIRVOperand(SPIRVOperandType::NUMBERID, VectorID));

      VMap[&I] = nextID;

      SPIRVInstruction *Inst =
          new SPIRVInstruction(4, spv::OpAll, nextID++, Ops);
      SPIRVInstList.push_back(Inst);
      break;
    }

    // any is converted to OpAny
    if (Callee->getName().equals("__spirv_anyDv2_i") ||
        Callee->getName().equals("__spirv_anyDv3_i") ||
        Callee->getName().equals("__spirv_anyDv4_i")) {
      //
      // Generate OpAny.
      //
      // Ops[0] = Result Type ID
      // Ops[1] = Vector ID
      //
      SPIRVOperandList Ops;

      uint32_t TyID = lookupType(I.getType());
      SPIRVOperand *ResTyIDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, TyID);
      Ops.push_back(ResTyIDOp);

      uint32_t VectorID = VMap[Call->getArgOperand(0)];
      Ops.push_back(new SPIRVOperand(SPIRVOperandType::NUMBERID, VectorID));

      VMap[&I] = nextID;

      SPIRVInstruction *Inst =
          new SPIRVInstruction(4, spv::OpAny, nextID++, Ops);
      SPIRVInstList.push_back(Inst);
      break;
    }

    // read_image is converted to OpSampledImage and OpImageSampleExplicitLod.
    // Additionally, OpTypeSampledImage is generated.
    if (Callee->getName().equals(
            "_Z11read_imagef14ocl_image2d_ro11ocl_samplerDv2_f") ||
        Callee->getName().equals(
            "_Z11read_imagef14ocl_image3d_ro11ocl_samplerDv4_f")) {
      //
      // Generate OpSampledImage.
      //
      // Ops[0] = Result Type ID
      // Ops[1] = Image ID
      // Ops[2] = Sampler ID
      //
      SPIRVOperandList Ops;

      Value *Image = Call->getArgOperand(0);
      Value *Sampler = Call->getArgOperand(1);
      Value *Coordinate = Call->getArgOperand(2);

      TypeMapType &OpImageTypeMap = getImageTypeMap();
      Type *ImageTy = Image->getType()->getPointerElementType();
      uint32_t ImageTyID = OpImageTypeMap[ImageTy];
      SPIRVOperand *ResTyIDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, ImageTyID);
      Ops.push_back(ResTyIDOp);

      uint32_t ImageID = VMap[Image];
      SPIRVOperand *ImageIDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, ImageID);
      Ops.push_back(ImageIDOp);

      uint32_t SamplerID = VMap[Sampler];
      SPIRVOperand *SamplerIDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, SamplerID);
      Ops.push_back(SamplerIDOp);

      uint32_t SampledImageID = nextID;

      SPIRVInstruction *Inst =
          new SPIRVInstruction(5, spv::OpSampledImage, nextID++, Ops);
      SPIRVInstList.push_back(Inst);

      //
      // Generate OpImageSampleExplicitLod.
      //
      // Ops[0] = Result Type ID
      // Ops[1] = Sampled Image ID
      // Ops[2] = Coordinate ID
      // Ops[3] = Image Operands Type ID
      // Ops[4] ... Ops[n] = Operands ID
      //
      Ops.clear();

      uint32_t RetTyID = lookupType(Call->getType());
      ResTyIDOp = new SPIRVOperand(SPIRVOperandType::NUMBERID, RetTyID);
      Ops.push_back(ResTyIDOp);

      SPIRVOperand *SampledImageIDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, SampledImageID);
      Ops.push_back(SampledImageIDOp);

      uint32_t CoordinateID = VMap[Coordinate];
      SPIRVOperand *CoordinateIDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, CoordinateID);
      Ops.push_back(CoordinateIDOp);

      std::vector<uint32_t> LiteralNum;
      LiteralNum.push_back(spv::ImageOperandsLodMask);
      SPIRVOperand *ImageOperandTyIDOp =
          new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, LiteralNum);
      Ops.push_back(ImageOperandTyIDOp);

      Constant *CstFP0 = ConstantFP::get(Context, APFloat(0.0f));
      uint32_t OperandID = VMap[CstFP0];
      SPIRVOperand *OperandIDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, OperandID);
      Ops.push_back(OperandIDOp);

      VMap[&I] = nextID;

      Inst =
          new SPIRVInstruction(7, spv::OpImageSampleExplicitLod, nextID++, Ops);
      SPIRVInstList.push_back(Inst);
      break;
    }

    // write_imagef is mapped to OpImageWrite.
    if (Callee->getName().equals(
            "_Z12write_imagef14ocl_image2d_woDv2_iDv4_f") ||
        Callee->getName().equals(
            "_Z12write_imagef14ocl_image3d_woDv4_iDv4_f")) {
      //
      // Generate OpImageWrite.
      //
      // Ops[0] = Image ID
      // Ops[1] = Coordinate ID
      // Ops[2] = Texel ID
      // Ops[3] = (Optional) Image Operands Type (Literal Number)
      // Ops[4] ... Ops[n] = (Optional) Operands ID
      //
      SPIRVOperandList Ops;

      Value *Image = Call->getArgOperand(0);
      Value *Coordinate = Call->getArgOperand(1);
      Value *Texel = Call->getArgOperand(2);

      uint32_t ImageID = VMap[Image];
      SPIRVOperand *ImageIDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, ImageID);
      Ops.push_back(ImageIDOp);

      uint32_t CoordinateID = VMap[Coordinate];
      SPIRVOperand *CoordinateIDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, CoordinateID);
      Ops.push_back(CoordinateIDOp);

      uint32_t TexelID = VMap[Texel];
      SPIRVOperand *TexelIDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, TexelID);
      Ops.push_back(TexelIDOp);

      SPIRVInstruction *Inst =
          new SPIRVInstruction(4, spv::OpImageWrite, 0 /* No id */, Ops);
      SPIRVInstList.push_back(Inst);
      break;
    }

    // Call instrucion is deferred because it needs function's ID. Record
    // slot's location on SPIRVInstructionList.
    DeferredInsts.push_back(
        std::make_tuple(&I, --SPIRVInstList.end(), nextID++));

    // Check whether this call is for extend instructions.
    glsl::ExtInst EInst = getExtInstEnum(Callee->getName());
    if (EInst == glsl::ExtInstFindUMsb) {
      // clz needs OpExtInst and OpISub with constant 31. Increase nextID.
      VMap[&I] = nextID;
      nextID++;
    }
    break;
  }
  case Instruction::Ret: {
    unsigned NumOps = I.getNumOperands();
    if (NumOps == 0) {
      //
      // Generate OpReturn.
      //

      // Empty Ops
      SPIRVOperandList Ops;
      SPIRVInstruction *Inst =
          new SPIRVInstruction(1, spv::OpReturn, 0 /* No id */, Ops);
      SPIRVInstList.push_back(Inst);
    } else {
      //
      // Generate OpReturnValue.
      //

      // Ops[0] = Return Value ID
      SPIRVOperandList Ops;
      uint32_t RetValID = VMap[I.getOperand(0)];
      SPIRVOperand *RetValIDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, RetValID);
      Ops.push_back(RetValIDOp);

      SPIRVInstruction *Inst =
          new SPIRVInstruction(2, spv::OpReturnValue, 0 /* No id */, Ops);
      SPIRVInstList.push_back(Inst);
      break;
    }
    break;
  }
  }
}

void SPIRVProducerPass::GenerateFuncEpilogue() {
  SPIRVInstructionList &SPIRVInstList = getSPIRVInstList();

  //
  // Generate OpFunctionEnd
  //

  // Empty Ops
  SPIRVOperandList Ops;
  SPIRVInstruction *Inst =
      new SPIRVInstruction(1, spv::OpFunctionEnd, 0 /* No id */, Ops);
  SPIRVInstList.push_back(Inst);
}

bool SPIRVProducerPass::is4xi8vec(Type *Ty) const {
  LLVMContext &Context = Ty->getContext();
  if (Ty->isVectorTy()) {
    if (Ty->getVectorElementType() == Type::getInt8Ty(Context) &&
        Ty->getVectorNumElements() == 4) {
      return true;
    }
  }

  return false;
}

void SPIRVProducerPass::HandleDeferredInstruction() {
  SPIRVInstructionList &SPIRVInstList = getSPIRVInstList();
  ValueMapType &VMap = getValueMap();
  DeferredInstVecType &DeferredInsts = getDeferredInstVec();

  for (auto DeferredInst = DeferredInsts.rbegin();
       DeferredInst != DeferredInsts.rend(); ++DeferredInst) {
    Value *Inst = std::get<0>(*DeferredInst);
    SPIRVInstructionList::iterator InsertPoint = ++std::get<1>(*DeferredInst);
    if (InsertPoint != SPIRVInstList.end()) {
      while ((*InsertPoint)->getOpcode() == spv::OpPhi) {
        ++InsertPoint;
      }
    }

    if (BranchInst *Br = dyn_cast<BranchInst>(Inst)) {
      // Check whether basic block, which has this branch instruction, is loop
      // header or not. If it is loop header, generate OpLoopMerge and
      // OpBranchConditional.
      Function *Func = Br->getParent()->getParent();
      DominatorTree &DT =
          getAnalysis<DominatorTreeWrapperPass>(*Func).getDomTree();
      const LoopInfo &LI =
          getAnalysis<LoopInfoWrapperPass>(*Func).getLoopInfo();

      BasicBlock *BrBB = Br->getParent();
      if (LI.isLoopHeader(BrBB)) {
        Value *ContinueBB = nullptr;
        Value *MergeBB = nullptr;

        Loop *L = LI.getLoopFor(BrBB);
        MergeBB = L->getExitBlock();
        if (!MergeBB) {
          // StructurizeCFG pass converts CFG into triangle shape and the cfg
          // has regions with single entry/exit. As a result, loop should not
          // have multiple exits.
          llvm_unreachable("Loop has multiple exits???");
        }

        if (L->isLoopLatch(BrBB)) {
          ContinueBB = BrBB;
        } else {
          // From SPIR-V spec 2.11, Continue Target must dominate that back-edge
          // block.
          BasicBlock *Header = L->getHeader();
          BasicBlock *Latch = L->getLoopLatch();
          for (BasicBlock *BB : L->blocks()) {
            if (BB == Header) {
              continue;
            }

            // Check whether block dominates block with back-edge.
            if (DT.dominates(BB, Latch)) {
              ContinueBB = BB;
            }
          }

          if (!ContinueBB) {
            llvm_unreachable("Wrong continue block from loop");
          }
        }

        //
        // Generate OpLoopMerge.
        //
        // Ops[0] = Merge Block ID
        // Ops[1] = Continue Target ID
        // Ops[2] = Selection Control
        SPIRVOperandList Ops;

        // StructurizeCFG pass already manipulated CFG. Just use false block of
        // branch instruction as merge block.
        uint32_t MergeBBID = VMap[MergeBB];
        SPIRVOperand *MergeBBIDOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, MergeBBID);
        Ops.push_back(MergeBBIDOp);

        uint32_t ContinueBBID = VMap[ContinueBB];
        SPIRVOperand *ContinueBBIDOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, ContinueBBID);
        Ops.push_back(ContinueBBIDOp);

        SPIRVOperand *SelectionControlOp = new SPIRVOperand(
            SPIRVOperandType::NUMBERID, spv::SelectionControlMaskNone);
        Ops.push_back(SelectionControlOp);

        SPIRVInstruction *MergeInst =
            new SPIRVInstruction(4, spv::OpLoopMerge, 0 /* No id */, Ops);
        SPIRVInstList.insert(InsertPoint, MergeInst);

      } else if (Br->isConditional()) {
        bool HasBackEdge = false;

        for (unsigned i = 0; i < Br->getNumSuccessors(); i++) {
          if (LI.isLoopHeader(Br->getSuccessor(i))) {
            HasBackEdge = true;
          }
        }
        if (!HasBackEdge) {
          //
          // Generate OpSelectionMerge.
          //
          // Ops[0] = Merge Block ID
          // Ops[1] = Selection Control
          SPIRVOperandList Ops;

          // StructurizeCFG pass already manipulated CFG. Just use false block
          // of branch instruction as merge block.
          uint32_t MergeBBID = VMap[Br->getSuccessor(1)];
          SPIRVOperand *MergeBBIDOp =
              new SPIRVOperand(SPIRVOperandType::NUMBERID, MergeBBID);
          Ops.push_back(MergeBBIDOp);

          SPIRVOperand *SelectionControlOp = new SPIRVOperand(
              SPIRVOperandType::NUMBERID, spv::SelectionControlMaskNone);
          Ops.push_back(SelectionControlOp);

          SPIRVInstruction *MergeInst = new SPIRVInstruction(
              3, spv::OpSelectionMerge, 0 /* No id */, Ops);
          SPIRVInstList.insert(InsertPoint, MergeInst);
        }
      }

      if (Br->isConditional()) {
        //
        // Generate OpBranchConditional.
        //
        // Ops[0] = Condition ID
        // Ops[1] = True Label ID
        // Ops[2] = False Label ID
        // Ops[3] ... Ops[n] = Branch weights (Literal Number)
        SPIRVOperandList Ops;

        uint32_t CondID = VMap[Br->getCondition()];
        SPIRVOperand *CondIDOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, CondID);
        Ops.push_back(CondIDOp);

        uint32_t TrueBBID = VMap[Br->getSuccessor(0)];
        SPIRVOperand *TrueBBIDOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, TrueBBID);
        Ops.push_back(TrueBBIDOp);

        uint32_t FalseBBID = VMap[Br->getSuccessor(1)];
        SPIRVOperand *FalseBBIDOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, FalseBBID);
        Ops.push_back(FalseBBIDOp);

        SPIRVInstruction *BrInst = new SPIRVInstruction(
            4, spv::OpBranchConditional, 0 /* No id */, Ops);
        SPIRVInstList.insert(InsertPoint, BrInst);
      } else {
        //
        // Generate OpBranch.
        //
        // Ops[0] = Target Label ID
        SPIRVOperandList Ops;

        uint32_t TargetID = VMap[Br->getSuccessor(0)];
        SPIRVOperand *TargetIDOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, TargetID);
        Ops.push_back(TargetIDOp);

        SPIRVInstList.insert(
            InsertPoint,
            new SPIRVInstruction(2, spv::OpBranch, 0 /* No id */, Ops));
      }
    } else if (PHINode *PHI = dyn_cast<PHINode>(Inst)) {
      //
      // Generate OpPhi.
      //
      // Ops[0] = Result Type ID
      // Ops[1] ... Ops[n] = (Variable ID, Parent ID) pairs
      SPIRVOperandList Ops;

      uint32_t ResTyID = lookupType(PHI->getType());
      SPIRVOperand *ResTyIDOp =
          new SPIRVOperand(SPIRVOperandType::NUMBERID, ResTyID);
      Ops.push_back(ResTyIDOp);

      uint16_t WordCount = 3;
      for (unsigned i = 0; i < PHI->getNumIncomingValues(); i++) {
        uint32_t VarID = VMap[PHI->getIncomingValue(i)];
        SPIRVOperand *VarIDOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, VarID);
        Ops.push_back(VarIDOp);

        uint32_t ParentID = VMap[PHI->getIncomingBlock(i)];
        SPIRVOperand *ParentIDOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, ParentID);
        Ops.push_back(ParentIDOp);

        WordCount += 2;
      }

      SPIRVInstList.insert(
          InsertPoint, new SPIRVInstruction(WordCount, spv::OpPhi,
                                            std::get<2>(*DeferredInst), Ops));
    } else if (CallInst *Call = dyn_cast<CallInst>(Inst)) {
      Function *Callee = Call->getCalledFunction();
      glsl::ExtInst EInst = getExtInstEnum(Callee->getName());

      if (EInst) {
        uint32_t &ExtInstImportID = getOpExtInstImportID();

        //
        // Generate OpExtInst.
        //

        // Ops[0] = Result Type ID
        // Ops[1] = Set ID (OpExtInstImport ID)
        // Ops[2] = Instruction Number (Literal Number)
        // Ops[3] ... Ops[n] = Operand 1, ... , Operand n
        SPIRVOperandList Ops;

        uint32_t ResTyID = lookupType(Call->getType());
        SPIRVOperand *ResTyIDOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, ResTyID);
        Ops.push_back(ResTyIDOp);

        SPIRVOperand *SetIDOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, ExtInstImportID);
        Ops.push_back(SetIDOp);

        std::vector<uint32_t> LiteralNum;
        LiteralNum.push_back(EInst);
        SPIRVOperand *InstructionOp =
            new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, LiteralNum);
        Ops.push_back(InstructionOp);

        uint16_t WordCount = 5;

        FunctionType *CalleeFTy = cast<FunctionType>(Call->getFunctionType());
        for (unsigned i = 0; i < CalleeFTy->getNumParams(); i++) {
          uint32_t ArgID = VMap[Call->getOperand(i)];
          SPIRVOperand *ArgIDOp =
              new SPIRVOperand(SPIRVOperandType::NUMBERID, ArgID);
          Ops.push_back(ArgIDOp);
          WordCount++;
        }

        SPIRVInstruction *ExtInst = new SPIRVInstruction(
            WordCount, spv::OpExtInst, std::get<2>(*DeferredInst), Ops);
        SPIRVInstList.insert(InsertPoint, ExtInst);

        // clz needs OpExtInst and OpISub with constant 31.
        if (EInst == glsl::ExtInstFindUMsb) {
          LLVMContext &Context =
              Call->getParent()->getParent()->getParent()->getContext();
          //
          // Generate OpISub with constant 31.
          //
          // Ops[0] = Result Type ID
          // Ops[1] = Operand 0
          // Ops[2] = Operand 1
          Ops.clear();

          Ops.push_back(new SPIRVOperand(SPIRVOperandType::NUMBERID,
                                         lookupType(Call->getType())));

          Type *IdxTy = Type::getInt32Ty(Context);
          Constant *Cst31 = ConstantInt::get(IdxTy, 31);
          uint32_t Op0ID = VMap[Cst31];
          SPIRVOperand *Op0IDOp =
              new SPIRVOperand(SPIRVOperandType::NUMBERID, Op0ID);
          Ops.push_back(Op0IDOp);

          SPIRVOperand *Op1IDOp = new SPIRVOperand(SPIRVOperandType::NUMBERID,
                                                   std::get<2>(*DeferredInst));
          Ops.push_back(Op1IDOp);

          SPIRVInstList.insert(
              InsertPoint,
              new SPIRVInstruction(5, spv::OpISub,
                                   std::get<2>(*DeferredInst) + 1, Ops));
        }
      } else if (Callee->getName().equals("_Z8popcounti") ||
                 Callee->getName().equals("_Z8popcountj") ||
                 Callee->getName().equals("_Z8popcountDv2_i") ||
                 Callee->getName().equals("_Z8popcountDv3_i") ||
                 Callee->getName().equals("_Z8popcountDv4_i") ||
                 Callee->getName().equals("_Z8popcountDv2_j") ||
                 Callee->getName().equals("_Z8popcountDv3_j") ||
                 Callee->getName().equals("_Z8popcountDv4_j")) {
        //
        // Generate OpBitCount
        //
        // Ops[0] = Result Type ID
        // Ops[1] = Base ID
        SPIRVOperand *Ops[2]{new SPIRVOperand(SPIRVOperandType::NUMBERID,
                                              lookupType(Call->getType())),
                             new SPIRVOperand(SPIRVOperandType::NUMBERID,
                                              VMap[Call->getOperand(0)])};

        SPIRVInstList.insert(
            InsertPoint, new SPIRVInstruction(4, spv::OpBitCount,
                                              std::get<2>(*DeferredInst), Ops));
      } else {
        //
        // Generate OpFunctionCall.
        //

        // Ops[0] = Result Type ID
        // Ops[1] = Callee Function ID
        // Ops[2] ... Ops[n] = Argument 0, ... , Argument n
        SPIRVOperandList Ops;

        uint32_t ResTyID = lookupType(Call->getType());
        SPIRVOperand *ResTyIDOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, ResTyID);
        Ops.push_back(ResTyIDOp);

        uint32_t CalleeID = VMap[Callee];

        SPIRVOperand *CalleeIDOp =
            new SPIRVOperand(SPIRVOperandType::NUMBERID, CalleeID);
        Ops.push_back(CalleeIDOp);

        uint16_t WordCount = 4;

        FunctionType *CalleeFTy = cast<FunctionType>(Call->getFunctionType());
        for (unsigned i = 0; i < CalleeFTy->getNumParams(); i++) {
          uint32_t ArgID = VMap[Call->getOperand(i)];
          SPIRVOperand *ArgIDOp =
              new SPIRVOperand(SPIRVOperandType::NUMBERID, ArgID);
          Ops.push_back(ArgIDOp);
          WordCount++;
        }

        SPIRVInstruction *CallInst = new SPIRVInstruction(
            WordCount, spv::OpFunctionCall, std::get<2>(*DeferredInst), Ops);
        SPIRVInstList.insert(InsertPoint, CallInst);
      }
    }
  }
}

void SPIRVProducerPass::HandleDeferredDecorations(const DataLayout &DL) {
  // Insert ArrayStride decorations on pointer types, due to OpPtrAccessChain
  // instructions we generated earlier.
  if (getPointerTypesNeedingArrayStride().empty())
    return;

  SPIRVInstructionList &SPIRVInstList = getSPIRVInstList();
  ValueMapType &VMap = getValueMap();

  // Find an iterator pointing just past the last decoration.
  bool seen_decorations = false;
  auto DecoInsertPoint =
      std::find_if(SPIRVInstList.begin(), SPIRVInstList.end(),
                   [&seen_decorations](SPIRVInstruction *Inst) -> bool {
                     const bool is_decoration =
                         Inst->getOpcode() == spv::OpDecorate ||
                         Inst->getOpcode() == spv::OpMemberDecorate;
                     if (is_decoration) {
                       seen_decorations = true;
                       return false;
                     } else {
                       return seen_decorations;
                     }
                   });

  for (auto *type : getPointerTypesNeedingArrayStride()) {
    auto *ptrType = cast<PointerType>(type);

    // Ops[0] = Target ID
    // Ops[1] = Decoration (ArrayStride)
    // Ops[2] = Stride number (Literal Number)
    SPIRVOperandList Ops;

    Ops.push_back(
        new SPIRVOperand(SPIRVOperandType::NUMBERID, lookupType(ptrType)));
    Ops.push_back(new SPIRVOperand(SPIRVOperandType::NUMBERID,
                                   spv::DecorationArrayStride));
    Type *elemTy = ptrType->getElementType();
    // Same as DL.getIndexedOfffsetInType( elemTy, { 1 } );
    const unsigned stride = DL.getTypeAllocSize(elemTy);
    Ops.push_back(new SPIRVOperand(SPIRVOperandType::LITERAL_INTEGER, stride));

    SPIRVInstruction *DecoInst =
        new SPIRVInstruction(4, spv::OpDecorate, 0 /* No id */, Ops);
    SPIRVInstList.insert(DecoInsertPoint, DecoInst);
  }
}

glsl::ExtInst SPIRVProducerPass::getExtInstEnum(StringRef Name) {
  return StringSwitch<glsl::ExtInst>(Name)
      .Case("_Z3absi", glsl::ExtInst::ExtInstSAbs)
      .Case("_Z3absDv2_i", glsl::ExtInst::ExtInstSAbs)
      .Case("_Z3absDv3_i", glsl::ExtInst::ExtInstSAbs)
      .Case("_Z3absDv4_i", glsl::ExtInst::ExtInstSAbs)
      .Case("_Z5clampiii", glsl::ExtInst::ExtInstSClamp)
      .Case("_Z5clampDv2_iS_S_", glsl::ExtInst::ExtInstSClamp)
      .Case("_Z5clampDv3_iS_S_", glsl::ExtInst::ExtInstSClamp)
      .Case("_Z5clampDv4_iS_S_", glsl::ExtInst::ExtInstSClamp)
      .Case("_Z5clampjjj", glsl::ExtInst::ExtInstUClamp)
      .Case("_Z5clampDv2_jS_S_", glsl::ExtInst::ExtInstUClamp)
      .Case("_Z5clampDv3_jS_S_", glsl::ExtInst::ExtInstUClamp)
      .Case("_Z5clampDv4_jS_S_", glsl::ExtInst::ExtInstUClamp)
      .Case("_Z5clampfff", glsl::ExtInst::ExtInstFClamp)
      .Case("_Z5clampDv2_fS_S_", glsl::ExtInst::ExtInstFClamp)
      .Case("_Z5clampDv3_fS_S_", glsl::ExtInst::ExtInstFClamp)
      .Case("_Z5clampDv4_fS_S_", glsl::ExtInst::ExtInstFClamp)
      .StartsWith("_Z3clz", glsl::ExtInst::ExtInstFindUMsb)
      .Case("_Z3maxii", glsl::ExtInst::ExtInstSMax)
      .Case("_Z3maxDv2_iS_", glsl::ExtInst::ExtInstSMax)
      .Case("_Z3maxDv3_iS_", glsl::ExtInst::ExtInstSMax)
      .Case("_Z3maxDv4_iS_", glsl::ExtInst::ExtInstSMax)
      .Case("_Z3maxjj", glsl::ExtInst::ExtInstUMax)
      .Case("_Z3maxDv2_jS_", glsl::ExtInst::ExtInstUMax)
      .Case("_Z3maxDv3_jS_", glsl::ExtInst::ExtInstUMax)
      .Case("_Z3maxDv4_jS_", glsl::ExtInst::ExtInstUMax)
      .Case("_Z3maxff", glsl::ExtInst::ExtInstFMax)
      .Case("_Z3maxDv2_fS_", glsl::ExtInst::ExtInstFMax)
      .Case("_Z3maxDv3_fS_", glsl::ExtInst::ExtInstFMax)
      .Case("_Z3maxDv4_fS_", glsl::ExtInst::ExtInstFMax)
      .StartsWith("_Z4fmax", glsl::ExtInst::ExtInstFMax)
      .Case("_Z3minii", glsl::ExtInst::ExtInstSMin)
      .Case("_Z3minDv2_iS_", glsl::ExtInst::ExtInstSMin)
      .Case("_Z3minDv3_iS_", glsl::ExtInst::ExtInstSMin)
      .Case("_Z3minDv4_iS_", glsl::ExtInst::ExtInstSMin)
      .Case("_Z3minjj", glsl::ExtInst::ExtInstUMin)
      .Case("_Z3minDv2_jS_", glsl::ExtInst::ExtInstUMin)
      .Case("_Z3minDv3_jS_", glsl::ExtInst::ExtInstUMin)
      .Case("_Z3minDv4_jS_", glsl::ExtInst::ExtInstUMin)
      .Case("_Z3minff", glsl::ExtInst::ExtInstFMin)
      .Case("_Z3minDv2_fS_", glsl::ExtInst::ExtInstFMin)
      .Case("_Z3minDv3_fS_", glsl::ExtInst::ExtInstFMin)
      .Case("_Z3minDv4_fS_", glsl::ExtInst::ExtInstFMin)
      .StartsWith("_Z4fmin", glsl::ExtInst::ExtInstFMin)
      .StartsWith("_Z7degrees", glsl::ExtInst::ExtInstDegrees)
      .StartsWith("_Z7radians", glsl::ExtInst::ExtInstRadians)
      .StartsWith("_Z3mix", glsl::ExtInst::ExtInstFMix)
      .StartsWith("_Z4acos", glsl::ExtInst::ExtInstAcos)
      .StartsWith("_Z5acosh", glsl::ExtInst::ExtInstAcosh)
      .StartsWith("_Z4asin", glsl::ExtInst::ExtInstAsin)
      .StartsWith("_Z5asinh", glsl::ExtInst::ExtInstAsinh)
      .StartsWith("_Z4atan", glsl::ExtInst::ExtInstAtan)
      .StartsWith("_Z5atan2", glsl::ExtInst::ExtInstAtan2)
      .StartsWith("_Z5atanh", glsl::ExtInst::ExtInstAtanh)
      .StartsWith("_Z4ceil", glsl::ExtInst::ExtInstCeil)
      .StartsWith("_Z3sin", glsl::ExtInst::ExtInstSin)
      .StartsWith("_Z4sinh", glsl::ExtInst::ExtInstSinh)
      .StartsWith("_Z8half_sin", glsl::ExtInst::ExtInstSin)
      .StartsWith("_Z10native_sin", glsl::ExtInst::ExtInstSin)
      .StartsWith("_Z3cos", glsl::ExtInst::ExtInstCos)
      .StartsWith("_Z4cosh", glsl::ExtInst::ExtInstCosh)
      .StartsWith("_Z8half_cos", glsl::ExtInst::ExtInstCos)
      .StartsWith("_Z10native_cos", glsl::ExtInst::ExtInstCos)
      .StartsWith("_Z3tan", glsl::ExtInst::ExtInstTan)
      .StartsWith("_Z4tanh", glsl::ExtInst::ExtInstTanh)
      .StartsWith("_Z8half_tan", glsl::ExtInst::ExtInstTan)
      .StartsWith("_Z10native_tan", glsl::ExtInst::ExtInstTan)
      .StartsWith("_Z3exp", glsl::ExtInst::ExtInstExp)
      .StartsWith("_Z8half_exp", glsl::ExtInst::ExtInstExp)
      .StartsWith("_Z10native_exp", glsl::ExtInst::ExtInstExp)
      .StartsWith("_Z4exp2", glsl::ExtInst::ExtInstExp2)
      .StartsWith("_Z9half_exp2", glsl::ExtInst::ExtInstExp2)
      .StartsWith("_Z11native_exp2", glsl::ExtInst::ExtInstExp2)
      .StartsWith("_Z3log", glsl::ExtInst::ExtInstLog)
      .StartsWith("_Z8half_log", glsl::ExtInst::ExtInstLog)
      .StartsWith("_Z10native_log", glsl::ExtInst::ExtInstLog)
      .StartsWith("_Z4log2", glsl::ExtInst::ExtInstLog2)
      .StartsWith("_Z9half_log2", glsl::ExtInst::ExtInstLog2)
      .StartsWith("_Z11native_log2", glsl::ExtInst::ExtInstLog2)
      .StartsWith("_Z4fabs", glsl::ExtInst::ExtInstFAbs)
      .StartsWith("_Z5floor", glsl::ExtInst::ExtInstFloor)
      .StartsWith("_Z5ldexp", glsl::ExtInst::ExtInstLdexp)
      .StartsWith("_Z3pow", glsl::ExtInst::ExtInstPow)
      .StartsWith("_Z4powr", glsl::ExtInst::ExtInstPow)
      .StartsWith("_Z9half_powr", glsl::ExtInst::ExtInstPow)
      .StartsWith("_Z11native_powr", glsl::ExtInst::ExtInstPow)
      .StartsWith("_Z5round", glsl::ExtInst::ExtInstRound)
      .StartsWith("_Z4sqrt", glsl::ExtInst::ExtInstSqrt)
      .StartsWith("_Z9half_sqrt", glsl::ExtInst::ExtInstSqrt)
      .StartsWith("_Z11native_sqrt", glsl::ExtInst::ExtInstSqrt)
      .StartsWith("_Z5rsqrt", glsl::ExtInst::ExtInstInverseSqrt)
      .StartsWith("_Z10half_rsqrt", glsl::ExtInst::ExtInstInverseSqrt)
      .StartsWith("_Z12native_rsqrt", glsl::ExtInst::ExtInstInverseSqrt)
      .StartsWith("_Z5trunc", glsl::ExtInst::ExtInstTrunc)
      .StartsWith("_Z5frexp", glsl::ExtInst::ExtInstFrexp)
      .StartsWith("_Z4sign", glsl::ExtInst::ExtInstFSign)
      .StartsWith("_Z6length", glsl::ExtInst::ExtInstLength)
      .StartsWith("_Z8distance", glsl::ExtInst::ExtInstDistance)
      .Case("_Z5crossDv3_fS_", glsl::ExtInst::ExtInstCross)
      .StartsWith("_Z9normalize", glsl::ExtInst::ExtInstNormalize)
      .StartsWith("llvm.fmuladd.", glsl::ExtInst::ExtInstFma)
      .Case("spirv.unpack.v2f16", glsl::ExtInst::ExtInstUnpackHalf2x16)
      .Case("spirv.pack.v2f16", glsl::ExtInst::ExtInstPackHalf2x16)
      .Default(static_cast<glsl::ExtInst>(0));
}

void SPIRVProducerPass::PrintResID(SPIRVInstruction *Inst) {
  out << "%" << Inst->getResultID();
}

void SPIRVProducerPass::PrintOpcode(SPIRVInstruction *Inst) {
  spv::Op Opcode = static_cast<spv::Op>(Inst->getOpcode());
  out << "\t" << spv::getOpName(Opcode);
}

void SPIRVProducerPass::PrintOperand(SPIRVOperand *Op) {
  SPIRVOperandType OpTy = Op->getType();
  switch (OpTy) {
  default: {
    llvm_unreachable("Unsupported SPIRV Operand Type???");
    break;
  }
  case SPIRVOperandType::NUMBERID: {
    out << "%" << Op->getNumID();
    break;
  }
  case SPIRVOperandType::LITERAL_STRING: {
    out << "\"" << Op->getLiteralStr() << "\"";
    break;
  }
  case SPIRVOperandType::LITERAL_INTEGER: {
    // TODO: Handle LiteralNum carefully.
    for (auto Word : Op->getLiteralNum()) {
      out << Word;
    }
    break;
  }
  case SPIRVOperandType::LITERAL_FLOAT: {
    // TODO: Handle LiteralNum carefully.
    for (auto Word : Op->getLiteralNum()) {
      APFloat APF = APFloat(APFloat::IEEEsingle(), APInt(32, Word));
      SmallString<8> Str;
      APF.toString(Str, 6, 2);
      out << Str;
    }
    break;
  }
  }
}

void SPIRVProducerPass::PrintCapability(SPIRVOperand *Op) {
  spv::Capability Cap = static_cast<spv::Capability>(Op->getNumID());
  out << spv::getCapabilityName(Cap);
}

void SPIRVProducerPass::PrintExtInst(SPIRVOperand *Op) {
  auto LiteralNum = Op->getLiteralNum();
  glsl::ExtInst Ext = static_cast<glsl::ExtInst>(LiteralNum[0]);
  out << glsl::getExtInstName(Ext);
}

void SPIRVProducerPass::PrintAddrModel(SPIRVOperand *Op) {
  spv::AddressingModel AddrModel =
      static_cast<spv::AddressingModel>(Op->getNumID());
  out << spv::getAddressingModelName(AddrModel);
}

void SPIRVProducerPass::PrintMemModel(SPIRVOperand *Op) {
  spv::MemoryModel MemModel = static_cast<spv::MemoryModel>(Op->getNumID());
  out << spv::getMemoryModelName(MemModel);
}

void SPIRVProducerPass::PrintExecModel(SPIRVOperand *Op) {
  spv::ExecutionModel ExecModel =
      static_cast<spv::ExecutionModel>(Op->getNumID());
  out << spv::getExecutionModelName(ExecModel);
}

void SPIRVProducerPass::PrintExecMode(SPIRVOperand *Op) {
  spv::ExecutionMode ExecMode = static_cast<spv::ExecutionMode>(Op->getNumID());
  out << spv::getExecutionModeName(ExecMode);
}

void SPIRVProducerPass::PrintSourceLanguage(SPIRVOperand *Op) {
  spv::SourceLanguage SourceLang = static_cast<spv::SourceLanguage>(Op->getNumID());
  out << spv::getSourceLanguageName(SourceLang);
}

void SPIRVProducerPass::PrintFuncCtrl(SPIRVOperand *Op) {
  spv::FunctionControlMask FuncCtrl =
      static_cast<spv::FunctionControlMask>(Op->getNumID());
  out << spv::getFunctionControlName(FuncCtrl);
}

void SPIRVProducerPass::PrintStorageClass(SPIRVOperand *Op) {
  spv::StorageClass StClass = static_cast<spv::StorageClass>(Op->getNumID());
  out << getStorageClassName(StClass);
}

void SPIRVProducerPass::PrintDecoration(SPIRVOperand *Op) {
  spv::Decoration Deco = static_cast<spv::Decoration>(Op->getNumID());
  out << getDecorationName(Deco);
}

void SPIRVProducerPass::PrintBuiltIn(SPIRVOperand *Op) {
  spv::BuiltIn BIn = static_cast<spv::BuiltIn>(Op->getNumID());
  out << getBuiltInName(BIn);
}

void SPIRVProducerPass::PrintSelectionControl(SPIRVOperand *Op) {
  spv::SelectionControlMask BIn =
      static_cast<spv::SelectionControlMask>(Op->getNumID());
  out << getSelectionControlName(BIn);
}

void SPIRVProducerPass::PrintLoopControl(SPIRVOperand *Op) {
  spv::LoopControlMask BIn = static_cast<spv::LoopControlMask>(Op->getNumID());
  out << getLoopControlName(BIn);
}

void SPIRVProducerPass::PrintDimensionality(SPIRVOperand *Op) {
  spv::Dim DIM = static_cast<spv::Dim>(Op->getNumID());
  out << getDimName(DIM);
}

void SPIRVProducerPass::PrintImageFormat(SPIRVOperand *Op) {
  spv::ImageFormat Format = static_cast<spv::ImageFormat>(Op->getNumID());
  out << getImageFormatName(Format);
}

void SPIRVProducerPass::PrintMemoryAccess(SPIRVOperand *Op) {
  out << spv::getMemoryAccessName(
      static_cast<spv::MemoryAccessMask>(Op->getNumID()));
}

void SPIRVProducerPass::PrintImageOperandsType(SPIRVOperand *Op) {
  auto LiteralNum = Op->getLiteralNum();
  spv::ImageOperandsMask Type =
      static_cast<spv::ImageOperandsMask>(LiteralNum[0]);
  out << getImageOperandsName(Type);
}

void SPIRVProducerPass::WriteSPIRVAssembly() {
  SPIRVInstructionList &SPIRVInstList = getSPIRVInstList();

  for (auto Inst : SPIRVInstList) {
    SPIRVOperandList Ops = Inst->getOperands();
    spv::Op Opcode = static_cast<spv::Op>(Inst->getOpcode());

    switch (Opcode) {
    default: {
      llvm_unreachable("Unsupported SPIRV instruction");
      break;
    }
    case spv::OpCapability: {
      // Ops[0] = Capability
      PrintOpcode(Inst);
      out << " ";
      PrintCapability(Ops[0]);
      out << "\n";
      break;
    }
    case spv::OpMemoryModel: {
      // Ops[0] = Addressing Model
      // Ops[1] = Memory Model
      PrintOpcode(Inst);
      out << " ";
      PrintAddrModel(Ops[0]);
      out << " ";
      PrintMemModel(Ops[1]);
      out << "\n";
      break;
    }
    case spv::OpEntryPoint: {
      // Ops[0] = Execution Model
      // Ops[1] = EntryPoint ID
      // Ops[2] = Name (Literal String)
      // Ops[3] ... Ops[n] = Interface ID
      PrintOpcode(Inst);
      out << " ";
      PrintExecModel(Ops[0]);
      for (uint32_t i = 1; i < Ops.size(); i++) {
        out << " ";
        PrintOperand(Ops[i]);
      }
      out << "\n";
      break;
    }
    case spv::OpExecutionMode: {
      // Ops[0] = Entry Point ID
      // Ops[1] = Execution Mode
      // Ops[2] ... Ops[n] = Optional literals according to Execution Mode
      PrintOpcode(Inst);
      out << " ";
      PrintOperand(Ops[0]);
      out << " ";
      PrintExecMode(Ops[1]);
      for (uint32_t i = 2; i < Ops.size(); i++) {
        out << " ";
        PrintOperand(Ops[i]);
      }
      out << "\n";
      break;
    }
    case spv::OpSource: {
      // Ops[0] = SourceLanguage ID
      // Ops[1] = Version (LiteralNum)
      PrintOpcode(Inst);
      out << " ";
      PrintSourceLanguage(Ops[0]);
      out << " ";
      PrintOperand(Ops[1]);
      out << "\n";
      break;
    }
    case spv::OpDecorate: {
      // Ops[0] = Target ID
      // Ops[1] = Decoration (Block or BufferBlock)
      // Ops[2] ... Ops[n] = Optional literals according to Decoration
      PrintOpcode(Inst);
      out << " ";
      PrintOperand(Ops[0]);
      out << " ";
      PrintDecoration(Ops[1]);
      // Handle BuiltIn OpDecorate specially.
      if (Ops[1]->getNumID() == spv::DecorationBuiltIn) {
        out << " ";
        PrintBuiltIn(Ops[2]);
      } else {
        for (uint32_t i = 2; i < Ops.size(); i++) {
          out << " ";
          PrintOperand(Ops[i]);
        }
      }
      out << "\n";
      break;
    }
    case spv::OpMemberDecorate: {
      // Ops[0] = Structure Type ID
      // Ops[1] = Member Index(Literal Number)
      // Ops[2] = Decoration
      // Ops[3] ... Ops[n] = Optional literals according to Decoration
      PrintOpcode(Inst);
      out << " ";
      PrintOperand(Ops[0]);
      out << " ";
      PrintOperand(Ops[1]);
      out << " ";
      PrintDecoration(Ops[2]);
      for (uint32_t i = 3; i < Ops.size(); i++) {
        out << " ";
        PrintOperand(Ops[i]);
      }
      out << "\n";
      break;
    }
    case spv::OpTypePointer: {
      // Ops[0] = Storage Class
      // Ops[1] = Element Type ID
      PrintResID(Inst);
      out << " = ";
      PrintOpcode(Inst);
      out << " ";
      PrintStorageClass(Ops[0]);
      out << " ";
      PrintOperand(Ops[1]);
      out << "\n";
      break;
    }
    case spv::OpTypeImage: {
      // Ops[0] = Sampled Type ID
      // Ops[1] = Dim ID
      // Ops[2] = Depth (Literal Number)
      // Ops[3] = Arrayed (Literal Number)
      // Ops[4] = MS (Literal Number)
      // Ops[5] = Sampled (Literal Number)
      // Ops[6] = Image Format ID
      PrintResID(Inst);
      out << " = ";
      PrintOpcode(Inst);
      out << " ";
      PrintOperand(Ops[0]);
      out << " ";
      PrintDimensionality(Ops[1]);
      out << " ";
      PrintOperand(Ops[2]);
      out << " ";
      PrintOperand(Ops[3]);
      out << " ";
      PrintOperand(Ops[4]);
      out << " ";
      PrintOperand(Ops[5]);
      out << " ";
      PrintImageFormat(Ops[6]);
      out << "\n";
      break;
    }
    case spv::OpFunction: {
      // Ops[0] : Result Type ID
      // Ops[1] : Function Control
      // Ops[2] : Function Type ID
      PrintResID(Inst);
      out << " = ";
      PrintOpcode(Inst);
      out << " ";
      PrintOperand(Ops[0]);
      out << " ";
      PrintFuncCtrl(Ops[1]);
      out << " ";
      PrintOperand(Ops[2]);
      out << "\n";
      break;
    }
    case spv::OpSelectionMerge: {
      // Ops[0] = Merge Block ID
      // Ops[1] = Selection Control
      PrintOpcode(Inst);
      out << " ";
      PrintOperand(Ops[0]);
      out << " ";
      PrintSelectionControl(Ops[1]);
      out << "\n";
      break;
    }
    case spv::OpLoopMerge: {
      // Ops[0] = Merge Block ID
      // Ops[1] = Continue Target ID
      // Ops[2] = Selection Control
      PrintOpcode(Inst);
      out << " ";
      PrintOperand(Ops[0]);
      out << " ";
      PrintOperand(Ops[1]);
      out << " ";
      PrintLoopControl(Ops[2]);
      out << "\n";
      break;
    }
    case spv::OpImageSampleExplicitLod: {
      // Ops[0] = Result Type ID
      // Ops[1] = Sampled Image ID
      // Ops[2] = Coordinate ID
      // Ops[3] = Image Operands Type ID
      // Ops[4] ... Ops[n] = Operands ID
      PrintResID(Inst);
      out << " = ";
      PrintOpcode(Inst);
      for (uint32_t i = 0; i < 3; i++) {
        out << " ";
        PrintOperand(Ops[i]);
      }
      out << " ";
      PrintImageOperandsType(Ops[3]);
      for (uint32_t i = 4; i < Ops.size(); i++) {
        out << " ";
        PrintOperand(Ops[i]);
      }
      out << "\n";
      break;
    }
    case spv::OpVariable: {
      // Ops[0] : Result Type ID
      // Ops[1] : Storage Class
      // Ops[2] ... Ops[n] = Initializer IDs
      PrintResID(Inst);
      out << " = ";
      PrintOpcode(Inst);
      out << " ";
      PrintOperand(Ops[0]);
      out << " ";
      PrintStorageClass(Ops[1]);
      for (uint32_t i = 2; i < Ops.size(); i++) {
        out << " ";
        PrintOperand(Ops[i]);
      }
      out << "\n";
      break;
    }
    case spv::OpExtInst: {
      // Ops[0] = Result Type ID
      // Ops[1] = Set ID (OpExtInstImport ID)
      // Ops[2] = Instruction Number (Literal Number)
      // Ops[3] ... Ops[n] = Operand 1, ... , Operand n
      PrintResID(Inst);
      out << " = ";
      PrintOpcode(Inst);
      out << " ";
      PrintOperand(Ops[0]);
      out << " ";
      PrintOperand(Ops[1]);
      out << " ";
      PrintExtInst(Ops[2]);
      for (uint32_t i = 3; i < Ops.size(); i++) {
        out << " ";
        PrintOperand(Ops[i]);
      }
      out << "\n";
      break;
    }
    case spv::OpCopyMemory: {
      // Ops[0] = Addressing Model
      // Ops[1] = Memory Model
      PrintOpcode(Inst);
      out << " ";
      PrintOperand(Ops[0]);
      out << " ";
      PrintOperand(Ops[1]);
      out << " ";
      PrintMemoryAccess(Ops[2]);
      out << " ";
      PrintOperand(Ops[3]);
      out << "\n";
      break;
    }
    case spv::OpExtension:
    case spv::OpControlBarrier:
    case spv::OpMemoryBarrier:
    case spv::OpBranch:
    case spv::OpBranchConditional:
    case spv::OpStore:
    case spv::OpImageWrite:
    case spv::OpReturnValue:
    case spv::OpReturn:
    case spv::OpFunctionEnd: {
      PrintOpcode(Inst);
      for (uint32_t i = 0; i < Ops.size(); i++) {
        out << " ";
        PrintOperand(Ops[i]);
      }
      out << "\n";
      break;
    }
    case spv::OpExtInstImport:
    case spv::OpTypeRuntimeArray:
    case spv::OpTypeStruct:
    case spv::OpTypeSampler:
    case spv::OpTypeSampledImage:
    case spv::OpTypeInt:
    case spv::OpTypeFloat:
    case spv::OpTypeArray:
    case spv::OpTypeVector:
    case spv::OpTypeBool:
    case spv::OpTypeVoid:
    case spv::OpTypeFunction:
    case spv::OpFunctionParameter:
    case spv::OpLabel:
    case spv::OpPhi:
    case spv::OpLoad:
    case spv::OpSelect:
    case spv::OpAccessChain:
    case spv::OpPtrAccessChain:
    case spv::OpInBoundsAccessChain:
    case spv::OpUConvert:
    case spv::OpSConvert:
    case spv::OpConvertFToU:
    case spv::OpConvertFToS:
    case spv::OpConvertUToF:
    case spv::OpConvertSToF:
    case spv::OpFConvert:
    case spv::OpConvertPtrToU:
    case spv::OpConvertUToPtr:
    case spv::OpBitcast:
    case spv::OpIAdd:
    case spv::OpFAdd:
    case spv::OpISub:
    case spv::OpFSub:
    case spv::OpIMul:
    case spv::OpFMul:
    case spv::OpUDiv:
    case spv::OpSDiv:
    case spv::OpFDiv:
    case spv::OpUMod:
    case spv::OpSRem:
    case spv::OpFRem:
    case spv::OpBitwiseOr:
    case spv::OpBitwiseXor:
    case spv::OpBitwiseAnd:
    case spv::OpShiftLeftLogical:
    case spv::OpShiftRightLogical:
    case spv::OpShiftRightArithmetic:
    case spv::OpBitCount:
    case spv::OpCompositeExtract:
    case spv::OpVectorExtractDynamic:
    case spv::OpCompositeInsert:
    case spv::OpVectorInsertDynamic:
    case spv::OpVectorShuffle:
    case spv::OpIEqual:
    case spv::OpINotEqual:
    case spv::OpUGreaterThan:
    case spv::OpUGreaterThanEqual:
    case spv::OpULessThan:
    case spv::OpULessThanEqual:
    case spv::OpSGreaterThan:
    case spv::OpSGreaterThanEqual:
    case spv::OpSLessThan:
    case spv::OpSLessThanEqual:
    case spv::OpFOrdEqual:
    case spv::OpFOrdGreaterThan:
    case spv::OpFOrdGreaterThanEqual:
    case spv::OpFOrdLessThan:
    case spv::OpFOrdLessThanEqual:
    case spv::OpFOrdNotEqual:
    case spv::OpFUnordEqual:
    case spv::OpFUnordGreaterThan:
    case spv::OpFUnordGreaterThanEqual:
    case spv::OpFUnordLessThan:
    case spv::OpFUnordLessThanEqual:
    case spv::OpFUnordNotEqual:
    case spv::OpSampledImage:
    case spv::OpFunctionCall:
    case spv::OpConstantTrue:
    case spv::OpConstantFalse:
    case spv::OpConstant:
    case spv::OpSpecConstant:
    case spv::OpConstantComposite:
    case spv::OpSpecConstantComposite:
    case spv::OpConstantNull:
    case spv::OpLogicalOr:
    case spv::OpLogicalAnd:
    case spv::OpLogicalNot:
    case spv::OpLogicalNotEqual:
    case spv::OpUndef:
    case spv::OpIsInf:
    case spv::OpIsNan:
    case spv::OpAny:
    case spv::OpAll:
    case spv::OpAtomicIAdd:
    case spv::OpAtomicISub:
    case spv::OpAtomicExchange:
    case spv::OpAtomicIIncrement:
    case spv::OpAtomicIDecrement:
    case spv::OpAtomicCompareExchange:
    case spv::OpAtomicUMin:
    case spv::OpAtomicSMin:
    case spv::OpAtomicUMax:
    case spv::OpAtomicSMax:
    case spv::OpAtomicAnd:
    case spv::OpAtomicOr:
    case spv::OpAtomicXor:
    case spv::OpDot: {
      PrintResID(Inst);
      out << " = ";
      PrintOpcode(Inst);
      for (uint32_t i = 0; i < Ops.size(); i++) {
        out << " ";
        PrintOperand(Ops[i]);
      }
      out << "\n";
      break;
    }
    }
  }
}

void SPIRVProducerPass::WriteOneWord(uint32_t Word) {
  binaryOut->write(reinterpret_cast<const char *>(&Word), sizeof(uint32_t));
}

void SPIRVProducerPass::WriteResultID(SPIRVInstruction *Inst) {
  WriteOneWord(Inst->getResultID());
}

void SPIRVProducerPass::WriteWordCountAndOpcode(SPIRVInstruction *Inst) {
  // High 16 bit : Word Count
  // Low 16 bit  : Opcode
  uint32_t Word = Inst->getOpcode();
  Word |= Inst->getWordCount() << 16;
  WriteOneWord(Word);
}

void SPIRVProducerPass::WriteOperand(SPIRVOperand *Op) {
  SPIRVOperandType OpTy = Op->getType();
  switch (OpTy) {
  default: {
    llvm_unreachable("Unsupported SPIRV Operand Type???");
    break;
  }
  case SPIRVOperandType::NUMBERID: {
    WriteOneWord(Op->getNumID());
    break;
  }
  case SPIRVOperandType::LITERAL_STRING: {
    std::string Str = Op->getLiteralStr();
    const char *Data = Str.c_str();
    size_t WordSize = Str.size() / 4;
    for (unsigned Idx = 0; Idx < WordSize; Idx++) {
      WriteOneWord(*reinterpret_cast<const uint32_t *>(&Data[4 * Idx]));
    }

    uint32_t Remainder = Str.size() % 4;
    uint32_t LastWord = 0;
    if (Remainder) {
      for (unsigned Idx = 0; Idx < Remainder; Idx++) {
        LastWord |= Data[4 * WordSize + Idx] << 8 * Idx;
      }
    }

    WriteOneWord(LastWord);
    break;
  }
  case SPIRVOperandType::LITERAL_INTEGER:
  case SPIRVOperandType::LITERAL_FLOAT: {
    auto LiteralNum = Op->getLiteralNum();
    // TODO: Handle LiteranNum carefully.
    for (auto Word : LiteralNum) {
      WriteOneWord(Word);
    }
    break;
  }
  }
}

void SPIRVProducerPass::WriteSPIRVBinary() {
  SPIRVInstructionList &SPIRVInstList = getSPIRVInstList();

  for (auto Inst : SPIRVInstList) {
    SPIRVOperandList Ops = Inst->getOperands();
    spv::Op Opcode = static_cast<spv::Op>(Inst->getOpcode());

    switch (Opcode) {
    default: {
      llvm_unreachable("Unsupported SPIRV instruction");
      break;
    }
    case spv::OpCapability:
    case spv::OpExtension:
    case spv::OpMemoryModel:
    case spv::OpEntryPoint:
    case spv::OpExecutionMode:
    case spv::OpSource:
    case spv::OpDecorate:
    case spv::OpMemberDecorate:
    case spv::OpBranch:
    case spv::OpBranchConditional:
    case spv::OpSelectionMerge:
    case spv::OpLoopMerge:
    case spv::OpStore:
    case spv::OpImageWrite:
    case spv::OpReturnValue:
    case spv::OpControlBarrier:
    case spv::OpMemoryBarrier:
    case spv::OpReturn:
    case spv::OpFunctionEnd:
    case spv::OpCopyMemory: {
      WriteWordCountAndOpcode(Inst);
      for (uint32_t i = 0; i < Ops.size(); i++) {
        WriteOperand(Ops[i]);
      }
      break;
    }
    case spv::OpTypeBool:
    case spv::OpTypeVoid:
    case spv::OpTypeSampler:
    case spv::OpLabel:
    case spv::OpExtInstImport:
    case spv::OpTypePointer:
    case spv::OpTypeRuntimeArray:
    case spv::OpTypeStruct:
    case spv::OpTypeImage:
    case spv::OpTypeSampledImage:
    case spv::OpTypeInt:
    case spv::OpTypeFloat:
    case spv::OpTypeArray:
    case spv::OpTypeVector:
    case spv::OpTypeFunction: {
      WriteWordCountAndOpcode(Inst);
      WriteResultID(Inst);
      for (uint32_t i = 0; i < Ops.size(); i++) {
        WriteOperand(Ops[i]);
      }
      break;
    }
    case spv::OpFunction:
    case spv::OpFunctionParameter:
    case spv::OpAccessChain:
    case spv::OpPtrAccessChain:
    case spv::OpInBoundsAccessChain:
    case spv::OpUConvert:
    case spv::OpSConvert:
    case spv::OpConvertFToU:
    case spv::OpConvertFToS:
    case spv::OpConvertUToF:
    case spv::OpConvertSToF:
    case spv::OpFConvert:
    case spv::OpConvertPtrToU:
    case spv::OpConvertUToPtr:
    case spv::OpBitcast:
    case spv::OpIAdd:
    case spv::OpFAdd:
    case spv::OpISub:
    case spv::OpFSub:
    case spv::OpIMul:
    case spv::OpFMul:
    case spv::OpUDiv:
    case spv::OpSDiv:
    case spv::OpFDiv:
    case spv::OpUMod:
    case spv::OpSRem:
    case spv::OpFRem:
    case spv::OpBitwiseOr:
    case spv::OpBitwiseXor:
    case spv::OpBitwiseAnd:
    case spv::OpShiftLeftLogical:
    case spv::OpShiftRightLogical:
    case spv::OpShiftRightArithmetic:
    case spv::OpBitCount:
    case spv::OpCompositeExtract:
    case spv::OpVectorExtractDynamic:
    case spv::OpCompositeInsert:
    case spv::OpVectorInsertDynamic:
    case spv::OpVectorShuffle:
    case spv::OpIEqual:
    case spv::OpINotEqual:
    case spv::OpUGreaterThan:
    case spv::OpUGreaterThanEqual:
    case spv::OpULessThan:
    case spv::OpULessThanEqual:
    case spv::OpSGreaterThan:
    case spv::OpSGreaterThanEqual:
    case spv::OpSLessThan:
    case spv::OpSLessThanEqual:
    case spv::OpFOrdEqual:
    case spv::OpFOrdGreaterThan:
    case spv::OpFOrdGreaterThanEqual:
    case spv::OpFOrdLessThan:
    case spv::OpFOrdLessThanEqual:
    case spv::OpFOrdNotEqual:
    case spv::OpFUnordEqual:
    case spv::OpFUnordGreaterThan:
    case spv::OpFUnordGreaterThanEqual:
    case spv::OpFUnordLessThan:
    case spv::OpFUnordLessThanEqual:
    case spv::OpFUnordNotEqual:
    case spv::OpExtInst:
    case spv::OpIsInf:
    case spv::OpIsNan:
    case spv::OpAny:
    case spv::OpAll:
    case spv::OpUndef:
    case spv::OpConstantNull:
    case spv::OpLogicalOr:
    case spv::OpLogicalAnd:
    case spv::OpLogicalNot:
    case spv::OpLogicalNotEqual:
    case spv::OpConstantComposite:
    case spv::OpSpecConstantComposite:
    case spv::OpConstantTrue:
    case spv::OpConstantFalse:
    case spv::OpConstant:
    case spv::OpSpecConstant:
    case spv::OpVariable:
    case spv::OpFunctionCall:
    case spv::OpSampledImage:
    case spv::OpImageSampleExplicitLod:
    case spv::OpSelect:
    case spv::OpPhi:
    case spv::OpLoad:
    case spv::OpAtomicIAdd:
    case spv::OpAtomicISub:
    case spv::OpAtomicExchange:
    case spv::OpAtomicIIncrement:
    case spv::OpAtomicIDecrement:
    case spv::OpAtomicCompareExchange:
    case spv::OpAtomicUMin:
    case spv::OpAtomicSMin:
    case spv::OpAtomicUMax:
    case spv::OpAtomicSMax:
    case spv::OpAtomicAnd:
    case spv::OpAtomicOr:
    case spv::OpAtomicXor:
    case spv::OpDot: {
      WriteWordCountAndOpcode(Inst);
      WriteOperand(Ops[0]);
      WriteResultID(Inst);
      for (uint32_t i = 1; i < Ops.size(); i++) {
        WriteOperand(Ops[i]);
      }
      break;
    }
    }
  }
}