Try to fix compilation error

RWKV · Sep 18, 2023 · d36aa94 · d36aa94
1 parent 69306e6
commit d36aa94
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 10 deletions.
diff --git a/rwkv.cpp b/rwkv.cpp
@@ -1,12 +1,4 @@
 #include "rwkv.h"
-
-// Fix build on Linux.
-// https://stackoverflow.com/questions/8518264/where-is-the-declaration-of-cpu-alloc
-#if defined(__linux__)
-#define _GNU_SOURCE
-#include <sched.h>
-#endif
-
 #include "ggml.h"
 #include "ggml-alloc.h"
 

diff --git a/tests/test_ggml_basics.c b/tests/test_ggml_basics.c
@@ -1,4 +1,12 @@
 // Tests that ggml basics work.
+
+// Fix build on Linux.
+// https://stackoverflow.com/questions/8518264/where-is-the-declaration-of-cpu-alloc
+#if defined(__linux__)
+#define _GNU_SOURCE
+#include <sched.h>
+#endif
+
 #include <ggml.h>
 
 #include <stdio.h>
@@ -23,7 +31,7 @@
     }
 
 // Tests simple computation in a single context.
-static void test_computation() {
+static void test_computation(void) {
     struct ggml_init_params params = {
         .mem_size   = 16 * 1024,
         .mem_buffer = NULL,
@@ -62,7 +70,7 @@ static void test_computation() {
 
 // Tests that operations on tensors from different contexts work.
 // RWKV model loading code depends on this behavior.
-static void test_tensors_from_different_contexts() {
+static void test_tensors_from_different_contexts(void) {
     struct ggml_init_params params = {
         .mem_size   = 16 * 1024,
         .mem_buffer = NULL,