From 42dd1f1cf13d02381c10461a7bcfcddc63ad211d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?R=C3=A9mi=20Denis-Courmont?= <remi@remlab.net>
Date: Sat, 16 Nov 2024 09:19:12 +0200
Subject: [PATCH] tests/cpu: print the RISC-V Vector length

---
 libavutil/tests/cpu.c | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/libavutil/tests/cpu.c b/libavutil/tests/cpu.c
index 9e45527408..1eb3055ff0 100644
--- a/libavutil/tests/cpu.c
+++ b/libavutil/tests/cpu.c
@@ -25,6 +25,8 @@
 
 #if ARCH_AARCH64
 #include "libavutil/aarch64/cpu.h"
+#elif ARCH_RISCV
+#include "libavutil/riscv/cpu.h"
 #endif
 
 #if HAVE_UNISTD_H
@@ -170,6 +172,12 @@ int main(int argc, char **argv)
 #if ARCH_AARCH64 && HAVE_SVE
     if (cpu_flags_raw & AV_CPU_FLAG_SVE)
         printf("sve_vector_length = %d\n", 8 * ff_aarch64_sve_length());
+#elif ARCH_RISCV && HAVE_RVV
+    if (cpu_flags_raw & AV_CPU_FLAG_RVV_I32) {
+        size_t bytes = ff_get_rv_vlenb();
+
+        printf("rv_vlenb = %zu (%zu bits)\n", bytes, 8 * bytes);
+    }
 #endif
 
     return 0;