diff mbox series

[FFmpeg-devel,2/3] lavu/riscv: add ff_rv_vlen_least()

Message ID 20240510163943.109471-2-remi@remlab.net
State Accepted
Commit 247c5b2b97b1de6645d1cb9f69a21ca0e76341d8
Headers show
Series [FFmpeg-devel,1/3] lavc/vp9dsp: fix indentation | expand

Commit Message

Rémi Denis-Courmont May 10, 2024, 4:39 p.m. UTC 
This inline function checks that the vector length is at least a given
value. With this, most run-time VLEN checks can be optimised away.
---
 libavutil/riscv/cpu.h | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)
diff mbox series

Patch

diff --git a/libavutil/riscv/cpu.h b/libavutil/riscv/cpu.h
index 56035f8556..af1440f626 100644
--- a/libavutil/riscv/cpu.h
+++ b/libavutil/riscv/cpu.h
@@ -22,6 +22,7 @@ 
 #define AVUTIL_RISCV_CPU_H
 
 #include "config.h"
+#include <stdbool.h>
 #include <stddef.h>
 #include "libavutil/cpu.h"
 
@@ -42,4 +43,24 @@  static inline size_t ff_get_rv_vlenb(void)
     return vlenb;
 }
 #endif
+
+/**
+ * Checks that the vector bit-size is at least the given value.
+ * This is potentially undefined behaviour if vectors are not implemented.
+ */
+static inline bool ff_rv_vlen_least(unsigned int bits)
+{
+#ifdef __riscv_v_min_vlen
+    if (bits <= __riscv_min_vlen)
+        return true;
+#else
+    /*
+     * Vector lengths smaller than 128 bits are only possible in embedded cases
+     * and cannot be run-time detected, so we can assume 128 bits at least.
+     */
+    if (bits <= 128)
+        return true;
+#endif
+    return bits <= (8 * ff_get_rv_vlenb());
+}
 #endif