1 /* 2 * Copyright (c) 1999 Peter Wemm <peter@FreeBSD.org>. All rights reserved. 3 * Copyright (c) 2012-2020 The DragonFly Project. All rights reserved. 4 * 5 * This code is derived from software contributed to The DragonFly Project 6 * by Matthew Dillon <dillon@backplane.com>, 7 * by Mihai Carabas <mihai.carabas@gmail.com> 8 * and many others. 9 * 10 * Redistribution and use in source and binary forms, with or without 11 * modification, are permitted provided that the following conditions 12 * are met: 13 * 14 * 1. Redistributions of source code must retain the above copyright 15 * notice, this list of conditions and the following disclaimer. 16 * 2. Redistributions in binary form must reproduce the above copyright 17 * notice, this list of conditions and the following disclaimer in 18 * the documentation and/or other materials provided with the 19 * distribution. 20 * 3. Neither the name of The DragonFly Project nor the names of its 21 * contributors may be used to endorse or promote products derived 22 * from this software without specific, prior written permission. 23 * 24 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 25 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 26 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS 27 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE 28 * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, 29 * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING, 30 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 31 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED 32 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, 33 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT 34 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 35 * SUCH DAMAGE. 36 */ 37 #include <sys/param.h> 38 #include <sys/systm.h> 39 #include <sys/kernel.h> 40 #include <sys/lock.h> 41 #include <sys/queue.h> 42 #include <sys/proc.h> 43 #include <sys/rtprio.h> 44 #include <sys/uio.h> 45 #include <sys/sysctl.h> 46 #include <sys/resourcevar.h> 47 #include <sys/spinlock.h> 48 #include <sys/cpu_topology.h> 49 #include <sys/thread2.h> 50 #include <sys/spinlock2.h> 51 52 #include <sys/ktr.h> 53 54 #include <machine/cpu.h> 55 #include <machine/smp.h> 56 57 #include <sys/usched_dfly.h> 58 59 /*static void dfly_acquire_curproc(struct lwp *lp); see sys/usched.h */ 60 static void dfly_release_curproc(struct lwp *lp); 61 static void dfly_select_curproc(globaldata_t gd); 62 static void dfly_setrunqueue(struct lwp *lp); 63 static void dfly_setrunqueue_dd(dfly_pcpu_t rdd, struct lwp *lp); 64 static void dfly_schedulerclock(struct lwp *lp, sysclock_t period, 65 sysclock_t cpstamp); 66 static void dfly_recalculate_estcpu(struct lwp *lp); 67 static void dfly_resetpriority(struct lwp *lp); 68 static void dfly_forking(struct lwp *plp, struct lwp *lp); 69 static void dfly_exiting(struct lwp *lp, struct proc *); 70 static void dfly_uload_update(struct lwp *lp); 71 static void dfly_yield(struct lwp *lp); 72 static void dfly_changeqcpu_locked(struct lwp *lp, 73 dfly_pcpu_t dd, dfly_pcpu_t rdd); 74 static dfly_pcpu_t dfly_choose_best_queue(struct lwp *lp); 75 static dfly_pcpu_t dfly_choose_worst_queue(dfly_pcpu_t dd, int forceit); 76 static dfly_pcpu_t dfly_choose_queue_simple(dfly_pcpu_t dd, struct lwp *lp); 77 static void dfly_need_user_resched_remote(void *dummy); 78 static struct lwp *dfly_chooseproc_locked(dfly_pcpu_t rdd, dfly_pcpu_t dd, 79 struct lwp *chklp, int worst); 80 static void dfly_remrunqueue_locked(dfly_pcpu_t dd, struct lwp *lp); 81 static void dfly_setrunqueue_locked(dfly_pcpu_t dd, struct lwp *lp); 82 static void dfly_changedcpu(struct lwp *lp); 83 84 struct usched usched_dfly = { 85 { NULL }, 86 "dfly", "Original DragonFly Scheduler", 87 NULL, /* default registration */ 88 NULL, /* default deregistration */ 89 dfly_acquire_curproc, 90 dfly_release_curproc, 91 dfly_setrunqueue, 92 dfly_schedulerclock, 93 dfly_recalculate_estcpu, 94 dfly_resetpriority, 95 dfly_forking, 96 dfly_exiting, 97 dfly_uload_update, 98 NULL, /* setcpumask not supported */ 99 dfly_yield, 100 dfly_changedcpu 101 }; 102 103 /* 104 * We have NQS (32) run queues per scheduling class. For the normal 105 * class, there are 128 priorities scaled onto these 32 queues. New 106 * processes are added to the last entry in each queue, and processes 107 * are selected for running by taking them from the head and maintaining 108 * a simple FIFO arrangement. Realtime and Idle priority processes have 109 * and explicit 0-31 priority which maps directly onto their class queue 110 * index. When a queue has something in it, the corresponding bit is 111 * set in the queuebits variable, allowing a single read to determine 112 * the state of all 32 queues and then a ffs() to find the first busy 113 * queue. 114 * 115 * curprocmask is used to publish cpus with assigned curprocs to the rest 116 * of the cpus. In certain situations curprocmask may leave a bit set 117 * (e.g. a yield or a token-based yield) even though dd->uschedcp is 118 * NULL'd out temporarily). 119 */ 120 /* currently running a user process */ 121 static cpumask_t dfly_curprocmask = CPUMASK_INITIALIZER_ALLONES; 122 static cpumask_t dfly_rdyprocmask; /* ready to accept a user process */ 123 static struct usched_dfly_pcpu dfly_pcpu[MAXCPU]; 124 static struct sysctl_ctx_list usched_dfly_sysctl_ctx; 125 static struct sysctl_oid *usched_dfly_sysctl_tree; 126 static struct lock usched_dfly_config_lk = LOCK_INITIALIZER("usdfs", 0, 0); 127 128 /* Debug info exposed through debug.* sysctl */ 129 130 static int usched_dfly_debug = -1; 131 SYSCTL_INT(_debug, OID_AUTO, dfly_scdebug, CTLFLAG_RW, 132 &usched_dfly_debug, 0, 133 "Print debug information for this pid"); 134 135 static int usched_dfly_pid_debug = -1; 136 SYSCTL_INT(_debug, OID_AUTO, dfly_pid_debug, CTLFLAG_RW, 137 &usched_dfly_pid_debug, 0, 138 "Print KTR debug information for this pid"); 139 140 static int usched_dfly_chooser = 0; 141 SYSCTL_INT(_debug, OID_AUTO, dfly_chooser, CTLFLAG_RW, 142 &usched_dfly_chooser, 0, 143 "Print KTR debug information for this pid"); 144 145 /* 146 * WARNING! 147 * 148 * The fork bias can have a large effect on the system in the face of a 149 * make -j N or other high-forking applications. 150 * 151 * Larger values are much less invasive vs other things that 152 * might be running in the system, but can cause exec chains 153 * such as those typically generated by make to have higher 154 * latencies in the face of modest load. 155 * 156 * Lower values are more invasive but have reduced latencies 157 * for such exec chains. 158 * 159 * make -j 10 buildkernel example, build times: 160 * 161 * +0 3:04 162 * +1 3:14 -5.2% <-- default 163 * +2 3:22 -8.9% 164 * 165 * This issue occurs due to the way the scheduler affinity heuristics work. 166 * There is no way to really 'fix' the affinity heuristics because when it 167 * comes right down to it trying to instantly schedule a process on an 168 * available cpu (even if it will become unavailable a microsecond later) 169 * tends to cause processes to shift around between cpus and sockets too much 170 * and breaks the affinity. 171 * 172 * NOTE: Heavily concurrent builds typically have enough things on the pan 173 * that they remain time-efficient even with a higher bias. 174 */ 175 static int usched_dfly_forkbias = 1; 176 SYSCTL_INT(_debug, OID_AUTO, dfly_forkbias, CTLFLAG_RW, 177 &usched_dfly_forkbias, 0, 178 "Fork bias for estcpu in whole queues"); 179 180 /* 181 * Tunning usched_dfly - configurable through kern.usched_dfly. 182 * 183 * weight1 - Tries to keep threads on their current cpu. If you 184 * make this value too large the scheduler will not be 185 * able to load-balance large loads. 186 * 187 * Generally set to a fairly low value, but high enough 188 * such that estcpu jitter doesn't move threads around. 189 * 190 * weight2 - If non-zero, detects thread pairs undergoing synchronous 191 * communications and tries to move them closer together. 192 * The weight advantages the same package and socket and 193 * disadvantages the same core and same cpu. 194 * 195 * WARNING! Weight2 is a ridiculously sensitive parameter, 196 * particularly against weight4. change the default at your 197 * peril. 198 * 199 * weight3 - Weighting based on the number of recently runnable threads 200 * on the userland scheduling queue (ignoring their loads). 201 * 202 * A nominal value here prevents high-priority (low-load) 203 * threads from accumulating on one cpu core when other 204 * cores are available. 205 * 206 * This value should be left fairly small because low-load 207 * high priority threads can still be mostly idle and too 208 * high a value will kick cpu-bound processes off the cpu 209 * unnecessarily. 210 * 211 * weight4 - Weighting based on availability of other logical cpus running 212 * less important threads (by upri) than the thread we are trying 213 * to schedule. 214 * 215 * This allows a thread to migrate to another nearby cpu if it 216 * is unable to run on the current cpu based on the other cpu 217 * being idle or running a less important (higher lwp_priority) 218 * thread. This value should be large enough to override weight1, 219 * but not so large as to override weight2. 220 * 221 * This parameter generally ensures fairness at the cost of some 222 * performance (if set to too high). It should generally be just 223 * a tad lower than weight2. 224 * 225 * weight5 - Weighting based on the relative amount of ram connected 226 * to the node a cpu resides on. 227 * 228 * This value should remain fairly low to allow assymetric 229 * NUMA nodes to get threads scheduled to them. Setting a very 230 * high level will prevent scheduling on assymetric NUMA nodes 231 * with low amounts of directly-attached memory. 232 * 233 * Note that when testing e.g. N threads on a machine with N 234 * cpu cores with assymtric NUMA nodes, a non-zero value will 235 * cause some cpu threads on the low-priority NUMA nodes to remain 236 * idle even when a few process threads are doubled-up on other 237 * cpus. But this is typically more ideal because it deschedules 238 * low-priority NUMA nodes at lighter nodes. 239 * 240 * Values between 50 and 200 are recommended. Default is 50. 241 * 242 * weight6 - rdd transfer weight hysteresis for regular pair rebalancing 243 * (feature 0x04). 244 * 245 * Defaults to 0, can be increased to improve stabillity at the 246 * cost of more mis-schedules. 247 * 248 * weight7 - rdd transfer weight hysteresis for idle cpu 'pull' (feature 0x01). 249 * 250 * Defaults to -100 to strongly promote a transfer. 251 * 252 * ipc_smt - If enabled, advantage IPC pairing to sibling cpu threads. 253 * If -1, automatic when load >= 1/2 ncpus (default). 254 * 255 * ipc_same- If enabled, advantage IPC pairing to the same logical cpu. 256 * If -1, automatic when load >= ncpus (default). 257 * 258 * features - These flags can be set or cleared to enable or disable various 259 * features. 260 * 261 * 0x01 Enable idle-cpu pulling (default) 262 * 0x02 Enable proactive pushing (default) 263 * 0x04 Enable rebalancing rover (default) 264 * 0x08 Enable more proactive pushing (default) 265 * 0x10 (unassigned) 266 * 0x20 choose best cpu for forked process (default) 267 * 0x40 choose current cpu for forked process 268 * 0x80 choose random cpu for forked process 269 * 270 * NOTE - The idea behind forking mechanic 0x20 is that most 271 * fork()ing is either followed by an exec in the child, 272 * or the parent wait*()s. If the child is short-lived, 273 * there is effectively an IPC dependency (td_wakefromcpu 274 * is also set in kern_fork.c) and we want to implement 275 * the weight2 behavior to reduce IPIs and to reduce CPU 276 * cache ping-ponging. 277 */ 278 __read_mostly static int usched_dfly_smt = 0; 279 __read_mostly static int usched_dfly_cache_coherent = 0; 280 __read_mostly static int usched_dfly_weight1 = 30; /* keep thread on cpu */ 281 __read_mostly static int usched_dfly_weight2 = 180; /* IPC locality */ 282 __read_mostly static int usched_dfly_weight3 = 10; /* threads on queue */ 283 __read_mostly static int usched_dfly_weight4 = 120; /* availability of cores */ 284 __read_mostly static int usched_dfly_weight5 = 50; /* node attached memory */ 285 __read_mostly static int usched_dfly_weight6 = 0; /* 0x04 transfer weight */ 286 __read_mostly static int usched_dfly_weight7 = -100;/* 0x01 transfer weight */ 287 __read_mostly static int usched_dfly_features = 0x2f; /* allow pulls */ 288 __read_mostly static int usched_dfly_fast_resched = PPQ / 2; /* delta pri */ 289 __read_mostly static int usched_dfly_swmask = ~PPQMASK; /* allow pulls */ 290 __read_mostly static int usched_dfly_rrinterval = (ESTCPUFREQ + 9) / 10; 291 __read_mostly static int usched_dfly_decay = 8; 292 __read_mostly static int usched_dfly_ipc_smt = -1; /* IPC auto smt pair */ 293 __read_mostly static int usched_dfly_ipc_same = -1; /* IPC auto same log cpu */ 294 __read_mostly static int usched_dfly_poll_ticks = 1; /* helper polling ticks */ 295 __read_mostly static long usched_dfly_node_mem; 296 297 /* KTR debug printings */ 298 299 KTR_INFO_MASTER(usched); 300 301 #if !defined(KTR_USCHED_DFLY) 302 #define KTR_USCHED_DFLY KTR_ALL 303 #endif 304 305 KTR_INFO(KTR_USCHED_DFLY, usched, chooseproc, 0, 306 "USCHED_DFLY(chooseproc: pid %d, old_cpuid %d, curr_cpuid %d)", 307 pid_t pid, int old_cpuid, int curr); 308 309 /* 310 * This function is called when the kernel intends to return to userland. 311 * It is responsible for making the thread the current designated userland 312 * thread for this cpu, blocking if necessary. 313 * 314 * The kernel will not depress our LWKT priority until after we return, 315 * in case we have to shove over to another cpu. 316 * 317 * We must determine our thread's disposition before we switch away. This 318 * is very sensitive code. 319 * 320 * WARNING! THIS FUNCTION IS ALLOWED TO CAUSE THE CURRENT THREAD TO MIGRATE 321 * TO ANOTHER CPU! Because most of the kernel assumes that no migration will 322 * occur, this function is called only under very controlled circumstances. 323 */ 324 void 325 dfly_acquire_curproc(struct lwp *lp) 326 { 327 globaldata_t gd; 328 dfly_pcpu_t dd; 329 dfly_pcpu_t rdd; 330 thread_t td; 331 int force_resched; 332 333 td = lp->lwp_thread; 334 gd = mycpu; 335 dd = &dfly_pcpu[gd->gd_cpuid]; 336 337 /* 338 * Quickly return if possible. 339 */ 340 if (__predict_true((td->td_flags & TDF_TSLEEPQ) == 0 && 341 !any_action_wanted_gd(gd) && 342 dd->uschedcp == lp)) { 343 return; 344 } 345 346 /* 347 * Make sure we aren't sitting on a tsleep queue. 348 */ 349 crit_enter_quick(td); 350 if (td->td_flags & TDF_TSLEEPQ) 351 tsleep_remove(td); 352 dfly_recalculate_estcpu(lp); 353 354 /* 355 * Process any pending interrupts/ipi's, then handle reschedule 356 * requests. dfly_release_curproc() will try to assign a new 357 * uschedcp that isn't us and otherwise NULL it out. 358 */ 359 force_resched = 0; 360 if (user_resched_wanted()) { 361 if (dd->uschedcp == lp) 362 force_resched = 1; 363 clear_user_resched(); 364 dfly_release_curproc(lp); 365 } 366 367 /* 368 * Loop until we are the current user thread. 369 * 370 * NOTE: dd spinlock not held at top of loop. 371 */ 372 if (dd->uschedcp == lp) 373 lwkt_yield_quick(); 374 375 while (dd->uschedcp != lp) { 376 /* 377 * Do not do a lwkt_yield_quick() here as it will prevent 378 * the lwp from being placed on the dfly_bsd runqueue for 379 * one cycle (possibly an entire round-robin), preventing 380 * it from being scheduled to another cpu. 381 */ 382 /* lwkt_yield_quick(); */ 383 384 if (usched_dfly_debug == lp->lwp_proc->p_pid) 385 kprintf(" pid %d acquire curcpu %d (force %d) ", 386 lp->lwp_proc->p_pid, gd->gd_cpuid, 387 force_resched); 388 389 390 spin_lock(&dd->spin); 391 392 /* This lwp is an outcast; force reschedule. */ 393 if (__predict_false( 394 CPUMASK_TESTBIT(lp->lwp_cpumask, gd->gd_cpuid) == 0) && 395 (rdd = dfly_choose_best_queue(lp)) != dd) { 396 dfly_changeqcpu_locked(lp, dd, rdd); 397 spin_unlock(&dd->spin); 398 lwkt_deschedule(lp->lwp_thread); 399 dfly_setrunqueue_dd(rdd, lp); 400 lwkt_switch(); 401 gd = mycpu; 402 dd = &dfly_pcpu[gd->gd_cpuid]; 403 if (usched_dfly_debug == lp->lwp_proc->p_pid) 404 kprintf("SEL-A cpu %d\n", gd->gd_cpuid); 405 continue; 406 } 407 408 /* 409 * We are not or are no longer the current lwp and a forced 410 * reschedule was requested. Figure out the best cpu to 411 * run on (our current cpu will be given significant weight). 412 * 413 * Doing this on many cpus simultaneously leads to 414 * instability so pace the operation. 415 * 416 * (if a reschedule was not requested we want to move this 417 * step after the uschedcp tests). 418 */ 419 if (force_resched && 420 (usched_dfly_features & 0x08) && 421 (u_int)sched_ticks / 8 % ncpus == gd->gd_cpuid) { 422 if ((rdd = dfly_choose_best_queue(lp)) != dd) { 423 dfly_changeqcpu_locked(lp, dd, rdd); 424 spin_unlock(&dd->spin); 425 lwkt_deschedule(lp->lwp_thread); 426 dfly_setrunqueue_dd(rdd, lp); 427 lwkt_switch(); 428 gd = mycpu; 429 dd = &dfly_pcpu[gd->gd_cpuid]; 430 if (usched_dfly_debug == lp->lwp_proc->p_pid) 431 kprintf("SEL-B cpu %d\n", gd->gd_cpuid); 432 continue; 433 } 434 if (usched_dfly_debug == lp->lwp_proc->p_pid) 435 kprintf("(SEL-B same cpu) "); 436 } 437 438 /* 439 * Either no reschedule was requested or the best queue was 440 * dd, and no current process has been selected. We can 441 * trivially become the current lwp on the current cpu. 442 */ 443 if (dd->uschedcp == NULL) { 444 atomic_clear_int(&lp->lwp_thread->td_mpflags, 445 TDF_MP_DIDYIELD); 446 if ((dd->flags & DFLY_PCPU_CURMASK) == 0) { 447 ATOMIC_CPUMASK_ORBIT(dfly_curprocmask, 448 gd->gd_cpuid); 449 dd->flags |= DFLY_PCPU_CURMASK; 450 } 451 dd->uschedcp = lp; 452 dd->upri = lp->lwp_priority; 453 KKASSERT(lp->lwp_qcpu == dd->cpuid); 454 spin_unlock(&dd->spin); 455 if (usched_dfly_debug == lp->lwp_proc->p_pid) 456 kprintf("SEL-C cpu %d (same cpu)\n", 457 gd->gd_cpuid); 458 break; 459 } 460 461 /* 462 * Can we steal the current designated user thread? 463 * 464 * If we do the other thread will stall when it tries to 465 * return to userland, possibly rescheduling elsewhere. 466 * Set need_user_resched() to get the thread to cycle soonest. 467 * 468 * It is important to do a masked test to avoid the edge 469 * case where two near-equal-priority threads are constantly 470 * interrupting each other. 471 * 472 * In the exact match case another thread has already gained 473 * uschedcp and lowered its priority, if we steal it the 474 * other thread will stay stuck on the LWKT runq and not 475 * push to another cpu. So don't steal on equal-priority even 476 * though it might appear to be more beneficial due to not 477 * having to switch back to the other thread's context. 478 * 479 * usched_dfly_fast_resched requires that two threads be 480 * significantly far apart in priority in order to interrupt. 481 * 482 * If better but not sufficiently far apart, the current 483 * uschedcp will be interrupted at the next scheduler clock. 484 */ 485 if (dd->uschedcp && 486 (dd->upri & ~PPQMASK) > 487 (lp->lwp_priority & ~PPQMASK) + usched_dfly_fast_resched) { 488 dd->uschedcp = lp; 489 dd->upri = lp->lwp_priority; 490 KKASSERT(lp->lwp_qcpu == dd->cpuid); 491 need_user_resched(); 492 spin_unlock(&dd->spin); 493 if (usched_dfly_debug == lp->lwp_proc->p_pid) 494 kprintf("SEL-D cpu %d (same cpu)\n", 495 gd->gd_cpuid); 496 break; 497 } 498 499 /* 500 * Requeue us at lwp_priority, which recalculate_estcpu() 501 * set for us. Reset the rrcount to force placement 502 * at the end of the queue. 503 * 504 * We used to move ourselves to the worst queue, but 505 * this creates a fairly serious priority inversion 506 * problem. 507 */ 508 if (lp->lwp_thread->td_mpflags & TDF_MP_DIDYIELD) { 509 spin_unlock(&dd->spin); 510 lp->lwp_rrcount = usched_dfly_rrinterval; 511 lp->lwp_rqindex = (lp->lwp_priority & PRIMASK) / PPQ; 512 513 lwkt_deschedule(lp->lwp_thread); 514 dfly_setrunqueue_dd(dd, lp); 515 atomic_clear_int(&lp->lwp_thread->td_mpflags, 516 TDF_MP_DIDYIELD); 517 lwkt_switch(); 518 gd = mycpu; 519 dd = &dfly_pcpu[gd->gd_cpuid]; 520 if (usched_dfly_debug == lp->lwp_proc->p_pid) 521 kprintf("SEL-E cpu %d (requeue)\n", 522 gd->gd_cpuid); 523 continue; 524 } 525 526 /* 527 * We are not the current lwp, figure out the best cpu 528 * to run on (our current cpu will be given significant 529 * weight). Loop on cpu change. 530 */ 531 if ((usched_dfly_features & 0x02) && 532 force_resched == 0 && 533 (rdd = dfly_choose_best_queue(lp)) != dd) { 534 dfly_changeqcpu_locked(lp, dd, rdd); 535 spin_unlock(&dd->spin); 536 lwkt_deschedule(lp->lwp_thread); 537 dfly_setrunqueue_dd(rdd, lp); 538 lwkt_switch(); 539 gd = mycpu; 540 dd = &dfly_pcpu[gd->gd_cpuid]; 541 if (usched_dfly_debug == lp->lwp_proc->p_pid) 542 kprintf("SEL-F cpu %d (requeue new cpu)\n", 543 gd->gd_cpuid); 544 continue; 545 } 546 547 /* 548 * We cannot become the current lwp, place the lp on the 549 * run-queue of this or another cpu and deschedule ourselves. 550 * 551 * When we are reactivated we will have another chance. 552 * 553 * Reload after a switch or setrunqueue/switch possibly 554 * moved us to another cpu. 555 */ 556 spin_unlock(&dd->spin); 557 lwkt_deschedule(lp->lwp_thread); 558 dfly_setrunqueue_dd(dd, lp); 559 lwkt_switch(); 560 gd = mycpu; 561 dd = &dfly_pcpu[gd->gd_cpuid]; 562 if (usched_dfly_debug == lp->lwp_proc->p_pid) 563 kprintf("SEL-G cpu %d (fallback setrunq)\n", 564 gd->gd_cpuid); 565 } 566 if (usched_dfly_debug == lp->lwp_proc->p_pid) 567 kprintf(" pid %d acquire DONE cpu %d\n", 568 lp->lwp_proc->p_pid, gd->gd_cpuid); 569 570 /* 571 * Make sure upri is synchronized, then yield to LWKT threads as 572 * needed before returning. This could result in another reschedule. 573 * XXX 574 */ 575 crit_exit_quick(td); 576 577 KKASSERT((lp->lwp_mpflags & LWP_MP_ONRUNQ) == 0); 578 } 579 580 /* 581 * DFLY_RELEASE_CURPROC 582 * 583 * This routine detaches the current thread from the userland scheduler, 584 * usually because the thread needs to run or block in the kernel (at 585 * kernel priority) for a while. 586 * 587 * This routine is also responsible for selecting a new thread to 588 * make the current thread. 589 * 590 * NOTE: This implementation differs from the dummy example in that 591 * dfly_select_curproc() is able to select the current process, whereas 592 * dummy_select_curproc() is not able to select the current process. 593 * This means we have to NULL out uschedcp. 594 * 595 * Additionally, note that we may already be on a run queue if releasing 596 * via the lwkt_switch() in dfly_setrunqueue(). 597 */ 598 static void 599 dfly_release_curproc(struct lwp *lp) 600 { 601 globaldata_t gd = mycpu; 602 dfly_pcpu_t dd = &dfly_pcpu[gd->gd_cpuid]; 603 604 /* 605 * Make sure td_wakefromcpu is defaulted. This will be overwritten 606 * by wakeup(). 607 */ 608 if (dd->uschedcp == lp) { 609 KKASSERT((lp->lwp_mpflags & LWP_MP_ONRUNQ) == 0); 610 spin_lock(&dd->spin); 611 if (dd->uschedcp == lp) { 612 dd->uschedcp = NULL; /* don't let lp be selected */ 613 dd->upri = PRIBASE_NULL; 614 615 /* 616 * We're just going to set it again, avoid the global 617 * cache line ping-pong. 618 */ 619 if ((lp->lwp_thread->td_mpflags & TDF_MP_DIDYIELD) == 0) { 620 if (dd->flags & DFLY_PCPU_CURMASK) { 621 ATOMIC_CPUMASK_NANDBIT(dfly_curprocmask, 622 gd->gd_cpuid); 623 dd->flags &= ~DFLY_PCPU_CURMASK; 624 } 625 } 626 spin_unlock(&dd->spin); 627 dfly_select_curproc(gd); 628 } else { 629 spin_unlock(&dd->spin); 630 } 631 } 632 } 633 634 /* 635 * DFLY_SELECT_CURPROC 636 * 637 * Select a new current process for this cpu and clear any pending user 638 * reschedule request. The cpu currently has no current process. 639 * 640 * This routine is also responsible for equal-priority round-robining, 641 * typically triggered from dfly_schedulerclock(). In our dummy example 642 * all the 'user' threads are LWKT scheduled all at once and we just 643 * call lwkt_switch(). 644 * 645 * The calling process is not on the queue and cannot be selected. 646 */ 647 static 648 void 649 dfly_select_curproc(globaldata_t gd) 650 { 651 dfly_pcpu_t dd = &dfly_pcpu[gd->gd_cpuid]; 652 struct lwp *nlp; 653 int cpuid = gd->gd_cpuid; 654 655 crit_enter_gd(gd); 656 657 spin_lock(&dd->spin); 658 nlp = dfly_chooseproc_locked(dd, dd, dd->uschedcp, 0); 659 660 if (nlp) { 661 if ((dd->flags & DFLY_PCPU_CURMASK) == 0) { 662 ATOMIC_CPUMASK_ORBIT(dfly_curprocmask, cpuid); 663 dd->flags |= DFLY_PCPU_CURMASK; 664 } 665 dd->upri = nlp->lwp_priority; 666 dd->uschedcp = nlp; 667 #if 0 668 dd->rrcount = 0; /* reset round robin */ 669 #endif 670 spin_unlock(&dd->spin); 671 lwkt_acquire(nlp->lwp_thread); 672 lwkt_schedule(nlp->lwp_thread); 673 } else { 674 spin_unlock(&dd->spin); 675 } 676 crit_exit_gd(gd); 677 } 678 679 /* 680 * Place the specified lwp on the user scheduler's run queue. This routine 681 * must be called with the thread descheduled. The lwp must be runnable. 682 * It must not be possible for anyone else to explicitly schedule this thread. 683 * 684 * The thread may be the current thread as a special case. 685 */ 686 static void 687 dfly_setrunqueue(struct lwp *lp) 688 { 689 dfly_pcpu_t dd; 690 dfly_pcpu_t rdd; 691 692 /* 693 * First validate the process LWKT state. 694 */ 695 KASSERT(lp->lwp_stat == LSRUN, ("setrunqueue: lwp not LSRUN")); 696 KASSERT((lp->lwp_mpflags & LWP_MP_ONRUNQ) == 0, 697 ("lwp %d/%d already on runq! flag %08x/%08x", lp->lwp_proc->p_pid, 698 lp->lwp_tid, lp->lwp_proc->p_flags, lp->lwp_flags)); 699 KKASSERT((lp->lwp_thread->td_flags & TDF_RUNQ) == 0); 700 701 /* 702 * NOTE: dd/rdd do not necessarily represent the current cpu. 703 * Instead they may represent the cpu the thread was last 704 * scheduled on or inherited by its parent. 705 */ 706 dd = &dfly_pcpu[lp->lwp_qcpu]; 707 rdd = dd; 708 709 /* 710 * This process is not supposed to be scheduled anywhere or assigned 711 * as the current process anywhere. Assert the condition. 712 */ 713 KKASSERT(rdd->uschedcp != lp); 714 715 /* 716 * Ok, we have to setrunqueue some target cpu and request a reschedule 717 * if necessary. 718 * 719 * We have to choose the best target cpu. It might not be the current 720 * target even if the current cpu has no running user thread (for 721 * example, because the current cpu might be a hyperthread and its 722 * sibling has a thread assigned). 723 * 724 * If we just forked it is most optimal to run the child on the same 725 * cpu just in case the parent decides to wait for it (thus getting 726 * off that cpu). As long as there is nothing else runnable on the 727 * cpu, that is. If we did this unconditionally a parent forking 728 * multiple children before waiting (e.g. make -j N) leaves other 729 * cpus idle that could be working. 730 */ 731 if (lp->lwp_forked) { 732 lp->lwp_forked = 0; 733 if (usched_dfly_features & 0x20) 734 rdd = dfly_choose_best_queue(lp); 735 else if (usched_dfly_features & 0x40) 736 rdd = &dfly_pcpu[lp->lwp_qcpu]; 737 else if (usched_dfly_features & 0x80) 738 rdd = dfly_choose_queue_simple(rdd, lp); 739 else if (dfly_pcpu[lp->lwp_qcpu].runqcount) 740 rdd = dfly_choose_best_queue(lp); 741 else 742 rdd = &dfly_pcpu[lp->lwp_qcpu]; 743 } else { 744 rdd = dfly_choose_best_queue(lp); 745 /* rdd = &dfly_pcpu[lp->lwp_qcpu]; */ 746 } 747 if (lp->lwp_qcpu != rdd->cpuid) { 748 spin_lock(&dd->spin); 749 dfly_changeqcpu_locked(lp, dd, rdd); 750 spin_unlock(&dd->spin); 751 } 752 dfly_setrunqueue_dd(rdd, lp); 753 } 754 755 /* 756 * Change qcpu to rdd->cpuid. The dd the lp is CURRENTLY on must be 757 * spin-locked on-call. rdd does not have to be. 758 */ 759 static void 760 dfly_changeqcpu_locked(struct lwp *lp, dfly_pcpu_t dd, dfly_pcpu_t rdd) 761 { 762 if (lp->lwp_qcpu != rdd->cpuid) { 763 spin_lock(&lp->lwp_spin); 764 if (lp->lwp_mpflags & LWP_MP_ULOAD) { 765 atomic_clear_int(&lp->lwp_mpflags, LWP_MP_ULOAD); 766 atomic_add_long(&dd->uload, -lp->lwp_uload); 767 atomic_add_int(&dd->ucount, -1); 768 } 769 lp->lwp_qcpu = rdd->cpuid; 770 spin_unlock(&lp->lwp_spin); 771 } 772 } 773 774 /* 775 * Place lp on rdd's runqueue. Nothing is locked on call. This function 776 * also performs all necessary ancillary notification actions. 777 */ 778 static void 779 dfly_setrunqueue_dd(dfly_pcpu_t rdd, struct lwp *lp) 780 { 781 globaldata_t rgd; 782 783 /* 784 * We might be moving the lp to another cpu's run queue, and once 785 * on the runqueue (even if it is our cpu's), another cpu can rip 786 * it away from us. 787 * 788 * TDF_MIGRATING might already be set if this is part of a 789 * remrunqueue+setrunqueue sequence. 790 */ 791 if ((lp->lwp_thread->td_flags & TDF_MIGRATING) == 0) 792 lwkt_giveaway(lp->lwp_thread); 793 794 rgd = rdd->gd; 795 796 /* 797 * We lose control of the lp the moment we release the spinlock 798 * after having placed it on the queue. i.e. another cpu could pick 799 * it up, or it could exit, or its priority could be further 800 * adjusted, or something like that. 801 * 802 * WARNING! rdd can point to a foreign cpu! 803 */ 804 spin_lock(&rdd->spin); 805 dfly_setrunqueue_locked(rdd, lp); 806 807 /* 808 * Potentially interrupt the currently-running thread 809 */ 810 if ((rdd->upri & ~PPQMASK) <= (lp->lwp_priority & ~PPQMASK)) { 811 /* 812 * Currently running thread is better or same, do not 813 * interrupt. 814 */ 815 spin_unlock(&rdd->spin); 816 } else if ((rdd->upri & ~PPQMASK) <= (lp->lwp_priority & ~PPQMASK) + 817 usched_dfly_fast_resched) { 818 /* 819 * Currently running thread is not better, but not so bad 820 * that we need to interrupt it. Let it run for one more 821 * scheduler tick. 822 */ 823 if (rdd->uschedcp && 824 rdd->uschedcp->lwp_rrcount < usched_dfly_rrinterval) { 825 rdd->uschedcp->lwp_rrcount = usched_dfly_rrinterval - 1; 826 } 827 spin_unlock(&rdd->spin); 828 } else if (rgd == mycpu) { 829 /* 830 * We should interrupt the currently running thread, which 831 * is on the current cpu. However, if DIDYIELD is set we 832 * round-robin unconditionally and do not interrupt it. 833 */ 834 spin_unlock(&rdd->spin); 835 if (rdd->uschedcp == NULL) 836 wakeup_mycpu(rdd->helper_thread); /* XXX */ 837 if ((lp->lwp_thread->td_mpflags & TDF_MP_DIDYIELD) == 0) 838 need_user_resched(); 839 } else { 840 /* 841 * We should interrupt the currently running thread, which 842 * is on a different cpu. 843 */ 844 spin_unlock(&rdd->spin); 845 lwkt_send_ipiq(rgd, dfly_need_user_resched_remote, NULL); 846 } 847 } 848 849 /* 850 * This routine is called from a systimer IPI. It MUST be MP-safe and 851 * the BGL IS NOT HELD ON ENTRY. This routine is called at ESTCPUFREQ on 852 * each cpu. 853 */ 854 static 855 void 856 dfly_schedulerclock(struct lwp *lp, sysclock_t period, sysclock_t cpstamp) 857 { 858 globaldata_t gd = mycpu; 859 dfly_pcpu_t dd = &dfly_pcpu[gd->gd_cpuid]; 860 861 /* 862 * Spinlocks also hold a critical section so there should not be 863 * any active. 864 */ 865 KKASSERT(gd->gd_spinlocks == 0 || dumping); 866 867 /* 868 * If lp is NULL we might be contended and lwkt_switch() may have 869 * cycled into the idle thread. Apply the tick to the current 870 * process on this cpu if it is contended. 871 */ 872 if (gd->gd_curthread == &gd->gd_idlethread) { 873 lp = dd->uschedcp; 874 if (lp && (lp->lwp_thread == NULL || 875 lp->lwp_thread->td_contended == 0)) { 876 lp = NULL; 877 } 878 } 879 880 /* 881 * Dock thread for tick 882 */ 883 if (lp) { 884 /* 885 * Do we need to round-robin? We round-robin 10 times a 886 * second. This should only occur for cpu-bound batch 887 * processes. 888 */ 889 if (++lp->lwp_rrcount >= usched_dfly_rrinterval) 890 need_user_resched(); 891 if ((lp->lwp_thread->td_mpflags & TDF_MP_BATCH_DEMARC) && 892 lp->lwp_rrcount >= usched_dfly_rrinterval / 2) { 893 need_user_resched(); 894 } 895 896 /* 897 * Adjust estcpu upward using a real time equivalent 898 * calculation, and recalculate lp's priority. Estcpu 899 * is increased such that it will cap-out over a period 900 * of one second. 901 */ 902 lp->lwp_estcpu = ESTCPULIM(lp->lwp_estcpu + 903 ESTCPUMAX / ESTCPUFREQ + 1); 904 dfly_resetpriority(lp); 905 } 906 907 /* 908 * Rebalance two cpus every 8 ticks, pulling the worst thread 909 * from the worst cpu's queue into a rotating cpu number. 910 * Also require that the moving of the highest-load thread 911 * from rdd to dd does not cause the uload to cross over. 912 * 913 * This mechanic is needed because the push algorithms can 914 * steady-state in an non-optimal configuration. We need to mix it 915 * up a little, even if it means breaking up a paired thread, so 916 * the push algorithms can rebalance the degenerate conditions. 917 * This portion of the algorithm exists to ensure stability at the 918 * selected weightings. 919 * 920 * Because we might be breaking up optimal conditions we do not want 921 * to execute this too quickly, hence we only rebalance approximately 922 * ~7-8 times per second. The push's, on the otherhand, are capable 923 * moving threads to other cpus at a much higher rate. 924 * 925 * We choose the most heavily loaded thread from the worst queue 926 * in order to ensure that multiple heavy-weight threads on the same 927 * queue get broken up, and also because these threads are the most 928 * likely to be able to remain in place. Hopefully then any pairings, 929 * if applicable, migrate to where these threads are. 930 */ 931 if ((usched_dfly_features & 0x04) && 932 ((u_int)sched_ticks & 7) == 0 && 933 (u_int)sched_ticks / 8 % ncpus == gd->gd_cpuid) { 934 /* 935 * Our cpu is up. 936 */ 937 struct lwp *nlp; 938 dfly_pcpu_t rdd; 939 940 rdd = dfly_choose_worst_queue(dd, 1); 941 if (rdd && dd->uload + usched_dfly_weight6 / 2 < rdd->uload) { 942 spin_lock(&dd->spin); 943 if (spin_trylock(&rdd->spin)) { 944 nlp = dfly_chooseproc_locked(rdd, dd, NULL, 1); 945 spin_unlock(&rdd->spin); 946 if (nlp == NULL) 947 spin_unlock(&dd->spin); 948 } else { 949 spin_unlock(&dd->spin); 950 nlp = NULL; 951 } 952 } else { 953 nlp = NULL; 954 } 955 /* dd->spin held if nlp != NULL */ 956 957 /* 958 * Either schedule it or add it to our queue. 959 */ 960 if (nlp && 961 (nlp->lwp_priority & ~PPQMASK) < (dd->upri & ~PPQMASK)) { 962 if ((dd->flags & DFLY_PCPU_CURMASK) == 0) { 963 ATOMIC_CPUMASK_ORMASK(dfly_curprocmask, 964 dd->cpumask); 965 dd->flags |= DFLY_PCPU_CURMASK; 966 } 967 dd->upri = nlp->lwp_priority; 968 dd->uschedcp = nlp; 969 #if 0 970 dd->rrcount = 0; /* reset round robin */ 971 #endif 972 spin_unlock(&dd->spin); 973 lwkt_acquire(nlp->lwp_thread); 974 lwkt_schedule(nlp->lwp_thread); 975 } else if (nlp) { 976 dfly_setrunqueue_locked(dd, nlp); 977 spin_unlock(&dd->spin); 978 } 979 } 980 } 981 982 /* 983 * Called from acquire and from kern_synch's one-second timer (one of the 984 * callout helper threads) with a critical section held. 985 * 986 * Adjust p_estcpu based on our single-cpu load, p_nice, and compensate for 987 * overall system load. 988 * 989 * Note that no recalculation occurs for a process which sleeps and wakes 990 * up in the same tick. That is, a system doing thousands of context 991 * switches per second will still only do serious estcpu calculations 992 * ESTCPUFREQ times per second. 993 */ 994 static 995 void 996 dfly_recalculate_estcpu(struct lwp *lp) 997 { 998 globaldata_t gd = mycpu; 999 sysclock_t cpbase; 1000 sysclock_t ttlticks; 1001 int estcpu; 1002 int decay_factor; 1003 int ucount; 1004 1005 /* 1006 * We have to subtract periodic to get the last schedclock 1007 * timeout time, otherwise we would get the upcoming timeout. 1008 * Keep in mind that a process can migrate between cpus and 1009 * while the scheduler clock should be very close, boundary 1010 * conditions could lead to a small negative delta. 1011 */ 1012 cpbase = gd->gd_schedclock.time - gd->gd_schedclock.periodic; 1013 1014 if (lp->lwp_slptime > 1) { 1015 /* 1016 * Too much time has passed, do a coarse correction. 1017 */ 1018 lp->lwp_estcpu = lp->lwp_estcpu >> 1; 1019 dfly_resetpriority(lp); 1020 lp->lwp_cpbase = cpbase; 1021 lp->lwp_cpticks = 0; 1022 lp->lwp_estfast = 0; 1023 } else if (lp->lwp_cpbase != cpbase) { 1024 /* 1025 * Adjust estcpu if we are in a different tick. Don't waste 1026 * time if we are in the same tick. 1027 * 1028 * First calculate the number of ticks in the measurement 1029 * interval. The ttlticks calculation can wind up 0 due to 1030 * a bug in the handling of lwp_slptime (as yet not found), 1031 * so make sure we do not get a divide by 0 panic. 1032 */ 1033 ttlticks = (cpbase - lp->lwp_cpbase) / 1034 gd->gd_schedclock.periodic; 1035 if ((ssysclock_t)ttlticks < 0) { 1036 ttlticks = 0; 1037 lp->lwp_cpbase = cpbase; 1038 } 1039 if (ttlticks < 4) 1040 return; 1041 updatepcpu(lp, lp->lwp_cpticks, ttlticks); 1042 1043 /* 1044 * Calculate instant estcpu based percentage of (one) cpu 1045 * used and exponentially average it into the current 1046 * lwp_estcpu. 1047 */ 1048 ucount = dfly_pcpu[lp->lwp_qcpu].ucount; 1049 estcpu = lp->lwp_cpticks * ESTCPUMAX / ttlticks; 1050 1051 /* 1052 * The higher ttlticks gets, the more meaning the calculation 1053 * has and the smaller our decay_factor in the exponential 1054 * average. 1055 * 1056 * The uload calculation has been removed because it actually 1057 * makes things worse, causing processes which use less cpu 1058 * (such as a browser) to be pumped up and treated the same 1059 * as a cpu-bound process (such as a make). The same effect 1060 * can occur with sufficient load without the uload 1061 * calculation, but occurs less quickly and takes more load. 1062 * In addition, the less cpu a process uses the smaller the 1063 * effect of the overload. 1064 */ 1065 if (ttlticks >= hz) 1066 decay_factor = 1; 1067 else 1068 decay_factor = hz - ttlticks; 1069 1070 lp->lwp_estcpu = ESTCPULIM( 1071 (lp->lwp_estcpu * ttlticks + estcpu) / 1072 (ttlticks + 1)); 1073 dfly_resetpriority(lp); 1074 lp->lwp_cpbase += ttlticks * gd->gd_schedclock.periodic; 1075 lp->lwp_cpticks = 0; 1076 } 1077 } 1078 1079 /* 1080 * Compute the priority of a process when running in user mode. 1081 * Arrange to reschedule if the resulting priority is better 1082 * than that of the current process. 1083 * 1084 * This routine may be called with any process. 1085 * 1086 * This routine is called by fork1() for initial setup with the process of 1087 * the run queue, and also may be called normally with the process on or 1088 * off the run queue. 1089 */ 1090 static void 1091 dfly_resetpriority(struct lwp *lp) 1092 { 1093 dfly_pcpu_t rdd; 1094 int newpriority; 1095 u_short newrqtype; 1096 int rcpu; 1097 int checkpri; 1098 int estcpu; 1099 int delta_uload; 1100 1101 crit_enter(); 1102 1103 /* 1104 * Lock the scheduler (lp) belongs to. This can be on a different 1105 * cpu. Handle races. This loop breaks out with the appropriate 1106 * rdd locked. 1107 */ 1108 for (;;) { 1109 rcpu = lp->lwp_qcpu; 1110 cpu_ccfence(); 1111 rdd = &dfly_pcpu[rcpu]; 1112 spin_lock(&rdd->spin); 1113 if (rcpu == lp->lwp_qcpu) 1114 break; 1115 spin_unlock(&rdd->spin); 1116 } 1117 1118 /* 1119 * Calculate the new priority and queue type 1120 */ 1121 newrqtype = lp->lwp_rtprio.type; 1122 1123 switch(newrqtype) { 1124 case RTP_PRIO_REALTIME: 1125 case RTP_PRIO_FIFO: 1126 newpriority = PRIBASE_REALTIME + 1127 (lp->lwp_rtprio.prio & PRIMASK); 1128 break; 1129 case RTP_PRIO_NORMAL: 1130 /* 1131 * Calculate the new priority. 1132 * 1133 * nice contributes up to NICE_QS queues (typ 32 - full range) 1134 * estcpu contributes up to EST_QS queues (typ 24) 1135 * 1136 * A nice +20 process receives 1/10 cpu vs nice+0. Niced 1137 * process more than 20 apart may receive no cpu, so cpu 1138 * bound nice -20 can prevent a nice +5 from getting any 1139 * cpu. A nice+0, being in the middle, always gets some cpu 1140 * no matter what. 1141 */ 1142 estcpu = lp->lwp_estcpu; 1143 newpriority = (lp->lwp_proc->p_nice - PRIO_MIN) * 1144 (NICE_QS * PPQ) / PRIO_RANGE; 1145 newpriority += estcpu * PPQ / ESTCPUPPQ; 1146 if (newpriority < 0) 1147 newpriority = 0; 1148 if (newpriority >= MAXPRI) 1149 newpriority = MAXPRI - 1; 1150 newpriority += PRIBASE_NORMAL; 1151 break; 1152 case RTP_PRIO_IDLE: 1153 newpriority = PRIBASE_IDLE + (lp->lwp_rtprio.prio & PRIMASK); 1154 break; 1155 case RTP_PRIO_THREAD: 1156 newpriority = PRIBASE_THREAD + (lp->lwp_rtprio.prio & PRIMASK); 1157 break; 1158 default: 1159 panic("Bad RTP_PRIO %d", newrqtype); 1160 /* NOT REACHED */ 1161 } 1162 1163 /* 1164 * The LWKT scheduler doesn't dive usched structures, give it a hint 1165 * on the relative priority of user threads running in the kernel. 1166 * The LWKT scheduler will always ensure that a user thread running 1167 * in the kernel will get cpu some time, regardless of its upri, 1168 * but can decide not to instantly switch from one kernel or user 1169 * mode user thread to a kernel-mode user thread when it has a less 1170 * desireable user priority. 1171 * 1172 * td_upri has normal sense (higher values are more desireable), so 1173 * negate it (this is a different field lp->lwp_priority) 1174 */ 1175 lp->lwp_thread->td_upri = -(newpriority & usched_dfly_swmask); 1176 1177 /* 1178 * The newpriority incorporates the queue type so do a simple masked 1179 * check to determine if the process has moved to another queue. If 1180 * it has, and it is currently on a run queue, then move it. 1181 * 1182 * Since uload is ~PPQMASK masked, no modifications are necessary if 1183 * we end up in the same run queue. 1184 * 1185 * Reset rrcount if moving to a higher-priority queue, otherwise 1186 * retain rrcount. 1187 */ 1188 if ((lp->lwp_priority ^ newpriority) & ~PPQMASK) { 1189 if (lp->lwp_priority < newpriority) 1190 lp->lwp_rrcount = 0; 1191 if (lp->lwp_mpflags & LWP_MP_ONRUNQ) { 1192 dfly_remrunqueue_locked(rdd, lp); 1193 lp->lwp_priority = newpriority; 1194 lp->lwp_rqtype = newrqtype; 1195 lp->lwp_rqindex = (newpriority & PRIMASK) / PPQ; 1196 dfly_setrunqueue_locked(rdd, lp); 1197 checkpri = 1; 1198 } else { 1199 lp->lwp_priority = newpriority; 1200 lp->lwp_rqtype = newrqtype; 1201 lp->lwp_rqindex = (newpriority & PRIMASK) / PPQ; 1202 checkpri = 0; 1203 } 1204 } else { 1205 /* 1206 * In the same PPQ, uload cannot change. 1207 */ 1208 lp->lwp_priority = newpriority; 1209 checkpri = 1; 1210 rcpu = -1; 1211 } 1212 1213 /* 1214 * Adjust effective load. 1215 * 1216 * Calculate load then scale up or down geometrically based on p_nice. 1217 * Processes niced up (positive) are less important, and processes 1218 * niced downard (negative) are more important. The higher the uload, 1219 * the more important the thread. 1220 */ 1221 /* 0-511, 0-100% cpu */ 1222 spin_lock(&lp->lwp_spin); 1223 delta_uload = lptouload(lp); 1224 delta_uload -= lp->lwp_uload; 1225 if (lp->lwp_uload + delta_uload < -32767) { 1226 delta_uload = -32768 - lp->lwp_uload; 1227 } else if (lp->lwp_uload + delta_uload > 32767) { 1228 delta_uload = 32767 - lp->lwp_uload; 1229 } 1230 lp->lwp_uload += delta_uload; 1231 if (lp->lwp_mpflags & LWP_MP_ULOAD) 1232 atomic_add_long(&dfly_pcpu[lp->lwp_qcpu].uload, delta_uload); 1233 spin_unlock(&lp->lwp_spin); 1234 1235 /* 1236 * Determine if we need to reschedule the target cpu. This only 1237 * occurs if the LWP is already on a scheduler queue, which means 1238 * that idle cpu notification has already occured. At most we 1239 * need only issue a need_user_resched() on the appropriate cpu. 1240 * 1241 * The LWP may be owned by a CPU different from the current one, 1242 * in which case dd->uschedcp may be modified without an MP lock 1243 * or a spinlock held. The worst that happens is that the code 1244 * below causes a spurious need_user_resched() on the target CPU 1245 * and dd->pri to be wrong for a short period of time, both of 1246 * which are harmless. 1247 * 1248 * If checkpri is 0 we are adjusting the priority of the current 1249 * process, possibly higher (less desireable), so ignore the upri 1250 * check which will fail in that case. 1251 */ 1252 if (rcpu >= 0) { 1253 if (CPUMASK_TESTBIT(dfly_rdyprocmask, rcpu) && 1254 (checkpri == 0 || 1255 (rdd->upri & ~PRIMASK) > 1256 (lp->lwp_priority & ~PRIMASK))) { 1257 if (rcpu == mycpu->gd_cpuid) { 1258 spin_unlock(&rdd->spin); 1259 need_user_resched(); 1260 } else { 1261 spin_unlock(&rdd->spin); 1262 lwkt_send_ipiq(globaldata_find(rcpu), 1263 dfly_need_user_resched_remote, 1264 NULL); 1265 } 1266 } else { 1267 spin_unlock(&rdd->spin); 1268 } 1269 } else { 1270 spin_unlock(&rdd->spin); 1271 } 1272 crit_exit(); 1273 } 1274 1275 static 1276 void 1277 dfly_yield(struct lwp *lp) 1278 { 1279 if (lp->lwp_qcpu != mycpu->gd_cpuid) 1280 return; 1281 KKASSERT(lp == curthread->td_lwp); 1282 1283 /* 1284 * Don't set need_user_resched() or mess with rrcount or anything. 1285 * the TDF flag will override everything as long as we release. 1286 */ 1287 atomic_set_int(&lp->lwp_thread->td_mpflags, TDF_MP_DIDYIELD); 1288 dfly_release_curproc(lp); 1289 } 1290 1291 /* 1292 * Thread was forcefully migrated to another cpu. Normally forced migrations 1293 * are used for iterations and the kernel returns to the original cpu before 1294 * returning and this is not needed. However, if the kernel migrates a 1295 * thread to another cpu and wants to leave it there, it has to call this 1296 * scheduler helper. 1297 * 1298 * Note that the lwkt_migratecpu() function also released the thread, so 1299 * we don't have to worry about that. 1300 */ 1301 static 1302 void 1303 dfly_changedcpu(struct lwp *lp) 1304 { 1305 dfly_pcpu_t dd = &dfly_pcpu[lp->lwp_qcpu]; 1306 dfly_pcpu_t rdd = &dfly_pcpu[mycpu->gd_cpuid]; 1307 1308 if (dd != rdd) { 1309 spin_lock(&dd->spin); 1310 dfly_changeqcpu_locked(lp, dd, rdd); 1311 spin_unlock(&dd->spin); 1312 } 1313 } 1314 1315 /* 1316 * Called from fork1() when a new child process is being created. 1317 * 1318 * Give the child process an initial estcpu that is more batch then 1319 * its parent and dock the parent for the fork (but do not 1320 * reschedule the parent). 1321 * 1322 * fast 1323 * 1324 * XXX lwp should be "spawning" instead of "forking" 1325 */ 1326 static void 1327 dfly_forking(struct lwp *plp, struct lwp *lp) 1328 { 1329 int estcpu; 1330 1331 /* 1332 * Put the child 4 queue slots (out of 32) higher than the parent 1333 * (less desireable than the parent). 1334 */ 1335 lp->lwp_estcpu = ESTCPULIM(plp->lwp_estcpu + 1336 ESTCPUPPQ * usched_dfly_forkbias); 1337 lp->lwp_forked = 1; 1338 lp->lwp_estfast = 0; 1339 1340 /* 1341 * Even though the lp will be scheduled specially the first time 1342 * due to lp->lwp_forked, it is important to initialize lwp_qcpu 1343 * to avoid favoring a fixed cpu. XXX 1344 */ 1345 #if 0 1346 static uint16_t save_cpu; 1347 lp->lwp_qcpu = ++save_cpu % ncpus; 1348 #else 1349 lp->lwp_qcpu = plp->lwp_qcpu; 1350 if (CPUMASK_TESTBIT(lp->lwp_cpumask, lp->lwp_qcpu) == 0) 1351 lp->lwp_qcpu = BSFCPUMASK(lp->lwp_cpumask); 1352 #endif 1353 1354 /* 1355 * Dock the parent a cost for the fork, protecting us from fork 1356 * bombs. If the parent is forking quickly this makes both the 1357 * parent and child more batchy. 1358 */ 1359 estcpu = plp->lwp_estcpu + ESTCPUPPQ / 16; 1360 plp->lwp_estcpu = ESTCPULIM(estcpu); 1361 } 1362 1363 /* 1364 * Called when a lwp is being removed from this scheduler, typically 1365 * during lwp_exit(). We have to clean out any ULOAD accounting before 1366 * we can let the lp go. 1367 * 1368 * Scheduler dequeueing has already occurred, no further action in that 1369 * regard is needed. 1370 */ 1371 static void 1372 dfly_exiting(struct lwp *lp, struct proc *child_proc) 1373 { 1374 dfly_pcpu_t dd; 1375 1376 spin_lock(&lp->lwp_spin); 1377 dd = &dfly_pcpu[lp->lwp_qcpu]; 1378 if (lp->lwp_mpflags & LWP_MP_ULOAD) { 1379 atomic_clear_int(&lp->lwp_mpflags, LWP_MP_ULOAD); 1380 atomic_add_long(&dd->uload, -lp->lwp_uload); 1381 atomic_add_int(&dd->ucount, -1); 1382 } 1383 spin_unlock(&lp->lwp_spin); 1384 } 1385 1386 /* 1387 * This function cannot block in any way, but spinlocks are ok. 1388 * 1389 * Update the uload based on the state of the thread (whether it is going 1390 * to sleep or running again). The uload is meant to be a longer-term 1391 * load and not an instantanious load. 1392 */ 1393 static void 1394 dfly_uload_update(struct lwp *lp) 1395 { 1396 dfly_pcpu_t dd; 1397 1398 if (lp->lwp_thread->td_flags & TDF_RUNQ) { 1399 if ((lp->lwp_mpflags & LWP_MP_ULOAD) == 0) { 1400 spin_lock(&lp->lwp_spin); 1401 dd = &dfly_pcpu[lp->lwp_qcpu]; 1402 if ((lp->lwp_mpflags & LWP_MP_ULOAD) == 0) { 1403 atomic_set_int(&lp->lwp_mpflags, 1404 LWP_MP_ULOAD); 1405 atomic_add_long(&dd->uload, lp->lwp_uload); 1406 atomic_add_int(&dd->ucount, 1); 1407 } 1408 spin_unlock(&lp->lwp_spin); 1409 } 1410 } else if (lp->lwp_slptime > 0) { 1411 if (lp->lwp_mpflags & LWP_MP_ULOAD) { 1412 spin_lock(&lp->lwp_spin); 1413 dd = &dfly_pcpu[lp->lwp_qcpu]; 1414 if (lp->lwp_mpflags & LWP_MP_ULOAD) { 1415 atomic_clear_int(&lp->lwp_mpflags, 1416 LWP_MP_ULOAD); 1417 atomic_add_long(&dd->uload, -lp->lwp_uload); 1418 atomic_add_int(&dd->ucount, -1); 1419 } 1420 spin_unlock(&lp->lwp_spin); 1421 } 1422 } 1423 } 1424 1425 /* 1426 * chooseproc() is called when a cpu needs a user process to LWKT schedule, 1427 * it selects a user process and returns it. If chklp is non-NULL and chklp 1428 * has a better or equal priority then the process that would otherwise be 1429 * chosen, NULL is returned. 1430 * 1431 * Until we fix the RUNQ code the chklp test has to be strict or we may 1432 * bounce between processes trying to acquire the current process designation. 1433 * 1434 * Must be called with rdd->spin locked. The spinlock is left intact through 1435 * the entire routine. dd->spin does not have to be locked. 1436 * 1437 * If worst is non-zero this function finds the worst thread instead of the 1438 * best thread (used by the schedulerclock-based rover). 1439 */ 1440 static 1441 struct lwp * 1442 dfly_chooseproc_locked(dfly_pcpu_t rdd, dfly_pcpu_t dd, 1443 struct lwp *chklp, int worst) 1444 { 1445 struct lwp *lp; 1446 struct rq *q; 1447 u_int32_t *which; 1448 u_int32_t pri; 1449 u_int32_t rtqbits; 1450 u_int32_t tsqbits; 1451 u_int32_t idqbits; 1452 1453 /* 1454 * Select best or worst process. Once selected, clear the bit 1455 * in our local variable (idqbits, tsqbits, or rtqbits) just 1456 * in case we have to loop. 1457 */ 1458 rtqbits = rdd->rtqueuebits; 1459 tsqbits = rdd->queuebits; 1460 idqbits = rdd->idqueuebits; 1461 1462 loopfar: 1463 if (worst) { 1464 if (idqbits) { 1465 pri = bsrl(idqbits); 1466 idqbits &= ~(1U << pri); 1467 q = &rdd->idqueues[pri]; 1468 which = &rdd->idqueuebits; 1469 } else if (tsqbits) { 1470 pri = bsrl(tsqbits); 1471 tsqbits &= ~(1U << pri); 1472 q = &rdd->queues[pri]; 1473 which = &rdd->queuebits; 1474 } else if (rtqbits) { 1475 pri = bsrl(rtqbits); 1476 rtqbits &= ~(1U << pri); 1477 q = &rdd->rtqueues[pri]; 1478 which = &rdd->rtqueuebits; 1479 } else { 1480 return (NULL); 1481 } 1482 lp = TAILQ_LAST(q, rq); 1483 } else { 1484 if (rtqbits) { 1485 pri = bsfl(rtqbits); 1486 rtqbits &= ~(1U << pri); 1487 q = &rdd->rtqueues[pri]; 1488 which = &rdd->rtqueuebits; 1489 } else if (tsqbits) { 1490 pri = bsfl(tsqbits); 1491 tsqbits &= ~(1U << pri); 1492 q = &rdd->queues[pri]; 1493 which = &rdd->queuebits; 1494 } else if (idqbits) { 1495 pri = bsfl(idqbits); 1496 idqbits &= ~(1U << pri); 1497 q = &rdd->idqueues[pri]; 1498 which = &rdd->idqueuebits; 1499 } else { 1500 return (NULL); 1501 } 1502 lp = TAILQ_FIRST(q); 1503 } 1504 KASSERT(lp, ("chooseproc: no lwp on busy queue")); 1505 1506 loopnear: 1507 /* 1508 * If the passed lwp <chklp> is reasonably close to the selected 1509 * lwp <lp>, return NULL (indicating that <chklp> should be kept). 1510 * 1511 * Note that we must error on the side of <chklp> to avoid bouncing 1512 * between threads in the acquire code. 1513 */ 1514 if (chklp) { 1515 if (chklp->lwp_priority < lp->lwp_priority + PPQ) 1516 return(NULL); 1517 } 1518 1519 /* 1520 * When rdd != dd, we have to make sure that the process we 1521 * are pulling is allow to run on our cpu. This alternative 1522 * path is a bit more expensive but its not considered to be 1523 * in the critical path. 1524 */ 1525 if (rdd != dd && CPUMASK_TESTBIT(lp->lwp_cpumask, dd->cpuid) == 0) { 1526 if (worst) 1527 lp = TAILQ_PREV(lp, rq, lwp_procq); 1528 else 1529 lp = TAILQ_NEXT(lp, lwp_procq); 1530 if (lp) 1531 goto loopnear; 1532 goto loopfar; 1533 } 1534 1535 KTR_COND_LOG(usched_chooseproc, 1536 lp->lwp_proc->p_pid == usched_dfly_pid_debug, 1537 lp->lwp_proc->p_pid, 1538 lp->lwp_thread->td_gd->gd_cpuid, 1539 mycpu->gd_cpuid); 1540 1541 KASSERT((lp->lwp_mpflags & LWP_MP_ONRUNQ) != 0, ("not on runq6!")); 1542 atomic_clear_int(&lp->lwp_mpflags, LWP_MP_ONRUNQ); 1543 TAILQ_REMOVE(q, lp, lwp_procq); 1544 --rdd->runqcount; 1545 if (TAILQ_EMPTY(q)) 1546 *which &= ~(1 << pri); 1547 1548 /* 1549 * If we are choosing a process from rdd with the intent to 1550 * move it to dd, lwp_qcpu must be adjusted while rdd's spinlock 1551 * is still held. 1552 */ 1553 if (rdd != dd) { 1554 spin_lock(&lp->lwp_spin); 1555 if (lp->lwp_mpflags & LWP_MP_ULOAD) { 1556 atomic_add_long(&rdd->uload, -lp->lwp_uload); 1557 atomic_add_int(&rdd->ucount, -1); 1558 } 1559 lp->lwp_qcpu = dd->cpuid; 1560 atomic_add_long(&dd->uload, lp->lwp_uload); 1561 atomic_add_int(&dd->ucount, 1); 1562 atomic_set_int(&lp->lwp_mpflags, LWP_MP_ULOAD); 1563 spin_unlock(&lp->lwp_spin); 1564 } 1565 return lp; 1566 } 1567 1568 /* 1569 * USED TO PUSH RUNNABLE LWPS TO THE LEAST LOADED CPU. 1570 * 1571 * Choose a cpu node to schedule lp on, hopefully nearby its current 1572 * node. 1573 * 1574 * We give the current node a modest advantage for obvious reasons. 1575 * 1576 * We also give the node the thread was woken up FROM a slight advantage 1577 * in order to try to schedule paired threads which synchronize/block waiting 1578 * for each other fairly close to each other. Similarly in a network setting 1579 * this feature will also attempt to place a user process near the kernel 1580 * protocol thread that is feeding it data. THIS IS A CRITICAL PART of the 1581 * algorithm as it heuristically groups synchronizing processes for locality 1582 * of reference in multi-socket systems. 1583 * 1584 * We check against running processes and give a big advantage if there 1585 * are none running. 1586 * 1587 * The caller will normally dfly_setrunqueue() lp on the returned queue. 1588 * 1589 * When the topology is known choose a cpu whos group has, in aggregate, 1590 * has the lowest weighted load. 1591 */ 1592 static 1593 dfly_pcpu_t 1594 dfly_choose_best_queue(struct lwp *lp) 1595 { 1596 cpumask_t wakemask; 1597 cpumask_t mask; 1598 cpu_node_t *cpup; 1599 cpu_node_t *cpun; 1600 cpu_node_t *cpub; 1601 dfly_pcpu_t dd = &dfly_pcpu[lp->lwp_qcpu]; 1602 dfly_pcpu_t rdd; 1603 int wakecpu; 1604 int cpuid; 1605 int n; 1606 int loadav; 1607 long load; 1608 long lowest_load; 1609 1610 /* 1611 * When the topology is unknown choose a random cpu that is hopefully 1612 * idle. 1613 */ 1614 if (dd->cpunode == NULL) 1615 return (dfly_choose_queue_simple(dd, lp)); 1616 1617 loadav = (averunnable.ldavg[0] + FSCALE / 2) >> FSHIFT; 1618 1619 /* 1620 * Pairing mask 1621 */ 1622 if ((wakecpu = lp->lwp_thread->td_wakefromcpu) >= 0) 1623 wakemask = dfly_pcpu[wakecpu].cpumask; 1624 else 1625 CPUMASK_ASSZERO(wakemask); 1626 1627 if (usched_dfly_debug == lp->lwp_proc->p_pid) 1628 kprintf("choosebest wakefromcpu %d:\n", 1629 lp->lwp_thread->td_wakefromcpu); 1630 1631 /* 1632 * When the topology is known choose a cpu whos group has, in 1633 * aggregate, has the lowest weighted load. 1634 */ 1635 cpup = root_cpu_node; 1636 rdd = dd; 1637 1638 while (cpup) { 1639 /* 1640 * Degenerate case super-root 1641 */ 1642 if (cpup->child_no == 1) { 1643 cpup = cpup->child_node[0]; 1644 continue; 1645 } 1646 1647 /* 1648 * Terminal cpunode 1649 */ 1650 if (cpup->child_no == 0) { 1651 rdd = &dfly_pcpu[BSFCPUMASK(cpup->members)]; 1652 if (usched_dfly_debug == lp->lwp_proc->p_pid) 1653 kprintf(" last cpu %d\n", rdd->cpuid); 1654 break; 1655 } 1656 1657 cpub = NULL; 1658 lowest_load = 0x7FFFFFFFFFFFFFFFLL; 1659 if (usched_dfly_debug == lp->lwp_proc->p_pid) 1660 kprintf(" reset lowest_load for scan\n"); 1661 1662 for (n = 0; n < cpup->child_no; ++n) { 1663 /* 1664 * Accumulate load information for all cpus 1665 * which are members of this node. 1666 */ 1667 int count; 1668 1669 cpun = cpup->child_node[n]; 1670 mask = cpun->members; 1671 CPUMASK_ANDMASK(mask, usched_global_cpumask); 1672 CPUMASK_ANDMASK(mask, smp_active_mask); 1673 CPUMASK_ANDMASK(mask, lp->lwp_cpumask); 1674 if (CPUMASK_TESTZERO(mask)) 1675 continue; 1676 1677 load = 0; 1678 count = 0; 1679 1680 if (usched_dfly_debug == lp->lwp_proc->p_pid) 1681 kprintf(" mask:"); 1682 while (CPUMASK_TESTNZERO(mask)) { 1683 cpuid = BSFCPUMASK(mask); 1684 rdd = &dfly_pcpu[cpuid]; 1685 1686 if (usched_dfly_debug == lp->lwp_proc->p_pid) 1687 kprintf(" %d", cpuid); 1688 1689 /* 1690 * Cumulative load for members. Note that 1691 * if (lp) is part of the group, lp's 1692 * contribution will be backed out later. 1693 */ 1694 load += rdd->uload; 1695 load += rdd->ucount * 1696 usched_dfly_weight3; 1697 1698 /* 1699 * If the node is running a less important 1700 * thread than our thread, give it an 1701 * advantage. Witha high-enough weighting 1702 * this can override most other considerations 1703 * to provide ultimate priority fairness at 1704 * the cost of localization. 1705 */ 1706 if ((rdd->upri & ~PPQMASK) > 1707 (lp->lwp_priority & ~PPQMASK)) { 1708 load -= usched_dfly_weight4; 1709 } 1710 1711 #if 0 1712 if (rdd->uschedcp == NULL && 1713 rdd->runqcount == 0 && 1714 rdd->gd->gd_tdrunqcount == 0 1715 ) { 1716 load += rdd->uload / 2; 1717 load += rdd->ucount * 1718 usched_dfly_weight3 / 2; 1719 } else { 1720 load += rdd->uload; 1721 load += rdd->ucount * 1722 usched_dfly_weight3; 1723 } 1724 #endif 1725 CPUMASK_NANDBIT(mask, cpuid); 1726 ++count; 1727 } 1728 1729 /* 1730 * Compensate if the lp is already accounted for in 1731 * the aggregate uload for this mask set. We want 1732 * to calculate the loads as if lp were not present, 1733 * otherwise the calculation is bogus. 1734 */ 1735 if ((lp->lwp_mpflags & LWP_MP_ULOAD) && 1736 CPUMASK_TESTMASK(dd->cpumask, cpun->members)) { 1737 load -= lp->lwp_uload; 1738 load -= usched_dfly_weight3; /* ucount */ 1739 } 1740 1741 if (usched_dfly_debug == lp->lwp_proc->p_pid) 1742 kprintf("\n accum_start c=%d ld=%ld " 1743 "cpu=%d ld/cnt=%ld ", 1744 count, load, rdd->cpuid, 1745 load / count); 1746 1747 /* 1748 * load is the aggregate load of count CPUs in the 1749 * group. For the weightings to work as intended, 1750 * we want an average per-cpu load. 1751 */ 1752 load = load / count; 1753 1754 /* 1755 * Advantage the cpu group (lp) is already on. 1756 */ 1757 if (CPUMASK_TESTMASK(cpun->members, dd->cpumask)) 1758 load -= usched_dfly_weight1; 1759 1760 if (usched_dfly_debug == lp->lwp_proc->p_pid) 1761 kprintf("B:%ld ", load); 1762 1763 /* 1764 * Advantage nodes with more memory 1765 */ 1766 if (usched_dfly_node_mem) { 1767 load -= cpun->phys_mem * usched_dfly_weight5 / 1768 usched_dfly_node_mem; 1769 } 1770 1771 if (usched_dfly_debug == lp->lwp_proc->p_pid) 1772 kprintf("C:%ld ", load); 1773 1774 /* 1775 * Advantage the cpu group we desire to pair (lp) 1776 * to, but Disadvantage hyperthreads on the same 1777 * core, or the same thread as the ipc peer. 1778 * 1779 * Under very heavy loads it is usually beneficial 1780 * to set kern.usched_dfly.ipc_smt to 1, and under 1781 * extreme loads it might be beneficial to also set 1782 * kern.usched_dfly.ipc_same to 1. 1783 * 1784 * load+ disadvantage 1785 * load- advantage 1786 */ 1787 if (CPUMASK_TESTMASK(cpun->members, wakemask)) { 1788 if (cpun->child_no) { 1789 if (cpun->type == CORE_LEVEL && 1790 usched_dfly_ipc_smt < 0 && 1791 loadav >= (ncpus >> 1)) { 1792 /* 1793 * Advantage at higher levels 1794 * of the topology. 1795 */ 1796 load -= usched_dfly_weight2; 1797 } else if (cpun->type == CORE_LEVEL && 1798 usched_dfly_ipc_smt == 0) { 1799 /* 1800 * Disadvantage the same core 1801 * when there are hyperthreads. 1802 */ 1803 load += usched_dfly_weight2; 1804 } else { 1805 /* 1806 * Advantage at higher levels 1807 * of the topology. 1808 */ 1809 load -= usched_dfly_weight2; 1810 } 1811 } else { 1812 /* 1813 * Disadvantage the last level (core 1814 * or hyperthread). Try to schedule 1815 * the ipc 1816 */ 1817 if (usched_dfly_ipc_same < 0 && 1818 loadav >= ncpus) { 1819 load -= usched_dfly_weight2; 1820 } else if (usched_dfly_ipc_same) { 1821 load -= usched_dfly_weight2; 1822 } else { 1823 load += usched_dfly_weight2; 1824 } 1825 } 1826 #if 0 1827 if (cpun->child_no != 0) { 1828 /* advantage */ 1829 load -= usched_dfly_weight2; 1830 } else { 1831 /* 1832 * 0x10 (disadvantage) 1833 * 0x00 (advantage) - default 1834 */ 1835 if (usched_dfly_features & 0x10) 1836 load += usched_dfly_weight2; 1837 else 1838 load -= usched_dfly_weight2; 1839 } 1840 #endif 1841 } 1842 1843 if (usched_dfly_debug == lp->lwp_proc->p_pid) 1844 kprintf("D:%ld ", load); 1845 1846 /* 1847 * Calculate the best load 1848 */ 1849 if (cpub == NULL || lowest_load > load || 1850 (lowest_load == load && 1851 CPUMASK_TESTMASK(cpun->members, dd->cpumask)) 1852 ) { 1853 lowest_load = load; 1854 cpub = cpun; 1855 } 1856 1857 if (usched_dfly_debug == lp->lwp_proc->p_pid) 1858 kprintf("low=%ld]\n", lowest_load); 1859 } 1860 cpup = cpub; 1861 } 1862 /* Dispatch this outcast to a proper CPU. */ 1863 if (__predict_false(CPUMASK_TESTBIT(lp->lwp_cpumask, rdd->cpuid) == 0)) 1864 rdd = &dfly_pcpu[BSFCPUMASK(lp->lwp_cpumask)]; 1865 if (usched_dfly_chooser > 0) { 1866 --usched_dfly_chooser; /* only N lines */ 1867 kprintf("lp %02d->%02d %s\n", 1868 lp->lwp_qcpu, rdd->cpuid, lp->lwp_proc->p_comm); 1869 } 1870 if (usched_dfly_debug == lp->lwp_proc->p_pid) 1871 kprintf("final cpu %d\n", rdd->cpuid); 1872 return (rdd); 1873 } 1874 1875 /* 1876 * USED TO PULL RUNNABLE LWPS FROM THE MOST LOADED CPU. 1877 * 1878 * Choose the worst queue close to dd's cpu node with a non-empty runq 1879 * that is NOT dd. 1880 * 1881 * This is used by the thread chooser when the current cpu's queues are 1882 * empty to steal a thread from another cpu's queue. We want to offload 1883 * the most heavily-loaded queue. 1884 * 1885 * However, we do not want to steal from far-away nodes who themselves 1886 * have idle cpu's that are more suitable to distribute the far-away 1887 * thread to. 1888 */ 1889 static 1890 dfly_pcpu_t 1891 dfly_choose_worst_queue(dfly_pcpu_t dd, int forceit) 1892 { 1893 cpumask_t mask; 1894 cpu_node_t *cpup; 1895 cpu_node_t *cpun; 1896 cpu_node_t *cpub; 1897 dfly_pcpu_t rdd; 1898 int cpuid; 1899 int n; 1900 int highest_runqcount; 1901 long load; 1902 long highest_load; 1903 #if 0 1904 int pri; 1905 int hpri; 1906 #endif 1907 1908 /* 1909 * When the topology is unknown choose a random cpu that is hopefully 1910 * idle. 1911 */ 1912 if (dd->cpunode == NULL) { 1913 return (NULL); 1914 } 1915 1916 /* 1917 * When the topology is known choose a cpu whos group has, in 1918 * aggregate, has the highest weighted load. 1919 */ 1920 cpup = root_cpu_node; 1921 rdd = dd; 1922 while (cpup) { 1923 /* 1924 * Degenerate case super-root 1925 */ 1926 if (cpup->child_no == 1) { 1927 cpup = cpup->child_node[0]; 1928 continue; 1929 } 1930 1931 /* 1932 * Terminal cpunode 1933 */ 1934 if (cpup->child_no == 0) { 1935 rdd = &dfly_pcpu[BSFCPUMASK(cpup->members)]; 1936 break; 1937 } 1938 1939 cpub = NULL; 1940 highest_load = -0x7FFFFFFFFFFFFFFFLL; 1941 1942 for (n = 0; n < cpup->child_no; ++n) { 1943 /* 1944 * Accumulate load information for all cpus 1945 * which are members of this node. 1946 */ 1947 int count; 1948 int runqcount; 1949 1950 cpun = cpup->child_node[n]; 1951 mask = cpun->members; 1952 CPUMASK_ANDMASK(mask, usched_global_cpumask); 1953 CPUMASK_ANDMASK(mask, smp_active_mask); 1954 if (CPUMASK_TESTZERO(mask)) 1955 continue; 1956 1957 load = 0; 1958 count = 0; 1959 runqcount = 0; 1960 1961 while (CPUMASK_TESTNZERO(mask)) { 1962 cpuid = BSFCPUMASK(mask); 1963 rdd = &dfly_pcpu[cpuid]; 1964 1965 load += rdd->uload; 1966 load += rdd->ucount * usched_dfly_weight3; 1967 1968 #if 0 1969 if (rdd->uschedcp == NULL && 1970 rdd->runqcount == 0 && 1971 rdd->gd->gd_tdrunqcount == 0 1972 ) { 1973 load += rdd->uload / 2; 1974 load += rdd->ucount * 1975 usched_dfly_weight3 / 2; 1976 } else { 1977 load += rdd->uload; 1978 load += rdd->ucount * 1979 usched_dfly_weight3; 1980 } 1981 #endif 1982 CPUMASK_NANDBIT(mask, cpuid); 1983 ++count; 1984 runqcount += rdd->runqcount; 1985 } 1986 load /= count; 1987 1988 /* 1989 * Advantage the cpu group (dd) is already on. 1990 * 1991 * When choosing the worst queue we reverse the 1992 * sign, but only count half the weight. 1993 * 1994 * weight1 needs to be high enough to be stable, 1995 * but this can also cause it to be too sticky, 1996 * so the iterator which rebalances the load sets 1997 * forceit to ignore it. 1998 */ 1999 if (forceit == 0 && 2000 CPUMASK_TESTMASK(dd->cpumask, cpun->members)) { 2001 load += usched_dfly_weight1 / 2; 2002 } 2003 2004 /* 2005 * Disadvantage nodes with more memory (same sign). 2006 */ 2007 if (usched_dfly_node_mem) { 2008 load -= cpun->phys_mem * usched_dfly_weight5 / 2009 usched_dfly_node_mem; 2010 } 2011 2012 2013 /* 2014 * The best candidate is the one with the worst 2015 * (highest) load, as long as it also has processes 2016 * on the run queue (verses running one and nothing 2017 * on the run queue). 2018 */ 2019 if (cpub == NULL || 2020 (runqcount && (highest_load < load || 2021 (highest_load == load && 2022 CPUMASK_TESTMASK(cpun->members, 2023 dd->cpumask)))) || 2024 (runqcount && highest_runqcount < runqcount + 1)) { 2025 highest_load = load; 2026 highest_runqcount = runqcount; 2027 cpub = cpun; 2028 } 2029 } 2030 cpup = cpub; 2031 } 2032 2033 /* 2034 * We never return our own node (dd), and only return a remote 2035 * node if it's load is significantly worse than ours (i.e. where 2036 * stealing a thread would be considered reasonable). 2037 * 2038 * This also helps us avoid breaking paired threads apart which 2039 * can have disastrous effects on performance. 2040 */ 2041 if (rdd == dd) 2042 return(NULL); 2043 2044 #if 0 2045 hpri = 0; 2046 if (rdd->rtqueuebits && hpri < (pri = bsrl(rdd->rtqueuebits))) 2047 hpri = pri; 2048 if (rdd->queuebits && hpri < (pri = bsrl(rdd->queuebits))) 2049 hpri = pri; 2050 if (rdd->idqueuebits && hpri < (pri = bsrl(rdd->idqueuebits))) 2051 hpri = pri; 2052 hpri *= PPQ; 2053 if (rdd->uload - hpri < dd->uload + hpri) 2054 return(NULL); 2055 #endif 2056 return (rdd); 2057 } 2058 2059 static 2060 dfly_pcpu_t 2061 dfly_choose_queue_simple(dfly_pcpu_t dd, struct lwp *lp) 2062 { 2063 dfly_pcpu_t rdd; 2064 cpumask_t tmpmask; 2065 cpumask_t mask; 2066 int cpubase; 2067 int cpuid; 2068 2069 /* 2070 * Fallback to the original heuristic, select random cpu, 2071 * first checking the cpus not currently running a user thread. 2072 * 2073 * Use cpuid as the base cpu in our scan, first checking 2074 * cpuid...(ncpus-1), then 0...(cpuid-1). This avoid favoring 2075 * lower-numbered cpus. 2076 */ 2077 ++dd->scancpu; /* SMP race ok */ 2078 mask = dfly_rdyprocmask; 2079 CPUMASK_NANDMASK(mask, dfly_curprocmask); 2080 CPUMASK_ANDMASK(mask, lp->lwp_cpumask); 2081 CPUMASK_ANDMASK(mask, smp_active_mask); 2082 CPUMASK_ANDMASK(mask, usched_global_cpumask); 2083 2084 cpubase = (int)(dd->scancpu % ncpus); 2085 CPUMASK_ASSBMASK(tmpmask, cpubase); 2086 CPUMASK_INVMASK(tmpmask); 2087 CPUMASK_ANDMASK(tmpmask, mask); 2088 while (CPUMASK_TESTNZERO(tmpmask)) { 2089 cpuid = BSFCPUMASK(tmpmask); 2090 rdd = &dfly_pcpu[cpuid]; 2091 2092 if ((rdd->upri & ~PPQMASK) >= (lp->lwp_priority & ~PPQMASK)) 2093 goto found; 2094 CPUMASK_NANDBIT(tmpmask, cpuid); 2095 } 2096 2097 CPUMASK_ASSBMASK(tmpmask, cpubase); 2098 CPUMASK_ANDMASK(tmpmask, mask); 2099 while (CPUMASK_TESTNZERO(tmpmask)) { 2100 cpuid = BSFCPUMASK(tmpmask); 2101 rdd = &dfly_pcpu[cpuid]; 2102 2103 if ((rdd->upri & ~PPQMASK) >= (lp->lwp_priority & ~PPQMASK)) 2104 goto found; 2105 CPUMASK_NANDBIT(tmpmask, cpuid); 2106 } 2107 2108 /* 2109 * Then cpus which might have a currently running lp 2110 */ 2111 mask = dfly_rdyprocmask; 2112 CPUMASK_ANDMASK(mask, dfly_curprocmask); 2113 CPUMASK_ANDMASK(mask, lp->lwp_cpumask); 2114 CPUMASK_ANDMASK(mask, smp_active_mask); 2115 CPUMASK_ANDMASK(mask, usched_global_cpumask); 2116 2117 CPUMASK_ASSBMASK(tmpmask, cpubase); 2118 CPUMASK_INVMASK(tmpmask); 2119 CPUMASK_ANDMASK(tmpmask, mask); 2120 while (CPUMASK_TESTNZERO(tmpmask)) { 2121 cpuid = BSFCPUMASK(tmpmask); 2122 rdd = &dfly_pcpu[cpuid]; 2123 2124 if ((rdd->upri & ~PPQMASK) > (lp->lwp_priority & ~PPQMASK)) 2125 goto found; 2126 CPUMASK_NANDBIT(tmpmask, cpuid); 2127 } 2128 2129 CPUMASK_ASSBMASK(tmpmask, cpubase); 2130 CPUMASK_ANDMASK(tmpmask, mask); 2131 while (CPUMASK_TESTNZERO(tmpmask)) { 2132 cpuid = BSFCPUMASK(tmpmask); 2133 rdd = &dfly_pcpu[cpuid]; 2134 2135 if ((rdd->upri & ~PPQMASK) > (lp->lwp_priority & ~PPQMASK)) 2136 goto found; 2137 CPUMASK_NANDBIT(tmpmask, cpuid); 2138 } 2139 2140 /* 2141 * If we cannot find a suitable cpu we round-robin using scancpu. 2142 * Other cpus will pickup as they release their current lwps or 2143 * become ready. 2144 * 2145 * Avoid a degenerate system lockup case if usched_global_cpumask 2146 * is set to 0 or otherwise does not cover lwp_cpumask. 2147 * 2148 * We only kick the target helper thread in this case, we do not 2149 * set the user resched flag because 2150 */ 2151 cpuid = cpubase; 2152 if (CPUMASK_TESTBIT(lp->lwp_cpumask, cpuid) == 0) 2153 cpuid = BSFCPUMASK(lp->lwp_cpumask); 2154 else if (CPUMASK_TESTBIT(usched_global_cpumask, cpuid) == 0) 2155 cpuid = 0; 2156 rdd = &dfly_pcpu[cpuid]; 2157 found: 2158 return (rdd); 2159 } 2160 2161 static 2162 void 2163 dfly_need_user_resched_remote(void *dummy) 2164 { 2165 globaldata_t gd = mycpu; 2166 dfly_pcpu_t dd = &dfly_pcpu[gd->gd_cpuid]; 2167 2168 /* 2169 * Flag reschedule needed 2170 */ 2171 need_user_resched(); 2172 2173 /* 2174 * If no user thread is currently running we need to kick the helper 2175 * on our cpu to recover. Otherwise the cpu will never schedule 2176 * anything again. 2177 * 2178 * We cannot schedule the process ourselves because this is an 2179 * IPI callback and we cannot acquire spinlocks in an IPI callback. 2180 * 2181 * Call wakeup_mycpu to avoid sending IPIs to other CPUs 2182 */ 2183 if (dd->uschedcp == NULL && (dd->flags & DFLY_PCPU_RDYMASK)) { 2184 ATOMIC_CPUMASK_NANDBIT(dfly_rdyprocmask, gd->gd_cpuid); 2185 dd->flags &= ~DFLY_PCPU_RDYMASK; 2186 wakeup_mycpu(dd->helper_thread); 2187 } 2188 } 2189 2190 /* 2191 * dfly_remrunqueue_locked() removes a given process from the run queue 2192 * that it is on, clearing the queue busy bit if it becomes empty. 2193 * 2194 * Note that user process scheduler is different from the LWKT schedule. 2195 * The user process scheduler only manages user processes but it uses LWKT 2196 * underneath, and a user process operating in the kernel will often be 2197 * 'released' from our management. 2198 * 2199 * uload is NOT adjusted here. It is only adjusted if the lwkt_thread goes 2200 * to sleep or the lwp is moved to a different runq. 2201 */ 2202 static void 2203 dfly_remrunqueue_locked(dfly_pcpu_t rdd, struct lwp *lp) 2204 { 2205 struct rq *q; 2206 u_int32_t *which; 2207 u_int8_t pri; 2208 2209 KKASSERT(rdd->runqcount >= 0); 2210 2211 pri = lp->lwp_rqindex; 2212 2213 switch(lp->lwp_rqtype) { 2214 case RTP_PRIO_NORMAL: 2215 q = &rdd->queues[pri]; 2216 which = &rdd->queuebits; 2217 break; 2218 case RTP_PRIO_REALTIME: 2219 case RTP_PRIO_FIFO: 2220 q = &rdd->rtqueues[pri]; 2221 which = &rdd->rtqueuebits; 2222 break; 2223 case RTP_PRIO_IDLE: 2224 q = &rdd->idqueues[pri]; 2225 which = &rdd->idqueuebits; 2226 break; 2227 default: 2228 panic("remrunqueue: invalid rtprio type"); 2229 /* NOT REACHED */ 2230 } 2231 KKASSERT(lp->lwp_mpflags & LWP_MP_ONRUNQ); 2232 atomic_clear_int(&lp->lwp_mpflags, LWP_MP_ONRUNQ); 2233 TAILQ_REMOVE(q, lp, lwp_procq); 2234 --rdd->runqcount; 2235 if (TAILQ_EMPTY(q)) { 2236 KASSERT((*which & (1 << pri)) != 0, 2237 ("remrunqueue: remove from empty queue")); 2238 *which &= ~(1 << pri); 2239 } 2240 } 2241 2242 /* 2243 * dfly_setrunqueue_locked() 2244 * 2245 * Add a process whos rqtype and rqindex had previously been calculated 2246 * onto the appropriate run queue. Determine if the addition requires 2247 * a reschedule on a cpu and return the cpuid or -1. 2248 * 2249 * NOTE: Lower priorities are better priorities. 2250 * 2251 * NOTE ON ULOAD: This variable specifies the aggregate load on a cpu, the 2252 * sum of the rough lwp_priority for all running and runnable 2253 * processes. Lower priority processes (higher lwp_priority 2254 * values) actually DO count as more load, not less, because 2255 * these are the programs which require the most care with 2256 * regards to cpu selection. 2257 */ 2258 static void 2259 dfly_setrunqueue_locked(dfly_pcpu_t rdd, struct lwp *lp) 2260 { 2261 u_int32_t *which; 2262 struct rq *q; 2263 int pri; 2264 2265 KKASSERT(lp->lwp_qcpu == rdd->cpuid); 2266 2267 spin_lock(&lp->lwp_spin); 2268 if ((lp->lwp_mpflags & LWP_MP_ULOAD) == 0) { 2269 atomic_set_int(&lp->lwp_mpflags, LWP_MP_ULOAD); 2270 atomic_add_long(&rdd->uload, lp->lwp_uload); 2271 atomic_add_int(&rdd->ucount, 1); 2272 } 2273 spin_unlock(&lp->lwp_spin); 2274 2275 pri = lp->lwp_rqindex; 2276 2277 switch(lp->lwp_rqtype) { 2278 case RTP_PRIO_NORMAL: 2279 q = &rdd->queues[pri]; 2280 which = &rdd->queuebits; 2281 break; 2282 case RTP_PRIO_REALTIME: 2283 case RTP_PRIO_FIFO: 2284 q = &rdd->rtqueues[pri]; 2285 which = &rdd->rtqueuebits; 2286 break; 2287 case RTP_PRIO_IDLE: 2288 q = &rdd->idqueues[pri]; 2289 which = &rdd->idqueuebits; 2290 break; 2291 default: 2292 panic("remrunqueue: invalid rtprio type"); 2293 /* NOT REACHED */ 2294 } 2295 2296 /* 2297 * Place us on the selected queue. Determine if we should be 2298 * placed at the head of the queue or at the end. 2299 * 2300 * We are placed at the tail if our round-robin count has expired, 2301 * or is about to expire and the system thinks its a good place to 2302 * round-robin, or there is already a next thread on the queue 2303 * (it might be trying to pick up where it left off and we don't 2304 * want to interfere). 2305 */ 2306 KKASSERT((lp->lwp_mpflags & LWP_MP_ONRUNQ) == 0); 2307 atomic_set_int(&lp->lwp_mpflags, LWP_MP_ONRUNQ); 2308 ++rdd->runqcount; 2309 2310 if (lp->lwp_rrcount >= usched_dfly_rrinterval || 2311 (lp->lwp_rrcount >= usched_dfly_rrinterval / 2 && 2312 (lp->lwp_thread->td_mpflags & TDF_MP_BATCH_DEMARC)) 2313 ) { 2314 /* 2315 * Place on tail 2316 */ 2317 atomic_clear_int(&lp->lwp_thread->td_mpflags, 2318 TDF_MP_BATCH_DEMARC); 2319 lp->lwp_rrcount = 0; 2320 TAILQ_INSERT_TAIL(q, lp, lwp_procq); 2321 } else { 2322 /* 2323 * Retain rrcount and place on head. Count is retained 2324 * even if the queue is empty. 2325 */ 2326 TAILQ_INSERT_HEAD(q, lp, lwp_procq); 2327 } 2328 *which |= 1 << pri; 2329 } 2330 2331 /* 2332 * For SMP systems a user scheduler helper thread is created for each 2333 * cpu and is used to allow one cpu to wakeup another for the purposes of 2334 * scheduling userland threads from setrunqueue(). 2335 * 2336 * UP systems do not need the helper since there is only one cpu. 2337 * 2338 * We can't use the idle thread for this because we might block. 2339 * Additionally, doing things this way allows us to HLT idle cpus 2340 * on MP systems. 2341 */ 2342 static void 2343 dfly_helper_thread(void *dummy) 2344 { 2345 globaldata_t gd; 2346 dfly_pcpu_t dd; 2347 dfly_pcpu_t rdd; 2348 struct lwp *nlp; 2349 cpumask_t mask; 2350 int sleepok; 2351 int cpuid; 2352 2353 gd = mycpu; 2354 cpuid = gd->gd_cpuid; /* doesn't change */ 2355 mask = gd->gd_cpumask; /* doesn't change */ 2356 dd = &dfly_pcpu[cpuid]; 2357 2358 /* 2359 * Initial interlock, make sure all dfly_pcpu[] structures have 2360 * been initialized before proceeding. 2361 */ 2362 lockmgr(&usched_dfly_config_lk, LK_SHARED); 2363 lockmgr(&usched_dfly_config_lk, LK_RELEASE); 2364 2365 /* 2366 * Since we only want to be woken up only when no user processes 2367 * are scheduled on a cpu, run at an ultra low priority. 2368 */ 2369 lwkt_setpri_self(TDPRI_USER_SCHEDULER); 2370 2371 for (;;) { 2372 /* 2373 * We use the LWKT deschedule-interlock trick to avoid racing 2374 * dfly_rdyprocmask. This means we cannot block through to the 2375 * manual lwkt_switch() call we make below. 2376 */ 2377 sleepok = 1; 2378 crit_enter_gd(gd); 2379 tsleep_interlock(dd->helper_thread, 0); 2380 2381 spin_lock(&dd->spin); 2382 if ((dd->flags & DFLY_PCPU_RDYMASK) == 0) { 2383 ATOMIC_CPUMASK_ORMASK(dfly_rdyprocmask, mask); 2384 dd->flags |= DFLY_PCPU_RDYMASK; 2385 } 2386 clear_user_resched(); /* This satisfied the reschedule request */ 2387 #if 0 2388 dd->rrcount = 0; /* Reset the round-robin counter */ 2389 #endif 2390 2391 if (dd->runqcount || dd->uschedcp != NULL) { 2392 /* 2393 * Threads are available. A thread may or may not be 2394 * currently scheduled. Get the best thread already queued 2395 * to this cpu. 2396 */ 2397 nlp = dfly_chooseproc_locked(dd, dd, dd->uschedcp, 0); 2398 if (nlp) { 2399 if ((dd->flags & DFLY_PCPU_CURMASK) == 0) { 2400 ATOMIC_CPUMASK_ORMASK(dfly_curprocmask, mask); 2401 dd->flags |= DFLY_PCPU_CURMASK; 2402 } 2403 dd->upri = nlp->lwp_priority; 2404 dd->uschedcp = nlp; 2405 #if 0 2406 dd->rrcount = 0; /* reset round robin */ 2407 #endif 2408 spin_unlock(&dd->spin); 2409 lwkt_acquire(nlp->lwp_thread); 2410 lwkt_schedule(nlp->lwp_thread); 2411 } else { 2412 /* 2413 * This situation should not occur because we had 2414 * at least one thread available. 2415 */ 2416 spin_unlock(&dd->spin); 2417 } 2418 } else if (usched_dfly_features & 0x01) { 2419 /* 2420 * This cpu is devoid of runnable threads, steal a thread 2421 * from another nearby cpu that is both running something 2422 * and has runnable threads queued. Since we're stealing, 2423 * we might as well load balance at the same time. 2424 * 2425 * We choose the worst thread from the worst queue. This 2426 * can be a bit problematic if the worst queue intends to 2427 * run the thread we choose, 2428 * 2429 * NOTE! This function only returns a non-NULL rdd when 2430 * another cpu's queue is obviously overloaded. We 2431 * do not want to perform the type of rebalancing 2432 * the schedclock does here because it would result 2433 * in insane process pulling when 'steady' state is 2434 * partially unbalanced (e.g. 6 runnables and only 2435 * 4 cores). 2436 */ 2437 rdd = dfly_choose_worst_queue(dd, 0); 2438 if (rdd && dd->uload + usched_dfly_weight7 < rdd->uload) { 2439 if (rdd->uschedcp && spin_trylock(&rdd->spin)) { 2440 nlp = dfly_chooseproc_locked(rdd, dd, NULL, 1); 2441 spin_unlock(&rdd->spin); 2442 } else { 2443 nlp = NULL; 2444 } 2445 } else { 2446 nlp = NULL; 2447 } 2448 if (nlp) { 2449 if ((dd->flags & DFLY_PCPU_CURMASK) == 0) { 2450 ATOMIC_CPUMASK_ORMASK(dfly_curprocmask, mask); 2451 dd->flags |= DFLY_PCPU_CURMASK; 2452 } 2453 dd->upri = nlp->lwp_priority; 2454 dd->uschedcp = nlp; 2455 #if 0 2456 dd->rrcount = 0; /* reset round robin */ 2457 #endif 2458 spin_unlock(&dd->spin); 2459 lwkt_acquire(nlp->lwp_thread); 2460 lwkt_schedule(nlp->lwp_thread); 2461 } else { 2462 /* 2463 * Leave the thread on our run queue. Another 2464 * scheduler will try to pull it later. 2465 */ 2466 spin_unlock(&dd->spin); 2467 } 2468 } else { 2469 /* 2470 * devoid of runnable threads and not allowed to steal 2471 * any. 2472 */ 2473 spin_unlock(&dd->spin); 2474 } 2475 2476 /* 2477 * We're descheduled unless someone scheduled us. Switch away. 2478 * Exiting the critical section will cause splz() to be called 2479 * for us if interrupts and such are pending. 2480 */ 2481 crit_exit_gd(gd); 2482 if (sleepok) { 2483 tsleep(dd->helper_thread, PINTERLOCKED, "schslp", 2484 usched_dfly_poll_ticks); 2485 } 2486 } 2487 } 2488 2489 #if 0 2490 static int 2491 sysctl_usched_dfly_stick_to_level(SYSCTL_HANDLER_ARGS) 2492 { 2493 int error, new_val; 2494 2495 new_val = usched_dfly_stick_to_level; 2496 2497 error = sysctl_handle_int(oidp, &new_val, 0, req); 2498 if (error != 0 || req->newptr == NULL) 2499 return (error); 2500 if (new_val > cpu_topology_levels_number - 1 || new_val < 0) 2501 return (EINVAL); 2502 usched_dfly_stick_to_level = new_val; 2503 return (0); 2504 } 2505 #endif 2506 2507 /* 2508 * Setup the queues and scheduler helpers (scheduler helpers are SMP only). 2509 * Note that curprocmask bit 0 has already been cleared by rqinit() and 2510 * we should not mess with it further. 2511 */ 2512 static void 2513 usched_dfly_cpu_init(void) 2514 { 2515 int i; 2516 int j; 2517 int smt_not_supported = 0; 2518 int cache_coherent_not_supported = 0; 2519 2520 if (bootverbose) 2521 kprintf("Start usched_dfly helpers on cpus:\n"); 2522 2523 sysctl_ctx_init(&usched_dfly_sysctl_ctx); 2524 usched_dfly_sysctl_tree = 2525 SYSCTL_ADD_NODE(&usched_dfly_sysctl_ctx, 2526 SYSCTL_STATIC_CHILDREN(_kern), OID_AUTO, 2527 "usched_dfly", CTLFLAG_RD, 0, ""); 2528 2529 usched_dfly_node_mem = get_highest_node_memory(); 2530 2531 lockmgr(&usched_dfly_config_lk, LK_EXCLUSIVE); 2532 2533 for (i = 0; i < ncpus; ++i) { 2534 dfly_pcpu_t dd = &dfly_pcpu[i]; 2535 cpumask_t mask; 2536 2537 CPUMASK_ASSBIT(mask, i); 2538 if (CPUMASK_TESTMASK(mask, smp_active_mask) == 0) 2539 continue; 2540 2541 spin_init(&dd->spin, "uschedcpuinit"); 2542 dd->cpunode = get_cpu_node_by_cpuid(i); 2543 dd->cpuid = i; 2544 dd->gd = globaldata_find(i); 2545 CPUMASK_ASSBIT(dd->cpumask, i); 2546 for (j = 0; j < NQS; j++) { 2547 TAILQ_INIT(&dd->queues[j]); 2548 TAILQ_INIT(&dd->rtqueues[j]); 2549 TAILQ_INIT(&dd->idqueues[j]); 2550 } 2551 ATOMIC_CPUMASK_NANDBIT(dfly_curprocmask, 0); 2552 if (i == 0) 2553 dd->flags &= ~DFLY_PCPU_CURMASK; 2554 2555 if (dd->cpunode == NULL) { 2556 smt_not_supported = 1; 2557 cache_coherent_not_supported = 1; 2558 if (bootverbose) 2559 kprintf (" cpu%d - WARNING: No CPU NODE " 2560 "found for cpu\n", i); 2561 } else { 2562 switch (dd->cpunode->type) { 2563 case THREAD_LEVEL: 2564 if (bootverbose) 2565 kprintf (" cpu%d - HyperThreading " 2566 "available. Core siblings: ", 2567 i); 2568 break; 2569 case CORE_LEVEL: 2570 smt_not_supported = 1; 2571 2572 if (bootverbose) 2573 kprintf (" cpu%d - No HT available, " 2574 "multi-core/physical " 2575 "cpu. Physical siblings: ", 2576 i); 2577 break; 2578 case CHIP_LEVEL: 2579 smt_not_supported = 1; 2580 2581 if (bootverbose) 2582 kprintf (" cpu%d - No HT available, " 2583 "single-core/physical cpu. " 2584 "Package siblings: ", 2585 i); 2586 break; 2587 default: 2588 /* Let's go for safe defaults here */ 2589 smt_not_supported = 1; 2590 cache_coherent_not_supported = 1; 2591 if (bootverbose) 2592 kprintf (" cpu%d - Unknown cpunode->" 2593 "type=%u. siblings: ", 2594 i, 2595 (u_int)dd->cpunode->type); 2596 break; 2597 } 2598 2599 if (bootverbose) { 2600 if (dd->cpunode->parent_node != NULL) { 2601 kprint_cpuset(&dd->cpunode-> 2602 parent_node->members); 2603 kprintf("\n"); 2604 } else { 2605 kprintf(" no siblings\n"); 2606 } 2607 } 2608 } 2609 2610 lwkt_create(dfly_helper_thread, NULL, &dd->helper_thread, NULL, 2611 0, i, "usched %d", i); 2612 2613 /* 2614 * Allow user scheduling on the target cpu. cpu #0 has already 2615 * been enabled in rqinit(). 2616 */ 2617 if (i) { 2618 ATOMIC_CPUMASK_NANDMASK(dfly_curprocmask, mask); 2619 dd->flags &= ~DFLY_PCPU_CURMASK; 2620 } 2621 if ((dd->flags & DFLY_PCPU_RDYMASK) == 0) { 2622 ATOMIC_CPUMASK_ORMASK(dfly_rdyprocmask, mask); 2623 dd->flags |= DFLY_PCPU_RDYMASK; 2624 } 2625 dd->upri = PRIBASE_NULL; 2626 2627 } 2628 2629 /* usched_dfly sysctl configurable parameters */ 2630 2631 SYSCTL_ADD_INT(&usched_dfly_sysctl_ctx, 2632 SYSCTL_CHILDREN(usched_dfly_sysctl_tree), 2633 OID_AUTO, "rrinterval", CTLFLAG_RW, 2634 &usched_dfly_rrinterval, 0, ""); 2635 SYSCTL_ADD_INT(&usched_dfly_sysctl_ctx, 2636 SYSCTL_CHILDREN(usched_dfly_sysctl_tree), 2637 OID_AUTO, "decay", CTLFLAG_RW, 2638 &usched_dfly_decay, 0, "Extra decay when not running"); 2639 SYSCTL_ADD_INT(&usched_dfly_sysctl_ctx, 2640 SYSCTL_CHILDREN(usched_dfly_sysctl_tree), 2641 OID_AUTO, "ipc_smt", CTLFLAG_RW, 2642 &usched_dfly_ipc_smt, 0, "Pair IPC on hyper-threads"); 2643 SYSCTL_ADD_INT(&usched_dfly_sysctl_ctx, 2644 SYSCTL_CHILDREN(usched_dfly_sysctl_tree), 2645 OID_AUTO, "ipc_same", CTLFLAG_RW, 2646 &usched_dfly_ipc_same, 0, "Pair IPC on same thread"); 2647 SYSCTL_ADD_INT(&usched_dfly_sysctl_ctx, 2648 SYSCTL_CHILDREN(usched_dfly_sysctl_tree), 2649 OID_AUTO, "poll_ticks", CTLFLAG_RW, 2650 &usched_dfly_poll_ticks, 0, "Poll for work (0 ok)"); 2651 2652 /* Add enable/disable option for SMT scheduling if supported */ 2653 if (smt_not_supported) { 2654 usched_dfly_smt = 0; 2655 SYSCTL_ADD_STRING(&usched_dfly_sysctl_ctx, 2656 SYSCTL_CHILDREN(usched_dfly_sysctl_tree), 2657 OID_AUTO, "smt", CTLFLAG_RD, 2658 "NOT SUPPORTED", 0, "SMT NOT SUPPORTED"); 2659 } else { 2660 usched_dfly_smt = 1; 2661 SYSCTL_ADD_INT(&usched_dfly_sysctl_ctx, 2662 SYSCTL_CHILDREN(usched_dfly_sysctl_tree), 2663 OID_AUTO, "smt", CTLFLAG_RW, 2664 &usched_dfly_smt, 0, "Enable SMT scheduling"); 2665 } 2666 2667 /* 2668 * Add enable/disable option for cache coherent scheduling 2669 * if supported 2670 */ 2671 if (cache_coherent_not_supported) { 2672 usched_dfly_cache_coherent = 0; 2673 SYSCTL_ADD_STRING(&usched_dfly_sysctl_ctx, 2674 SYSCTL_CHILDREN(usched_dfly_sysctl_tree), 2675 OID_AUTO, "cache_coherent", CTLFLAG_RD, 2676 "NOT SUPPORTED", 0, 2677 "Cache coherence NOT SUPPORTED"); 2678 } else { 2679 usched_dfly_cache_coherent = 1; 2680 SYSCTL_ADD_INT(&usched_dfly_sysctl_ctx, 2681 SYSCTL_CHILDREN(usched_dfly_sysctl_tree), 2682 OID_AUTO, "cache_coherent", CTLFLAG_RW, 2683 &usched_dfly_cache_coherent, 0, 2684 "Enable/Disable cache coherent scheduling"); 2685 2686 SYSCTL_ADD_INT(&usched_dfly_sysctl_ctx, 2687 SYSCTL_CHILDREN(usched_dfly_sysctl_tree), 2688 OID_AUTO, "weight1", CTLFLAG_RW, 2689 &usched_dfly_weight1, 200, 2690 "Weight selection for current cpu"); 2691 2692 SYSCTL_ADD_INT(&usched_dfly_sysctl_ctx, 2693 SYSCTL_CHILDREN(usched_dfly_sysctl_tree), 2694 OID_AUTO, "weight2", CTLFLAG_RW, 2695 &usched_dfly_weight2, 180, 2696 "Weight selection for wakefrom cpu"); 2697 2698 SYSCTL_ADD_INT(&usched_dfly_sysctl_ctx, 2699 SYSCTL_CHILDREN(usched_dfly_sysctl_tree), 2700 OID_AUTO, "weight3", CTLFLAG_RW, 2701 &usched_dfly_weight3, 40, 2702 "Weight selection for num threads on queue"); 2703 2704 SYSCTL_ADD_INT(&usched_dfly_sysctl_ctx, 2705 SYSCTL_CHILDREN(usched_dfly_sysctl_tree), 2706 OID_AUTO, "weight4", CTLFLAG_RW, 2707 &usched_dfly_weight4, 160, 2708 "Availability of other idle cpus"); 2709 2710 SYSCTL_ADD_INT(&usched_dfly_sysctl_ctx, 2711 SYSCTL_CHILDREN(usched_dfly_sysctl_tree), 2712 OID_AUTO, "weight5", CTLFLAG_RW, 2713 &usched_dfly_weight5, 50, 2714 "Memory attached to node"); 2715 2716 SYSCTL_ADD_INT(&usched_dfly_sysctl_ctx, 2717 SYSCTL_CHILDREN(usched_dfly_sysctl_tree), 2718 OID_AUTO, "weight6", CTLFLAG_RW, 2719 &usched_dfly_weight6, 150, 2720 "Transfer weight Feat 0x04"); 2721 2722 SYSCTL_ADD_INT(&usched_dfly_sysctl_ctx, 2723 SYSCTL_CHILDREN(usched_dfly_sysctl_tree), 2724 OID_AUTO, "weight7", CTLFLAG_RW, 2725 &usched_dfly_weight7, -100, 2726 "Transfer weight Feat 0x01"); 2727 2728 SYSCTL_ADD_INT(&usched_dfly_sysctl_ctx, 2729 SYSCTL_CHILDREN(usched_dfly_sysctl_tree), 2730 OID_AUTO, "fast_resched", CTLFLAG_RW, 2731 &usched_dfly_fast_resched, 0, 2732 "Availability of other idle cpus"); 2733 2734 SYSCTL_ADD_INT(&usched_dfly_sysctl_ctx, 2735 SYSCTL_CHILDREN(usched_dfly_sysctl_tree), 2736 OID_AUTO, "features", CTLFLAG_RW, 2737 &usched_dfly_features, 0x8F, 2738 "Allow pulls into empty queues"); 2739 2740 SYSCTL_ADD_INT(&usched_dfly_sysctl_ctx, 2741 SYSCTL_CHILDREN(usched_dfly_sysctl_tree), 2742 OID_AUTO, "swmask", CTLFLAG_RW, 2743 &usched_dfly_swmask, ~PPQMASK, 2744 "Queue mask to force thread switch"); 2745 2746 #if 0 2747 SYSCTL_ADD_PROC(&usched_dfly_sysctl_ctx, 2748 SYSCTL_CHILDREN(usched_dfly_sysctl_tree), 2749 OID_AUTO, "stick_to_level", 2750 CTLTYPE_INT | CTLFLAG_RW, 2751 NULL, sizeof usched_dfly_stick_to_level, 2752 sysctl_usched_dfly_stick_to_level, "I", 2753 "Stick a process to this level. See sysctl" 2754 "paremter hw.cpu_topology.level_description"); 2755 #endif 2756 } 2757 lockmgr(&usched_dfly_config_lk, LK_RELEASE); 2758 } 2759 2760 SYSINIT(uschedtd, SI_BOOT2_USCHED, SI_ORDER_SECOND, 2761 usched_dfly_cpu_init, NULL); 2762