From 01e6fac4d61fdd7fff5433942ec93fc2ea1e4df1 Mon Sep 17 00:00:00 2001 From: Joshua Bakita Date: Wed, 28 Jun 2023 18:24:25 -0400 Subject: Include nvgpu headers These are needed to build on NVIDIA's Jetson boards for the time being. Only a couple structs are required, so it should be fairly easy to remove this dependency at some point in the future. --- include/os/linux/thread.c | 70 +++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 70 insertions(+) create mode 100644 include/os/linux/thread.c (limited to 'include/os/linux/thread.c') diff --git a/include/os/linux/thread.c b/include/os/linux/thread.c new file mode 100644 index 0000000..c56bff6 --- /dev/null +++ b/include/os/linux/thread.c @@ -0,0 +1,70 @@ +/* + * Copyright (c) 2017-2018, NVIDIA CORPORATION. All rights reserved. + * + * This program is free software; you can redistribute it and/or modify it + * under the terms and conditions of the GNU General Public License, + * version 2, as published by the Free Software Foundation. + * + * This program is distributed in the hope it will be useful, but WITHOUT + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or + * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for + * more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see . + */ + +#include + +#include +#include + +int nvgpu_thread_proxy(void *threaddata) +{ + struct nvgpu_thread *thread = threaddata; + int ret = thread->fn(thread->data); + + thread->running = false; + return ret; +} + +int nvgpu_thread_create(struct nvgpu_thread *thread, + void *data, + int (*threadfn)(void *data), const char *name) +{ + struct task_struct *task = kthread_create(nvgpu_thread_proxy, + thread, name); + if (IS_ERR(task)) + return PTR_ERR(task); + + thread->task = task; + thread->fn = threadfn; + thread->data = data; + thread->running = true; + wake_up_process(task); + return 0; +}; + +void nvgpu_thread_stop(struct nvgpu_thread *thread) +{ + if (thread->task) { + kthread_stop(thread->task); + thread->task = NULL; + } +}; + +bool nvgpu_thread_should_stop(struct nvgpu_thread *thread) +{ + return kthread_should_stop(); +}; + +bool nvgpu_thread_is_running(struct nvgpu_thread *thread) +{ + return ACCESS_ONCE(thread->running); +}; + +void nvgpu_thread_join(struct nvgpu_thread *thread) +{ + while (ACCESS_ONCE(thread->running)) + nvgpu_msleep(10); +}; -- cgit v1.2.2