From 068f0c3ed82349fce5eaeaf89d426c1dd7e39245 Mon Sep 17 00:00:00 2001
From: Kaiwan N Billimoria <kaiwan.billimoria@gmail.com>
Date: Tue, 26 Nov 2024 10:27:51 +0530
Subject: [PATCH] ARM64 requires some tweaks; different header &
 kernel_neon_begin|end() Reported-by: eibisidi

---
 ch5/fp_in_lkm/Makefile    | 2 ++
 ch5/fp_in_lkm/fp_in_lkm.c | 9 ++++++++-
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/ch5/fp_in_lkm/Makefile b/ch5/fp_in_lkm/Makefile
index 496fa51..8fe722c 100644
--- a/ch5/fp_in_lkm/Makefile
+++ b/ch5/fp_in_lkm/Makefile
@@ -100,6 +100,8 @@ ifeq (${MYDEBUG}, y)
 else
   ccflags-y   += -UDEBUG
 endif
+ccflags-remove-y += -mgeneral-regs-only
+
 # We always keep the dynamic debug facility enabled; this allows us to turn
 # dynamically turn on/off debug printk's later... To disable it simply comment
 # out the following line
diff --git a/ch5/fp_in_lkm/fp_in_lkm.c b/ch5/fp_in_lkm/fp_in_lkm.c
index e627c35..da85b2a 100644
--- a/ch5/fp_in_lkm/fp_in_lkm.c
+++ b/ch5/fp_in_lkm/fp_in_lkm.c
@@ -20,8 +20,15 @@
 
 #include <linux/init.h>
 #include <linux/module.h>
-//#include <linux/kernel.h>
+#ifdef CONFIG_X86
 #include <asm/fpu/api.h>
+#endif
+
+#ifdef CONFIG_ARM64
+#include <asm/neon.h>
+#define kernel_fpu_begin kernel_neon_begin
+#define kernel_fpu_end kernel_neon_end
+#endif
 
 MODULE_AUTHOR("<insert your name here>");
 MODULE_DESCRIPTION("LKP2E book:ch5/fp_in_lkm: no performing FP \