From af33d2eca05c453aa81c78bb8c863f1c40d69cf4 Mon Sep 17 00:00:00 2001 From: Tom Stellard Date: Tue, 22 Mar 2016 23:21:03 +0000 Subject: AMDGPU: Implement SIRegisterInfo::getRegPressureSetScore() This tells the scheduler it is better to increase SGPR register pressure than VGPR register pressure. This could probably be made smarter in the future. --- lib/Target/AMDGPU/SIRegisterInfo.cpp | 19 +++++++++++++++++++ lib/Target/AMDGPU/SIRegisterInfo.h | 3 +++ 2 files changed, 22 insertions(+) diff --git a/lib/Target/AMDGPU/SIRegisterInfo.cpp b/lib/Target/AMDGPU/SIRegisterInfo.cpp index 48de0cf1302..324e5c40db4 100644 --- a/lib/Target/AMDGPU/SIRegisterInfo.cpp +++ b/lib/Target/AMDGPU/SIRegisterInfo.cpp @@ -187,6 +187,25 @@ unsigned SIRegisterInfo::getRegPressureSetLimit(const MachineFunction &MF, return VGPRLimit; } +unsigned SIRegisterInfo::getRegPressureSetScore(const MachineFunction &MF, + unsigned PSetID) const { + + if (SGPRPressureSets.test(PSetID)) { + if (VGPRPressureSets.test(PSetID)) { + // If a pressure set contains vgprs and sgprs then we prefer incresing + // its pressure over regular register classes. This has the hightest + // score. + return 2; + } + // If this is an SGPR pressure set we prefer increasing its pressure + // over VGPR pressure sets. + return 1; + } + + // VGPRS. + return 0; +} + bool SIRegisterInfo::requiresRegisterScavenging(const MachineFunction &Fn) const { return Fn.getFrameInfo()->hasStackObjects(); } diff --git a/lib/Target/AMDGPU/SIRegisterInfo.h b/lib/Target/AMDGPU/SIRegisterInfo.h index 1888398917a..e8e2d1f8734 100644 --- a/lib/Target/AMDGPU/SIRegisterInfo.h +++ b/lib/Target/AMDGPU/SIRegisterInfo.h @@ -51,6 +51,9 @@ public: unsigned getRegPressureSetLimit(const MachineFunction &MF, unsigned Idx) const override; + unsigned getRegPressureSetScore(const MachineFunction &MF, + unsigned PSetID) const override; + bool requiresRegisterScavenging(const MachineFunction &Fn) const override; bool requiresFrameIndexScavenging(const MachineFunction &MF) const override; -- cgit v1.2.3