blob: 496e32cea53562283f1a698204a6810f7e90430e (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
/* IBM_PROLOG_BEGIN_TAG */
/* This is an automatically generated prolog. */
/* */
/* $Source: src/kernel/scheduler.C $ */
/* */
/* OpenPOWER HostBoot Project */
/* */
/* COPYRIGHT International Business Machines Corp. 2010,2014 */
/* */
/* Licensed under the Apache License, Version 2.0 (the "License"); */
/* you may not use this file except in compliance with the License. */
/* You may obtain a copy of the License at */
/* */
/* http://www.apache.org/licenses/LICENSE-2.0 */
/* */
/* Unless required by applicable law or agreed to in writing, software */
/* distributed under the License is distributed on an "AS IS" BASIS, */
/* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or */
/* implied. See the License for the specific language governing */
/* permissions and limitations under the License. */
/* */
/* IBM_PROLOG_END_TAG */
#include <kernel/task.H>
#include <kernel/scheduler.H>
#include <kernel/taskmgr.H>
#include <kernel/cpu.H>
#include <kernel/cpumgr.H>
#include <kernel/console.H>
#include <kernel/timemgr.H>
#include <util/lockfree/atomic_construct.H>
void Scheduler::addTask(task_t* t)
{
t->state = TASK_STATE_READY;
if (t->cpu->idle_task != t)
{
// If task is pinned to this CPU, add to the per-CPU queue.
if (0 != t->affinity_pinned)
{
// Allocate a per-CPU queue if this is the first pinning to CPU.
Util::Lockfree::atomic_construct
(reinterpret_cast<Runqueue_t**>(&t->cpu->scheduler_extra));
// Insert into queue.
static_cast<Runqueue_t*>(t->cpu->scheduler_extra)->insert(t);
}
// Not pinned, add to global run-queue.
else
{
iv_taskList.insert(t);
}
}
}
void Scheduler::addTaskMasterCPU(task_t* t)
{
t->state = TASK_STATE_READY;
if (t->cpu->idle_task != t)
{
cpu_t* master = CpuManager::getMasterCPU();
// Allocate a per-CPU queue if this is the first pinning to CPU.
Util::Lockfree::atomic_construct(
reinterpret_cast<Runqueue_t**>(&master->scheduler_extra));
// Insert into queue.
static_cast<Runqueue_t*>(master->scheduler_extra)->insert(t);
}
}
void Scheduler::returnRunnable()
{
this->addTask(TaskManager::getCurrentTask());
}
void Scheduler::setNextRunnable()
{
task_t* t = NULL;
cpu_t* cpu = CpuManager::getCurrentCPU();
// Check for ready task in local run-queue, if it exists.
if (NULL != cpu->scheduler_extra)
{
t = static_cast<Runqueue_t*>(cpu->scheduler_extra)->remove();
}
// Check for ready task in global run-queue.
if (NULL == t)
{
t = iv_taskList.remove();
}
// Choose idle task if no other ready task is available.
if (NULL == t)
{
t = cpu->idle_task;
setDEC(TimeManager::getIdleTimeSliceCount());
}
else // Set normal timeslice to decrementer.
{
setDEC(TimeManager::getTimeSliceCount());
}
TaskManager::setCurrentTask(t);
}
|