b8e1544b9d
The patch adds SPIRVPrepareFunctions pass, which modifies function signatures containing aggregate arguments and/or return values before IR translation. Information about the original signatures is stored in metadata. It is used during call lowering to restore correct SPIR-V types of function arguments and return values. This pass also substitutes some llvm intrinsic calls to function calls, generating the necessary functions in the module, as the SPIRV translator does. The patch also includes changes in other modules, fixing errors and enabling many SPIR-V features that were omitted earlier. And 15 LIT tests are also added to demonstrate the new functionality. Differential Revision: https://reviews.llvm.org/D129730 Co-authored-by: Aleksandr Bezzubikov <zuban32s@gmail.com> Co-authored-by: Michal Paszkowski <michal.paszkowski@outlook.com> Co-authored-by: Andrey Tretyakov <andrey1.tretyakov@intel.com> Co-authored-by: Konrad Trifunovic <konrad.trifunovic@intel.com>
64 lines
2.2 KiB
C++
64 lines
2.2 KiB
C++
//=- SPIRVMCInstLower.cpp - Convert SPIR-V MachineInstr to MCInst -*- C++ -*-=//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file contains code to lower SPIR-V MachineInstrs to their corresponding
|
|
// MCInst records.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "SPIRVMCInstLower.h"
|
|
#include "SPIRV.h"
|
|
#include "SPIRVModuleAnalysis.h"
|
|
#include "SPIRVUtils.h"
|
|
#include "llvm/CodeGen/MachineInstr.h"
|
|
#include "llvm/IR/Constants.h"
|
|
|
|
using namespace llvm;
|
|
|
|
void SPIRVMCInstLower::lower(const MachineInstr *MI, MCInst &OutMI,
|
|
SPIRV::ModuleAnalysisInfo *MAI) const {
|
|
OutMI.setOpcode(MI->getOpcode());
|
|
const MachineFunction *MF = MI->getMF();
|
|
for (unsigned i = 0, e = MI->getNumOperands(); i != e; ++i) {
|
|
const MachineOperand &MO = MI->getOperand(i);
|
|
MCOperand MCOp;
|
|
switch (MO.getType()) {
|
|
default:
|
|
llvm_unreachable("unknown operand type");
|
|
case MachineOperand::MO_GlobalAddress: {
|
|
Register FuncReg = MAI->getFuncReg(MO.getGlobal()->getGlobalIdentifier());
|
|
assert(FuncReg.isValid() && "Cannot find function Id");
|
|
MCOp = MCOperand::createReg(FuncReg);
|
|
break;
|
|
}
|
|
case MachineOperand::MO_MachineBasicBlock:
|
|
MCOp = MCOperand::createReg(MAI->getOrCreateMBBRegister(*MO.getMBB()));
|
|
break;
|
|
case MachineOperand::MO_Register: {
|
|
Register NewReg = MAI->getRegisterAlias(MF, MO.getReg());
|
|
MCOp = MCOperand::createReg(NewReg.isValid() ? NewReg : MO.getReg());
|
|
break;
|
|
}
|
|
case MachineOperand::MO_Immediate:
|
|
if (MI->getOpcode() == SPIRV::OpExtInst && i == 2) {
|
|
Register Reg = MAI->getExtInstSetReg(MO.getImm());
|
|
MCOp = MCOperand::createReg(Reg);
|
|
} else {
|
|
MCOp = MCOperand::createImm(MO.getImm());
|
|
}
|
|
break;
|
|
case MachineOperand::MO_FPImmediate:
|
|
MCOp = MCOperand::createDFPImm(
|
|
MO.getFPImm()->getValueAPF().convertToFloat());
|
|
break;
|
|
}
|
|
|
|
OutMI.addOperand(MCOp);
|
|
}
|
|
}
|