1 files changed, 32 insertions, 0 deletions

diff --git a/include/os/linux/os_sched.c b/include/os/linux/os_sched.c
new file mode 100644
index 0000000..9a25da1
--- /dev/null
+++ b/include/os/linux/os_sched.c
@@ -0,0 +1,32 @@
+/*
+ * 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.
+ */
+
+#include <nvgpu/os_sched.h>
+
+#include <linux/sched.h>
+
+int nvgpu_current_tid(struct gk20a *g)
+{
+        return current->pid;
+}
+
+int nvgpu_current_pid(struct gk20a *g)
+{
+        return current->tgid;
+}
+
+void __nvgpu_print_current(struct gk20a *g, const char *func_name, int line,
+                void *ctx, enum nvgpu_log_type type)
+{
+        __nvgpu_log_msg(g, func_name, line, type, current->comm);
+}