Auto-CPU-affinity from part./cluster assignment.

This patch replaces be_migrate_to() with several
be_migrate_*() APIs to automatically assign CPU affinity
masks from a task's partition (or cluster) assignment.
Routines are release-master-aware such that the release
master (if one exists) will NOT be included in the
task's affinity mask. (Note that release-master
avoidance may be overridden by calling
__be_migrate_thread_to_cluster().)

2 files changed, 25 insertions, 7 deletions

diff --git a/include/litmus.h b/include/litmus.h
index 58af6b7..b2e81f2 100644
--- a/include/litmus.h
+++ b/include/litmus.h
@@ -16,13 +16,7 @@ extern "C" {
 
 #include "asm/cycles.h" /* for null_call() */
 
-typedef int pid_t;       /* PID of a task */
-
-/* obtain the PID of a thread */
-pid_t gettid(void);
-
-/* migrate to partition */
-int be_migrate_to(int target_cpu);
+#include "migration.h"
 
 int set_rt_task_param(pid_t pid, struct rt_task* param);
 int get_rt_task_param(pid_t pid, struct rt_task* param);
diff --git a/include/migration.h b/include/migration.h
new file mode 100644
index 0000000..2413e7c
--- /dev/null
+++ b/include/migration.h
@@ -0,0 +1,24 @@
+
+typedef int pid_t;
+
+/* obtain the PID of a thread */
+pid_t gettid();
+
+/* Assign a task to a cpu/partition/cluster.
+ * PRECOND: tid is not yet in real-time mode (it's a best effort task).
+ * Set tid == 0 to migrate the caller */
+int be_migrate_thread_to_cpu(pid_t tid, int target_cpu);
+int be_migrate_thread_to_partition(pid_t tid, int partition);
+/* If using release master, set cluster_sz to size of largest cluster. tid
+ * will not be scheduled on release master. */
+int be_migrate_thread_to_cluster(pid_t tid, int cluster, int cluster_sz);
+
+/* set ignore_rm == 1 to include release master in tid's cpu affinity */
+int __be_migrate_thread_to_cluster(pid_t tid, int cluster, int cluster_sz, int ignore_rm);
+
+int be_migrate_to_cpu(int target_cpu);
+int be_migrate_to_partition(int partition);
+int be_migrate_to_cluster(int cluster, int cluster_sz);
+
+int num_online_cpus();
+int release_master();