From fc95237cf08975983634fd6d3ad85634f80a8a55 Mon Sep 17 00:00:00 2001 From: Terje Bergstrom Date: Tue, 25 Apr 2017 09:00:47 -0700 Subject: gpu: nvgpu: Abstract threads Add abstraction for threads. Implement them using kthreads in Linux kernel. JIRA NVGPU-14 Change-Id: I26996d844db6bdc4b5e6f3230514a1dadb3fb07c Signed-off-by: Terje Bergstrom Reviewed-on: http://git-master/r/1469636 Reviewed-by: mobile promotions Tested-by: mobile promotions --- drivers/gpu/nvgpu/Makefile.nvgpu | 1 + drivers/gpu/nvgpu/common/linux/thread.c | 42 +++++++++++++++++ drivers/gpu/nvgpu/include/nvgpu/linux/thread.h | 26 +++++++++++ drivers/gpu/nvgpu/include/nvgpu/thread.h | 62 ++++++++++++++++++++++++++ 4 files changed, 131 insertions(+) create mode 100644 drivers/gpu/nvgpu/common/linux/thread.c create mode 100644 drivers/gpu/nvgpu/include/nvgpu/linux/thread.h create mode 100644 drivers/gpu/nvgpu/include/nvgpu/thread.h diff --git a/drivers/gpu/nvgpu/Makefile.nvgpu b/drivers/gpu/nvgpu/Makefile.nvgpu index 8ed7f125..08d939f0 100644 --- a/drivers/gpu/nvgpu/Makefile.nvgpu +++ b/drivers/gpu/nvgpu/Makefile.nvgpu @@ -37,6 +37,7 @@ nvgpu-y := \ common/linux/soc.o \ common/linux/driver_common.o \ common/linux/firmware.o \ + common/linux/thread.o \ common/mm/nvgpu_allocator.o \ common/mm/bitmap_allocator.o \ common/mm/buddy_allocator.o \ diff --git a/drivers/gpu/nvgpu/common/linux/thread.c b/drivers/gpu/nvgpu/common/linux/thread.c new file mode 100644 index 00000000..d37b2a10 --- /dev/null +++ b/drivers/gpu/nvgpu/common/linux/thread.c @@ -0,0 +1,42 @@ +/* + * Copyright (c) 2017, 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 + +int nvgpu_thread_create(struct nvgpu_thread *thread, + void *data, + int (*threadfn)(void *data), const char *name) +{ + struct task_struct *task = kthread_create(threadfn, data, name); + if (IS_ERR(task)) + return PTR_ERR(task); + + thread->task = task; + wake_up_process(task); + return 0; +}; + +void nvgpu_thread_stop(struct nvgpu_thread *thread) +{ + kthread_stop(thread->task); +}; + +bool nvgpu_thread_should_stop(struct nvgpu_thread *thread) +{ + return kthread_should_stop(); +}; diff --git a/drivers/gpu/nvgpu/include/nvgpu/linux/thread.h b/drivers/gpu/nvgpu/include/nvgpu/linux/thread.h new file mode 100644 index 00000000..13f29515 --- /dev/null +++ b/drivers/gpu/nvgpu/include/nvgpu/linux/thread.h @@ -0,0 +1,26 @@ +/* + * Copyright (c) 2017, 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 . + */ + +#ifndef __NVGPU_THREAD_LINUX_H__ +#define __NVGPU_THREAD_LINUX_H__ + +struct task_struct; + +struct nvgpu_thread { + struct task_struct *task; +}; + +#endif /* __NVGPU_THREAD_LINUX_H__ */ diff --git a/drivers/gpu/nvgpu/include/nvgpu/thread.h b/drivers/gpu/nvgpu/include/nvgpu/thread.h new file mode 100644 index 00000000..81dac7ca --- /dev/null +++ b/drivers/gpu/nvgpu/include/nvgpu/thread.h @@ -0,0 +1,62 @@ +/* + * Copyright (c) 2017, 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 . + */ + +#ifndef __NVGPU_THREAD_H__ +#define __NVGPU_THREAD_H__ + +#ifdef __KERNEL__ +#include +#endif + +/** + * nvgpu_thread_create - Create and run a new thread. + * + * @thread - thread structure to use + * @data - data to pass to threadfn + * @threadfn - Thread function + * @name - name of the thread + * + * Create a thread and run threadfn in it. The thread stays alive as long as + * threadfn is running. As soon as threadfn returns the thread is destroyed. + * + * threadfn needs to continuously poll nvgpu_thread_should_stop() to determine + * if it should exit. + */ +int nvgpu_thread_create(struct nvgpu_thread *thread, + void *data, + int (*threadfn)(void *data), const char *name); + +/** + * nvgpu_thread_stop - Destroy or request to destroy a thread + * + * @thread - thread to stop + * + * Request a thread to stop by setting nvgpu_thread_should_stop() to + * true and wait for thread to exit. + */ +void nvgpu_thread_stop(struct nvgpu_thread *thread); + +/** + * nvgpu_thread_should_stop - Query if thread should stop + * + * @thread + * + * Return true if thread should exit. Can be run only in the thread's own + * context and with the thread as parameter. + */ +bool nvgpu_thread_should_stop(struct nvgpu_thread *thread); + +#endif /* __NVGPU_THREAD_H__ */ -- cgit v1.2.2