mirror of
https://github.com/RPCS3/llvm-mirror.git
synced 2024-11-01 16:33:37 +01:00
182 lines
7.8 KiB
C++
182 lines
7.8 KiB
C++
//===-- PTXFPRoundingModePass.cpp - Assign rounding modes pass ------------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file defines a machine function pass that sets appropriate FP rounding
|
|
// modes for all relevant instructions.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#define DEBUG_TYPE "ptx-fp-rounding-mode"
|
|
|
|
#include "PTX.h"
|
|
#include "PTXTargetMachine.h"
|
|
#include "llvm/ADT/DenseMap.h"
|
|
#include "llvm/CodeGen/MachineFunctionPass.h"
|
|
#include "llvm/CodeGen/MachineRegisterInfo.h"
|
|
#include "llvm/Support/Debug.h"
|
|
#include "llvm/Support/ErrorHandling.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
|
|
using namespace llvm;
|
|
|
|
// NOTE: PTXFPRoundingModePass should be executed just before emission.
|
|
|
|
namespace {
|
|
/// PTXFPRoundingModePass - Pass to assign appropriate FP rounding modes to
|
|
/// all FP instructions. Essentially, this pass just looks for all FP
|
|
/// instructions that have a rounding mode set to RndDefault, and sets an
|
|
/// appropriate rounding mode based on the target device.
|
|
///
|
|
class PTXFPRoundingModePass : public MachineFunctionPass {
|
|
private:
|
|
static char ID;
|
|
|
|
typedef std::pair<unsigned, unsigned> RndModeDesc;
|
|
|
|
PTXTargetMachine& TargetMachine;
|
|
DenseMap<unsigned, RndModeDesc> Instrs;
|
|
|
|
public:
|
|
PTXFPRoundingModePass(PTXTargetMachine &TM, CodeGenOpt::Level OptLevel)
|
|
: MachineFunctionPass(ID),
|
|
TargetMachine(TM) {
|
|
initializeMap();
|
|
}
|
|
|
|
virtual bool runOnMachineFunction(MachineFunction &MF);
|
|
|
|
virtual const char *getPassName() const {
|
|
return "PTX FP Rounding Mode Pass";
|
|
}
|
|
|
|
private:
|
|
|
|
void initializeMap();
|
|
void processInstruction(MachineInstr &MI);
|
|
}; // class PTXFPRoundingModePass
|
|
} // end anonymous namespace
|
|
|
|
using namespace llvm;
|
|
|
|
char PTXFPRoundingModePass::ID = 0;
|
|
|
|
bool PTXFPRoundingModePass::runOnMachineFunction(MachineFunction &MF) {
|
|
// Look at each basic block
|
|
for (MachineFunction::iterator bbi = MF.begin(), bbe = MF.end(); bbi != bbe;
|
|
++bbi) {
|
|
MachineBasicBlock &MBB = *bbi;
|
|
// Look at each instruction
|
|
for (MachineBasicBlock::iterator ii = MBB.begin(), ie = MBB.end();
|
|
ii != ie; ++ii) {
|
|
MachineInstr &MI = *ii;
|
|
processInstruction(MI);
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
void PTXFPRoundingModePass::initializeMap() {
|
|
using namespace PTXRoundingMode;
|
|
const PTXSubtarget& ST = TargetMachine.getSubtarget<PTXSubtarget>();
|
|
|
|
// Build a map of default rounding mode for all instructions that need a
|
|
// rounding mode.
|
|
Instrs[PTX::FADDrr32] = std::make_pair(1U, (unsigned)RndNearestEven);
|
|
Instrs[PTX::FADDri32] = std::make_pair(1U, (unsigned)RndNearestEven);
|
|
Instrs[PTX::FADDrr64] = std::make_pair(1U, (unsigned)RndNearestEven);
|
|
Instrs[PTX::FADDri64] = std::make_pair(1U, (unsigned)RndNearestEven);
|
|
Instrs[PTX::FSUBrr32] = std::make_pair(1U, (unsigned)RndNearestEven);
|
|
Instrs[PTX::FSUBri32] = std::make_pair(1U, (unsigned)RndNearestEven);
|
|
Instrs[PTX::FSUBrr64] = std::make_pair(1U, (unsigned)RndNearestEven);
|
|
Instrs[PTX::FSUBri64] = std::make_pair(1U, (unsigned)RndNearestEven);
|
|
Instrs[PTX::FMULrr32] = std::make_pair(1U, (unsigned)RndNearestEven);
|
|
Instrs[PTX::FMULri32] = std::make_pair(1U, (unsigned)RndNearestEven);
|
|
Instrs[PTX::FMULrr64] = std::make_pair(1U, (unsigned)RndNearestEven);
|
|
Instrs[PTX::FMULri64] = std::make_pair(1U, (unsigned)RndNearestEven);
|
|
|
|
Instrs[PTX::FNEGrr32] = std::make_pair(1U, (unsigned)RndNone);
|
|
Instrs[PTX::FNEGri32] = std::make_pair(1U, (unsigned)RndNone);
|
|
Instrs[PTX::FNEGrr64] = std::make_pair(1U, (unsigned)RndNone);
|
|
Instrs[PTX::FNEGri64] = std::make_pair(1U, (unsigned)RndNone);
|
|
|
|
unsigned FDivRndMode = ST.fdivNeedsRoundingMode() ? RndNearestEven : RndNone;
|
|
Instrs[PTX::FDIVrr32] = std::make_pair(1U, FDivRndMode);
|
|
Instrs[PTX::FDIVri32] = std::make_pair(1U, FDivRndMode);
|
|
Instrs[PTX::FDIVrr64] = std::make_pair(1U, FDivRndMode);
|
|
Instrs[PTX::FDIVri64] = std::make_pair(1U, FDivRndMode);
|
|
|
|
unsigned FMADRndMode = ST.fmadNeedsRoundingMode() ? RndNearestEven : RndNone;
|
|
Instrs[PTX::FMADrrr32] = std::make_pair(1U, FMADRndMode);
|
|
Instrs[PTX::FMADrri32] = std::make_pair(1U, FMADRndMode);
|
|
Instrs[PTX::FMADrii32] = std::make_pair(1U, FMADRndMode);
|
|
Instrs[PTX::FMADrrr64] = std::make_pair(1U, FMADRndMode);
|
|
Instrs[PTX::FMADrri64] = std::make_pair(1U, FMADRndMode);
|
|
Instrs[PTX::FMADrii64] = std::make_pair(1U, FMADRndMode);
|
|
|
|
Instrs[PTX::FSQRTrr32] = std::make_pair(1U, (unsigned)RndNearestEven);
|
|
Instrs[PTX::FSQRTri32] = std::make_pair(1U, (unsigned)RndNearestEven);
|
|
Instrs[PTX::FSQRTrr64] = std::make_pair(1U, (unsigned)RndNearestEven);
|
|
Instrs[PTX::FSQRTri64] = std::make_pair(1U, (unsigned)RndNearestEven);
|
|
|
|
Instrs[PTX::FSINrr32] = std::make_pair(1U, (unsigned)RndApprox);
|
|
Instrs[PTX::FSINri32] = std::make_pair(1U, (unsigned)RndApprox);
|
|
Instrs[PTX::FSINrr64] = std::make_pair(1U, (unsigned)RndApprox);
|
|
Instrs[PTX::FSINri64] = std::make_pair(1U, (unsigned)RndApprox);
|
|
Instrs[PTX::FCOSrr32] = std::make_pair(1U, (unsigned)RndApprox);
|
|
Instrs[PTX::FCOSri32] = std::make_pair(1U, (unsigned)RndApprox);
|
|
Instrs[PTX::FCOSrr64] = std::make_pair(1U, (unsigned)RndApprox);
|
|
Instrs[PTX::FCOSri64] = std::make_pair(1U, (unsigned)RndApprox);
|
|
|
|
Instrs[PTX::CVTu16f32] = std::make_pair(1U, (unsigned)RndTowardsZeroInt);
|
|
Instrs[PTX::CVTs16f32] = std::make_pair(1U, (unsigned)RndTowardsZeroInt);
|
|
Instrs[PTX::CVTu16f64] = std::make_pair(1U, (unsigned)RndTowardsZeroInt);
|
|
Instrs[PTX::CVTs16f64] = std::make_pair(1U, (unsigned)RndTowardsZeroInt);
|
|
Instrs[PTX::CVTu32f32] = std::make_pair(1U, (unsigned)RndTowardsZeroInt);
|
|
Instrs[PTX::CVTs32f32] = std::make_pair(1U, (unsigned)RndTowardsZeroInt);
|
|
Instrs[PTX::CVTu32f64] = std::make_pair(1U, (unsigned)RndTowardsZeroInt);
|
|
Instrs[PTX::CVTs32f64] = std::make_pair(1U, (unsigned)RndTowardsZeroInt);
|
|
Instrs[PTX::CVTu64f32] = std::make_pair(1U, (unsigned)RndTowardsZeroInt);
|
|
Instrs[PTX::CVTs64f32] = std::make_pair(1U, (unsigned)RndTowardsZeroInt);
|
|
Instrs[PTX::CVTu64f64] = std::make_pair(1U, (unsigned)RndTowardsZeroInt);
|
|
Instrs[PTX::CVTs64f64] = std::make_pair(1U, (unsigned)RndTowardsZeroInt);
|
|
|
|
Instrs[PTX::CVTf32u16] = std::make_pair(1U, (unsigned)RndNearestEven);
|
|
Instrs[PTX::CVTf32s16] = std::make_pair(1U, (unsigned)RndNearestEven);
|
|
Instrs[PTX::CVTf32u32] = std::make_pair(1U, (unsigned)RndNearestEven);
|
|
Instrs[PTX::CVTf32s32] = std::make_pair(1U, (unsigned)RndNearestEven);
|
|
Instrs[PTX::CVTf32u64] = std::make_pair(1U, (unsigned)RndNearestEven);
|
|
Instrs[PTX::CVTf32s64] = std::make_pair(1U, (unsigned)RndNearestEven);
|
|
Instrs[PTX::CVTf32f64] = std::make_pair(1U, (unsigned)RndNearestEven);
|
|
Instrs[PTX::CVTf64u16] = std::make_pair(1U, (unsigned)RndNearestEven);
|
|
Instrs[PTX::CVTf64s16] = std::make_pair(1U, (unsigned)RndNearestEven);
|
|
Instrs[PTX::CVTf64u32] = std::make_pair(1U, (unsigned)RndNearestEven);
|
|
Instrs[PTX::CVTf64s32] = std::make_pair(1U, (unsigned)RndNearestEven);
|
|
Instrs[PTX::CVTf64u64] = std::make_pair(1U, (unsigned)RndNearestEven);
|
|
Instrs[PTX::CVTf64s64] = std::make_pair(1U, (unsigned)RndNearestEven);
|
|
}
|
|
|
|
void PTXFPRoundingModePass::processInstruction(MachineInstr &MI) {
|
|
// Is this an instruction that needs a rounding mode?
|
|
if (Instrs.count(MI.getOpcode())) {
|
|
const RndModeDesc &Desc = Instrs[MI.getOpcode()];
|
|
// Get the rounding mode operand
|
|
MachineOperand &Op = MI.getOperand(Desc.first);
|
|
// Update the rounding mode if needed
|
|
if (Op.getImm() == PTXRoundingMode::RndDefault) {
|
|
Op.setImm(Desc.second);
|
|
}
|
|
}
|
|
}
|
|
|
|
FunctionPass *llvm::createPTXFPRoundingModePass(PTXTargetMachine &TM,
|
|
CodeGenOpt::Level OptLevel) {
|
|
return new PTXFPRoundingModePass(TM, OptLevel);
|
|
}
|
|
|