Cluster assignment of nv klmirqd threads.

1 files changed, 8 insertions, 0 deletions

diff --git a/include/litmus/sched_plugin.h b/include/litmus/sched_plugin.h
index a13d1a2992fe..cfa218504d75 100644
--- a/include/litmus/sched_plugin.h
+++ b/include/litmus/sched_plugin.h
@@ -102,6 +102,10 @@ typedef int (*__higher_prio_t)(struct task_struct* a, comparison_mode_t a_mod,
                                                           struct task_struct* b, comparison_mode_t b_mod);
 #endif
 
+#if defined(CONFIG_LITMUS_NVIDIA) && defined(CONFIG_LITMUS_SOFTIRQD)
+typedef int (*default_cpu_for_gpu_t)(int gpu, int num_gpus);
+#endif
+
 
 /********************* sys call backends  ********************/
 /* This function causes the caller to sleep until the next release */
@@ -165,6 +169,10 @@ struct sched_plugin {
         change_prio_pai_tasklet_t       change_prio_pai_tasklet;
         run_tasklets_t                          run_tasklets;
 #endif
+
+#if defined(CONFIG_LITMUS_NVIDIA) && defined(CONFIG_LITMUS_SOFTIRQD)
+        default_cpu_for_gpu_t           map_gpu_to_cpu;
+#endif
 } __attribute__ ((__aligned__(SMP_CACHE_BYTES)));