1 files changed, 51 insertions, 0 deletions

diff --git a/include/litmus/nvidia_info.h b/include/litmus/nvidia_info.h
new file mode 100644
index 000000000000..8c2a5524512e
--- /dev/null
+++ b/include/litmus/nvidia_info.h
@@ -0,0 +1,51 @@
+#ifndef __LITMUS_NVIDIA_H
+#define __LITMUS_NVIDIA_H
+
+#include <linux/interrupt.h>
+
+
+#include <litmus/litmus_softirq.h>
+
+#define NV_DEVICE_NUM CONFIG_NV_DEVICE_NUM
+
+/* TODO: Make this a function that checks the PCIe bus or maybe proc settings */
+#define num_online_gpus() (NV_DEVICE_NUM)
+
+
+/* Functions used for decoding NVIDIA blobs. */
+
+int init_nvidia_info(void);
+void shutdown_nvidia_info(void);
+
+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);
+
+/* Functions for figuring out the priority of GPU-using tasks */
+
+struct task_struct* get_nv_max_device_owner(u32 target_device_id);
+
+#ifdef CONFIG_LITMUS_SOFTIRQD
+struct task_struct* get_nv_klmirqd_thread(u32 target_device_id);
+#endif
+
+/* 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