|
|
Message-Id: <20240614064934.657405-1-mzh@mzh.io>
Date: Fri, 14 Jun 2024 14:49:34 +0800
From: Meng Zhuo <mzh@....io>
To: musl@...ts.openwall.com
Subject: [PATCH] math: add riscv64 round/roundf
---
Thanks Szabolcs and Markus!
My first thought was the riscv fclass instruction.
---
v4 -> v5:
* Use single cmp+jmp do the check
---
src/math/riscv64/round.c | 22 ++++++++++++++++++++++
src/math/riscv64/roundf.c | 22 ++++++++++++++++++++++
2 files changed, 44 insertions(+)
create mode 100644 src/math/riscv64/round.c
create mode 100644 src/math/riscv64/roundf.c
diff --git a/src/math/riscv64/round.c b/src/math/riscv64/round.c
new file mode 100644
index 00000000..d6d24211
--- /dev/null
+++ b/src/math/riscv64/round.c
@@ -0,0 +1,22 @@
+#include <math.h>
+#include "libm.h"
+
+#if __riscv_flen >= 64
+
+double round(double x)
+{
+ if (((asuint64(x) >> 52) & 0x7ff) >= 52+0x3ff) return x;
+ double tmp;
+ long long n;
+ __asm__ ("fcvt.l.d %0, %1, rmm" : "=r"(n) : "f"(x));
+ __asm__ ("fcvt.d.l %0, %1" : "=f"(tmp) : "r"(n));
+ // the sign bit is only copied to handle round(-0.0)
+ __asm__ ("fsgnj.d %0, %1, %2" : "=f"(x) : "f"(tmp), "f"(x));
+ return x;
+}
+
+#else
+
+#include "../round.c"
+
+#endif
diff --git a/src/math/riscv64/roundf.c b/src/math/riscv64/roundf.c
new file mode 100644
index 00000000..3104dde6
--- /dev/null
+++ b/src/math/riscv64/roundf.c
@@ -0,0 +1,22 @@
+#include <math.h>
+#include "libm.h"
+
+#if __riscv_flen >= 32
+
+float roundf(float x)
+{
+ if (((asuint(x) >> 23) & 0xff) >= 23+0x7f) return x;
+ float tmp;
+ long n;
+ __asm__ ("fcvt.w.s %0, %1, rmm" : "=r"(n) : "f"(x));
+ __asm__ ("fcvt.s.w %0, %1" : "=f"(tmp) : "r"(n));
+ // the sign bit is only copied to handle round(-0.0)
+ __asm__ ("fsgnj.s %0, %1, %2" : "=f"(x) : "f"(tmp), "f"(x));
+ return x;
+}
+
+#else
+
+#include "../roundf.c"
+
+#endif
--
2.39.2
Powered by blists - more mailing lists
Confused about mailing lists and their use? Read about mailing lists on Wikipedia and check out these guidelines on proper formatting of your messages.