make klmirqd work like aux tasks. checkpoint.

this code is untested!

1 files changed, 19 insertions, 16 deletions

diff --git a/include/litmus/nvidia_info.h b/include/litmus/nvidia_info.h
index 97c9577141db..6f354c8b00ac 100644
--- a/include/litmus/nvidia_info.h
+++ b/include/litmus/nvidia_info.h
@@ -6,10 +6,9 @@
 
 #include <litmus/litmus_softirq.h>
 
-
-//#define NV_DEVICE_NUM NR_LITMUS_SOFTIRQD
 #define NV_DEVICE_NUM CONFIG_NV_DEVICE_NUM
-#define NV_MAX_SIMULT_USERS CONFIG_NV_MAX_SIMULT_USERS
+
+/* Functions used for decoding NVIDIA blobs. */
 
 int init_nvidia_info(void);
 void shutdown_nvidia_info(void);
@@ -18,29 +17,33 @@ int is_nvidia_func(void* func_addr);
 
 void dump_nvidia_info(const struct tasklet_struct *t);
 
-
 // Returns the Nvidia device # associated with provided tasklet and work_struct.
 u32 get_tasklet_nv_device_num(const struct tasklet_struct *t);
 u32 get_work_nv_device_num(const struct work_struct *t);
 
 
-int init_nv_device_reg(void);
-//int get_nv_device_id(struct task_struct* owner);
 
-
-int reg_nv_device(int reg_device_id, int register_device, struct task_struct *t);
+/* Functions for figuring out the priority of GPU-using tasks */
 
 struct task_struct* get_nv_max_device_owner(u32 target_device_id);
-//int is_nv_device_owner(u32 target_device_id);
-
-void lock_nv_registry(u32 reg_device_id, unsigned long* flags);
-void unlock_nv_registry(u32 reg_device_id, unsigned long* flags);
 
-#ifdef CONFIG_LITMUS_PAI_SOFTIRQD
-void pai_check_priority_increase(struct task_struct *t, int reg_device_id);
-void pai_check_priority_decrease(struct task_struct *t, int reg_device_id);
+#ifdef CONFIG_LITMUS_SOFTIRQD
+struct task_struct* get_nv_klmirqd_thread(u32 target_device_id);
 #endif
 
-//void increment_nv_int_count(u32 device);
+/* call when the GPU-holding task, t, blocks */
+long enable_gpu_owner(struct task_struct *t);
+
+/* call when the GPU-holding task, t, resumes */
+long disable_gpu_owner(struct task_struct *t);
+
+/* call when the GPU-holding task, t, increases its priority */
+int gpu_owner_increase_priority(struct task_struct *t);
+
+/* call when the GPU-holding task, t, decreases its priority */
+int gpu_owner_decrease_priority(struct task_struct *t);
+
+
+int reg_nv_device(int reg_device_id, int reg_action, struct task_struct *t);
 
 #endif