Skip to content

Commit 53b1da2

Browse files
jfuentessys_zuul
authored andcommitted
Obtain number of threads from kernel instead of builder in local scheduler.
Change-Id: Icaa00741155c6cb496b7b2671f042f9e9c6690dd
1 parent 98a7e5d commit 53b1da2

File tree

3 files changed

+11
-9
lines changed

3 files changed

+11
-9
lines changed

visa/FlowGraph.cpp

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3416,8 +3416,16 @@ void G4_Kernel::setKernelParameters()
34163416
default:
34173417
numAcc = 2;
34183418
}
3419+
}
34193420

3420-
3421+
// Set number of threads if it was not defined before
3422+
if (numThreads == 0)
3423+
{
3424+
switch (getGenxPlatform())
3425+
{
3426+
default:
3427+
numThreads = 7;
3428+
}
34213429
}
34223430
}
34233431

visa/HWCaps.inc

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -521,12 +521,6 @@
521521
return Type_HF;
522522
}
523523

524-
unsigned getHWThreadNumberPerEU() const
525-
{
526-
unsigned Val = getOptions()->getuInt32Option(vISA_HWThreadNumberPerEU);
527-
return (Val != 0) ? Val : 7;
528-
}
529-
530524
// each flag register is 16-bit
531525
uint32_t getNumFlagRegisters(void)
532526
{

visa/LocalScheduler/LocalScheduler_G4IR.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -236,7 +236,7 @@ G4_BB_Schedule::G4_BB_Schedule(G4_Kernel* k, Mem_Manager& m, G4_BB* block,
236236
// Update the listing of the basic block with the reordered code.
237237
INST_LIST_ITER inst_it = bb->begin();
238238
Node *prevNode = nullptr;
239-
unsigned HWThreadsPerEU = getBuilder()->getHWThreadNumberPerEU();
239+
unsigned HWThreadsPerEU = k->getNumThreads();
240240
size_t scheduleInstSize = 0;
241241
for (Node *currNode : scheduledNodes) {
242242
for (G4_INST *inst : *currNode->getInstructions()) {
@@ -1195,7 +1195,7 @@ DDD::DDD(Mem_Manager& m, G4_BB* bb, const LatencyTable& lt, G4_Kernel* k)
11951195
, kernel(k)
11961196
{
11971197
Node* lastBarrier = nullptr;
1198-
HWthreadsPerEU = getBuilder()->getHWThreadNumberPerEU();
1198+
HWthreadsPerEU = k->getNumThreads();
11991199
useMTLatencies = getBuilder()->useMultiThreadLatency();
12001200
totalGRFNum = kernel->getNumRegTotal();
12011201
bool BTIIsRestrict = getOptions()->getOption(vISA_ReorderDPSendToDifferentBti);

0 commit comments

Comments
 (0)