From 96db5f238ebb4254fca19c8b734d76a0442c0293 Mon Sep 17 00:00:00 2001 From: Jessica Paquette Date: Tue, 12 Feb 2019 17:38:34 +0000 Subject: [PATCH] [GlobalISel][NFC] Gardening: Make translateSimpleUnaryIntrinsic general Instead of only having this code work for unary intrinsics, have it work for an arbitrary number of parameters. Factor out the cases that fall under this (fma, pow). This makes it a bit easier to add more intrinsics which don't require any special work. Differential Revision: https://reviews.llvm.org/D58079 git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@353863 91177308-0d34-0410-b5e6-96231b3b80d8 --- .../llvm/CodeGen/GlobalISel/IRTranslator.h | 12 ++--- lib/CodeGen/GlobalISel/IRTranslator.cpp | 44 ++++++++----------- 2 files changed, 24 insertions(+), 32 deletions(-) diff --git a/include/llvm/CodeGen/GlobalISel/IRTranslator.h b/include/llvm/CodeGen/GlobalISel/IRTranslator.h index 8844260116c..c75d823501a 100644 --- a/include/llvm/CodeGen/GlobalISel/IRTranslator.h +++ b/include/llvm/CodeGen/GlobalISel/IRTranslator.h @@ -216,16 +216,16 @@ private: bool translateOverflowIntrinsic(const CallInst &CI, unsigned Op, MachineIRBuilder &MIRBuilder); - /// Helper function for translateSimpleUnaryIntrinsic. + /// Helper function for translateSimpleIntrinsic. /// \return The generic opcode for \p IntrinsicID if \p IntrinsicID is a - /// simple unary intrinsic (ceil, fabs, etc.). Otherwise, returns + /// simple intrinsic (ceil, fabs, etc.). Otherwise, returns /// Intrinsic::not_intrinsic. - unsigned getSimpleUnaryIntrinsicOpcode(Intrinsic::ID ID); + unsigned getSimpleIntrinsicOpcode(Intrinsic::ID ID); - /// Translates the intrinsics defined in getSimpleUnaryIntrinsicOpcode. + /// Translates the intrinsics defined in getSimpleIntrinsicOpcode. /// \return true if the translation succeeded. - bool translateSimpleUnaryIntrinsic(const CallInst &CI, Intrinsic::ID ID, - MachineIRBuilder &MIRBuilder); + bool translateSimpleIntrinsic(const CallInst &CI, Intrinsic::ID ID, + MachineIRBuilder &MIRBuilder); bool translateKnownIntrinsic(const CallInst &CI, Intrinsic::ID ID, MachineIRBuilder &MIRBuilder); diff --git a/lib/CodeGen/GlobalISel/IRTranslator.cpp b/lib/CodeGen/GlobalISel/IRTranslator.cpp index dd4e0df3188..7db54079af1 100644 --- a/lib/CodeGen/GlobalISel/IRTranslator.cpp +++ b/lib/CodeGen/GlobalISel/IRTranslator.cpp @@ -789,8 +789,7 @@ bool IRTranslator::translateOverflowIntrinsic(const CallInst &CI, unsigned Op, return true; } -unsigned -IRTranslator::getSimpleUnaryIntrinsicOpcode(Intrinsic::ID ID) { +unsigned IRTranslator::getSimpleIntrinsicOpcode(Intrinsic::ID ID) { switch (ID) { default: break; @@ -812,12 +811,16 @@ IRTranslator::getSimpleUnaryIntrinsicOpcode(Intrinsic::ID ID) { return TargetOpcode::G_FCANONICALIZE; case Intrinsic::floor: return TargetOpcode::G_FFLOOR; + case Intrinsic::fma: + return TargetOpcode::G_FMA; case Intrinsic::log: return TargetOpcode::G_FLOG; case Intrinsic::log2: return TargetOpcode::G_FLOG2; case Intrinsic::log10: return TargetOpcode::G_FLOG10; + case Intrinsic::pow: + return TargetOpcode::G_FPOW; case Intrinsic::round: return TargetOpcode::G_INTRINSIC_ROUND; case Intrinsic::sin: @@ -830,18 +833,22 @@ IRTranslator::getSimpleUnaryIntrinsicOpcode(Intrinsic::ID ID) { return Intrinsic::not_intrinsic; } -bool IRTranslator::translateSimpleUnaryIntrinsic( - const CallInst &CI, Intrinsic::ID ID, MachineIRBuilder &MIRBuilder) { +bool IRTranslator::translateSimpleIntrinsic(const CallInst &CI, + Intrinsic::ID ID, + MachineIRBuilder &MIRBuilder) { - unsigned Op = getSimpleUnaryIntrinsicOpcode(ID); + unsigned Op = getSimpleIntrinsicOpcode(ID); - // Is this a simple unary intrinsic? + // Is this a simple intrinsic? if (Op == Intrinsic::not_intrinsic) return false; // Yes. Let's translate it. - MIRBuilder.buildInstr(Op, {getOrCreateVReg(CI)}, - {getOrCreateVReg(*CI.getArgOperand(0))}, + SmallVector VRegs; + for (auto &Arg : CI.arg_operands()) + VRegs.push_back(getOrCreateVReg(*Arg)); + + MIRBuilder.buildInstr(Op, {getOrCreateVReg(CI)}, VRegs, MachineInstr::copyFlagsFromInstruction(CI)); return true; } @@ -849,9 +856,9 @@ bool IRTranslator::translateSimpleUnaryIntrinsic( bool IRTranslator::translateKnownIntrinsic(const CallInst &CI, Intrinsic::ID ID, MachineIRBuilder &MIRBuilder) { - // If this is a simple unary intrinsic (that is, we just need to add a def of - // a vreg, and a use of a vreg, then translate it. - if (translateSimpleUnaryIntrinsic(CI, ID, MIRBuilder)) + // If this is a simple intrinsic (that is, we just need to add a def of + // a vreg, and uses for each arg operand, then translate it. + if (translateSimpleIntrinsic(CI, ID, MIRBuilder)) return true; switch (ID) { @@ -974,21 +981,6 @@ bool IRTranslator::translateKnownIntrinsic(const CallInst &CI, Intrinsic::ID ID, return translateOverflowIntrinsic(CI, TargetOpcode::G_UMULO, MIRBuilder); case Intrinsic::smul_with_overflow: return translateOverflowIntrinsic(CI, TargetOpcode::G_SMULO, MIRBuilder); - case Intrinsic::pow: { - MIRBuilder.buildInstr(TargetOpcode::G_FPOW, {getOrCreateVReg(CI)}, - {getOrCreateVReg(*CI.getArgOperand(0)), - getOrCreateVReg(*CI.getArgOperand(1))}, - MachineInstr::copyFlagsFromInstruction(CI)); - return true; - } - case Intrinsic::fma: { - MIRBuilder.buildInstr(TargetOpcode::G_FMA, {getOrCreateVReg(CI)}, - {getOrCreateVReg(*CI.getArgOperand(0)), - getOrCreateVReg(*CI.getArgOperand(1)), - getOrCreateVReg(*CI.getArgOperand(2))}, - MachineInstr::copyFlagsFromInstruction(CI)); - return true; - } case Intrinsic::fmuladd: { const TargetMachine &TM = MF->getTarget(); const TargetLowering &TLI = *MF->getSubtarget().getTargetLowering(); -- 2.40.0