irqbalance/placement.c

/* 
 * Copyright (C) 2006, Intel Corporation
 * 
 * This file is part of irqbalance
 *
 * This program file is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by the
 * Free Software Foundation; version 2 of the License.
 * 
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 * for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program in a file named COPYING; if not, write to the 
 * Free Software Foundation, Inc., 
 * 51 Franklin Street, Fifth Floor, 
 * Boston, MA 02110-1301 USA
 */
#include "config.h"
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <stdint.h>

#include "types.h"
#include "irqbalance.h"


int power_mode;

GList *rebalance_irq_list;

struct obj_placement {
		struct common_obj_data *best;
		struct common_obj_data *least_irqs;
		uint64_t best_cost;
		struct irq_info *info;
};

static void find_best_object(struct common_obj_data *d, void *data)
{
	struct obj_placement *best = (struct obj_placement *)data;
	uint64_t newload;

	newload = d->load;
	if (newload < best->best_cost) {
		best->best = d;
		best->best_cost = newload;
		best->least_irqs = NULL;
	}

	if (newload == best->best_cost) {
		if (g_list_length(d->interrupts) < g_list_length(best->best->interrupts))
			best->least_irqs = d;
	}
}

static void place_irq_in_cache_domain(struct irq_info *info, void *data)
{
	struct package *p = data;
	struct obj_placement place;
	struct common_obj_data *asign;

	if (!info->moved)
		return;

	if (info->level <= BALANCE_PACKAGE)
		return;


	place.info = info;
	place.best = NULL;
	place.least_irqs = NULL;
	place.best_cost = INT_MAX;

	for_each_cache_domain(p->cache_domains, find_best_object, &place);

	asign = place.least_irqs ? place.least_irqs : place.best;

	if (asign) {
		migrate_irq(&p->common.interrupts, &asign->interrupts, info);
		info->assigned_obj = asign;
	}

}
	
static void place_cache_domain(struct common_obj_data *d, void *data __attribute__((unused)))
{
	struct package *package = (struct package *)d;
	if (package->common.interrupts)
		for_each_irq(package->common.interrupts, place_irq_in_cache_domain, package);
}

static void place_core(struct irq_info *info, void *data)
{
	struct cache_domain *c = data;
	struct obj_placement place;
	struct common_obj_data *asign;

	if (!info->moved)
		return;

	if ((info->level <= BALANCE_CACHE) &&
	    (!one_shot_mode))
		return;

	place.info = info;
	place.best = NULL;
	place.least_irqs = NULL;
	place.best_cost = INT_MAX;

	for_each_cpu_core(c->cpu_cores, find_best_object, &place);

	asign = place.least_irqs ? place.least_irqs : place.best;

	if (asign) {
		migrate_irq(&c->common.interrupts, &asign->interrupts, info);
		info->assigned_obj = asign;
		asign->load += info->load;
	}

}

static void place_cores(struct common_obj_data *d, void *data __attribute__((unused)))
{
	struct cache_domain *cache_domain = (struct cache_domain *)d;
	if (cache_domain->common.interrupts)
		for_each_irq(cache_domain->common.interrupts, place_core, cache_domain);
}

static void place_irq_in_package(struct irq_info *info, void *data)
{
	struct obj_placement place;
	struct numa_node *n = data;
	struct common_obj_data *asign;

	if (!info->moved)
		return;

	if (info->level == BALANCE_NONE)
		return;

	place.info = info;
	place.best = NULL;
	place.least_irqs = NULL;
	place.best_cost = INT_MAX;

	for_each_package(n->packages, find_best_object, &place);

	asign = place.least_irqs ? place.least_irqs : place.best;

	if (asign) {
		migrate_irq(&n->common.interrupts, &asign->interrupts, info);
		info->assigned_obj = asign;
		asign->load += info->load;
	}
}

static void place_packages(struct common_obj_data *d, void *data __attribute__((unused)))
{
	struct numa_node *n = (struct numa_node *)d;
	if (n->common.interrupts)
		for_each_irq(n->common.interrupts, place_irq_in_package, n);
}

static void place_irq_in_node(struct irq_info *info, void *data __attribute__((unused)))
{
	struct obj_placement place;
	struct common_obj_data *asign;

	if( info->level == BALANCE_NONE)
		return;

	if (irq_numa_node(info)->common.number != -1) {
		/*
 		 * This irq belongs to a device with a preferred numa node
 		 * put it on that node
 		 */
		migrate_irq(&rebalance_irq_list, &irq_numa_node(info)->common.interrupts, info);
		info->assigned_obj = (struct common_obj_data *)irq_numa_node(info);
		irq_numa_node(info)->common.load += info->load + 1;
		return;
	}

	place.best_cost = INT_MAX;
	place.best = NULL;
	place.least_irqs = NULL;
	place.info = info;

	for_each_numa_node(NULL, find_best_object, &place);

	asign = place.least_irqs ? place.least_irqs : place.best;

	if (asign) {
		migrate_irq(&rebalance_irq_list, &asign->interrupts, info);
		info->assigned_obj = asign;
		asign->load += info->load;
	}
}

static void validate_irq(struct irq_info *info, void *data)
{
	printf("Validating irq %d %p against %p\n", info->irq, info->assigned_obj, data);
	if (info->assigned_obj != data)
		printf("irq %d is wrong, points to %p, should be %p\n",
			info->irq, info->assigned_obj, data);
}

static void validate_object(struct common_obj_data *d, void *data __attribute__((unused)))
{
	if (d->interrupts)
		for_each_irq(d->interrupts, validate_irq, d);
}

static void validate_object_tree_placement()
{
	for_each_package(NULL, validate_object, NULL);	
	for_each_cache_domain(NULL, validate_object, NULL);
	for_each_cpu_core(NULL, validate_object, NULL);
}

void calculate_placement(void)
{
	/* first clear old data */ 
	clear_work_stats();

	sort_irq_list(&rebalance_irq_list);

	for_each_irq(rebalance_irq_list, place_irq_in_node, NULL);
	for_each_numa_node(NULL, place_packages, NULL);
	for_each_package(NULL, place_cache_domain, NULL);
	for_each_cache_domain(NULL, place_cores, NULL);

	if (debug_mode)
		validate_object_tree_placement();
}