1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
#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);
// 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_and_lock_nvklmirqd_thread(u32 target_device_id, unsigned long* flags);
void unlock_nvklmirqd_thread(u32 target_device_id, unsigned long* flags);
struct task_struct* get_nvklmirqd_thread(u32 target_device_id);
#if defined(CONFIG_LITMUS_NVIDIA_WORKQ_ON) || defined(CONFIG_LITMUS_NVIDIA_WORKQ_ON_DEDICATED)
struct task_struct* get_and_lock_nvklmworkqd_thread(u32 target_device_id, unsigned long* flags);
void unlock_nvklmworkqd_thread(u32 target_device_id, unsigned long* flags);
struct task_struct* get_nvklmworkqd_thread(u32 target_device_id);
#endif
#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
|
