Cluster assignment of nv klmirqd threads.

1 files changed, 22 insertions, 0 deletions

diff --git a/litmus/sched_cedf.c b/litmus/sched_cedf.c
index 84aafca78cde..35ea1544ce69 100644
--- a/litmus/sched_cedf.c
+++ b/litmus/sched_cedf.c
@@ -1652,6 +1652,25 @@ static void cleanup_cedf(void)
         }
 }
 
+#if defined(CONFIG_LITMUS_NVIDIA) && defined(CONFIG_LITMUS_SOFTIRQD)
+static int cedf_map_gpu_to_cpu(int gpu, int num_gpus)
+{
+        /* TODO: Some sort of smart clustering on the PCIe bus topology */
+        int num_gpu_clusters = num_clusters;
+        unsigned int gpu_cluster_size = num_gpus / num_gpu_clusters;
+        int cpu_cluster = gpu / gpu_cluster_size;
+        int default_cpu = cedf[cpu_cluster].cpus[0]->cpu;  // first CPU in given cluster
+
+        if(num_gpus % num_gpu_clusters != 0) {
+                TRACE("GPU clusters are of non-uniform size!\n");
+        }
+
+        TRACE("CPU %d is default for GPU %d interrupt threads.\n", default_cpu, gpu);
+
+        return default_cpu;
+}
+#endif
+
 static long cedf_activate_plugin(void)
 {
         int i, j, cpu, ccpu, cpu_count;
@@ -1826,6 +1845,9 @@ static struct sched_plugin cedf_plugin __cacheline_aligned_in_smp = {
         .change_prio_pai_tasklet = cedf_change_prio_pai_tasklet,
         .run_tasklets = cedf_run_tasklets,
 #endif
+#if defined(CONFIG_LITMUS_NVIDIA) && defined(CONFIG_LITMUS_SOFTIRQD)
+        .map_gpu_to_cpu = cedf_map_gpu_to_cpu,
+#endif
 };
 
 static struct proc_dir_entry *cluster_file = NULL, *cedf_dir = NULL;