diff options
Diffstat (limited to 'drivers/gpu/nvgpu/os/linux/thread.c')
-rw-r--r-- | drivers/gpu/nvgpu/os/linux/thread.c | 63 |
1 files changed, 63 insertions, 0 deletions
diff --git a/drivers/gpu/nvgpu/os/linux/thread.c b/drivers/gpu/nvgpu/os/linux/thread.c new file mode 100644 index 00000000..92c556f2 --- /dev/null +++ b/drivers/gpu/nvgpu/os/linux/thread.c | |||
@@ -0,0 +1,63 @@ | |||
1 | /* | ||
2 | * Copyright (c) 2017, NVIDIA CORPORATION. All rights reserved. | ||
3 | * | ||
4 | * This program is free software; you can redistribute it and/or modify it | ||
5 | * under the terms and conditions of the GNU General Public License, | ||
6 | * version 2, as published by the Free Software Foundation. | ||
7 | * | ||
8 | * This program is distributed in the hope it will be useful, but WITHOUT | ||
9 | * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or | ||
10 | * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for | ||
11 | * more details. | ||
12 | * | ||
13 | * You should have received a copy of the GNU General Public License | ||
14 | * along with this program. If not, see <http://www.gnu.org/licenses/>. | ||
15 | */ | ||
16 | |||
17 | #include <linux/kthread.h> | ||
18 | |||
19 | #include <nvgpu/thread.h> | ||
20 | |||
21 | int nvgpu_thread_proxy(void *threaddata) | ||
22 | { | ||
23 | struct nvgpu_thread *thread = threaddata; | ||
24 | int ret = thread->fn(thread->data); | ||
25 | |||
26 | thread->running = false; | ||
27 | return ret; | ||
28 | } | ||
29 | |||
30 | int nvgpu_thread_create(struct nvgpu_thread *thread, | ||
31 | void *data, | ||
32 | int (*threadfn)(void *data), const char *name) | ||
33 | { | ||
34 | struct task_struct *task = kthread_create(nvgpu_thread_proxy, | ||
35 | thread, name); | ||
36 | if (IS_ERR(task)) | ||
37 | return PTR_ERR(task); | ||
38 | |||
39 | thread->task = task; | ||
40 | thread->fn = threadfn; | ||
41 | thread->data = data; | ||
42 | thread->running = true; | ||
43 | wake_up_process(task); | ||
44 | return 0; | ||
45 | }; | ||
46 | |||
47 | void nvgpu_thread_stop(struct nvgpu_thread *thread) | ||
48 | { | ||
49 | if (thread->task) { | ||
50 | kthread_stop(thread->task); | ||
51 | thread->task = NULL; | ||
52 | } | ||
53 | }; | ||
54 | |||
55 | bool nvgpu_thread_should_stop(struct nvgpu_thread *thread) | ||
56 | { | ||
57 | return kthread_should_stop(); | ||
58 | }; | ||
59 | |||
60 | bool nvgpu_thread_is_running(struct nvgpu_thread *thread) | ||
61 | { | ||
62 | return ACCESS_ONCE(thread->running); | ||
63 | }; | ||