Skip to content

Commit cee82c0

Browse files
fangliu2020igcbot
authored andcommitted
Adding reg keys for scheduled BB range in local scheduler
1 parent 8479e81 commit cee82c0

File tree

2 files changed

+9
-0
lines changed

2 files changed

+9
-0
lines changed

visa/LocalScheduler/LocalScheduler_G4IR.cpp

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,8 +52,15 @@ void LocalScheduler::localScheduling()
5252
LatencyTable LT(fg.builder);
5353

5454
uint32_t totalCycles = 0;
55+
uint32_t scheduleStartBBId = m_options->getuInt32Option(vISA_LocalSchedulingStartBB);
56+
uint32_t shceduleEndBBId = m_options->getuInt32Option(vISA_LocalSchedulingEndBB);
5557
for (; ib != bend; ++ib)
5658
{
59+
if ((*ib)->getId() < scheduleStartBBId || (*ib)->getId() > shceduleEndBBId)
60+
{
61+
continue;
62+
}
63+
5764
unsigned instCountBefore = (uint32_t)(*ib)->size();
5865
#define SCH_THRESHOLD 2
5966
if (instCountBefore < SCH_THRESHOLD)

visa/include/VISAOptions.def

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -167,6 +167,8 @@ DEF_VISA_OPTION(vISA_HWThreadNumberPerEU, ET_INT32, "-HWThreadNumberPerEU", "USA
167167
DEF_VISA_OPTION(vISA_NoAtomicSend, ET_BOOL, "-noAtomicSend", UNUSED, false)
168168
DEF_VISA_OPTION(vISA_ReadSuppressionDepth, ET_INT32, "-readSuppressionDepth", UNUSED, 0)
169169
DEF_VISA_OPTION(vISA_ScheduleForReadSuppression, ET_BOOL, "-scheduleForReadSuppression", UNUSED, false)
170+
DEF_VISA_OPTION(vISA_LocalSchedulingStartBB, ET_INT32, "-scheduleStartBB", UNUSED, 0)
171+
DEF_VISA_OPTION(vISA_LocalSchedulingEndBB, ET_INT32, "-scheduleEndBB", UNUSED, UINT_MAX)
170172

171173
//=== SWSB options ===
172174
DEF_VISA_OPTION(vISA_USEL3HIT, ET_BOOL, "-SBIDL3Hit", UNUSED, false)

0 commit comments

Comments
 (0)