aboutsummaryrefslogtreecommitdiffstats
path: root/libavcodec
diff options
context:
space:
mode:
authorRémi Denis-Courmont <remi@remlab.net>2024-05-26 10:18:22 +0300
committerRémi Denis-Courmont <remi@remlab.net>2024-05-29 16:57:02 +0300
commit91b5ea7bb930d44e0ffd12fe0aabfb2838f0e67c (patch)
tree0aaecb07dc69c25247c5820264fb1ad31c652d12 /libavcodec
parentc53d42380d3f57c318fa76329c8fdcb47a05093a (diff)
downloadffmpeg-91b5ea7bb930d44e0ffd12fe0aabfb2838f0e67c.tar.gz
lavc/vp8dsp: R-V V vp8_luma_dc_wht
This is not great as transposition is poorly supported, but it works: vp8_luma_dc_wht_c: 2.5 vp8_luma_dc_wht_rvv_i32: 1.7
Diffstat (limited to 'libavcodec')
-rw-r--r--libavcodec/riscv/vp8dsp_init.c5
-rw-r--r--libavcodec/riscv/vp8dsp_rvv.S56
2 files changed, 61 insertions, 0 deletions
diff --git a/libavcodec/riscv/vp8dsp_init.c b/libavcodec/riscv/vp8dsp_init.c
index a272058801..836237b41c 100644
--- a/libavcodec/riscv/vp8dsp_init.c
+++ b/libavcodec/riscv/vp8dsp_init.c
@@ -26,6 +26,7 @@
#include "libavcodec/vp8dsp.h"
#include "vp8dsp.h"
+void ff_vp8_luma_dc_wht_rvv(int16_t block[4][4][16], int16_t dc[16]);
void ff_vp8_idct_dc_add_rvv(uint8_t *dst, int16_t block[16], ptrdiff_t stride);
void ff_vp8_idct_dc_add4y_rvv(uint8_t *dst, int16_t block[4][16], ptrdiff_t stride);
void ff_vp8_idct_dc_add4uv_rvv(uint8_t *dst, int16_t block[4][16], ptrdiff_t stride);
@@ -124,6 +125,10 @@ av_cold void ff_vp8dsp_init_riscv(VP8DSPContext *c)
int flags = av_get_cpu_flags();
if (flags & AV_CPU_FLAG_RVV_I32 && ff_rv_vlen_least(128)) {
+#if __riscv_xlen >= 64
+ if (flags & AV_CPU_FLAG_RVV_I64)
+ c->vp8_luma_dc_wht = ff_vp8_luma_dc_wht_rvv;
+#endif
c->vp8_idct_dc_add = ff_vp8_idct_dc_add_rvv;
c->vp8_idct_dc_add4y = ff_vp8_idct_dc_add4y_rvv;
if (flags & AV_CPU_FLAG_RVB_ADDR) {
diff --git a/libavcodec/riscv/vp8dsp_rvv.S b/libavcodec/riscv/vp8dsp_rvv.S
index 3b2d8600d6..3360a38cac 100644
--- a/libavcodec/riscv/vp8dsp_rvv.S
+++ b/libavcodec/riscv/vp8dsp_rvv.S
@@ -1,5 +1,6 @@
/*
* Copyright (c) 2024 Institue of Software Chinese Academy of Sciences (ISCAS).
+ * Copyright © 2024 Rémi Denis-Courmont.
*
* This file is part of FFmpeg.
*
@@ -42,6 +43,61 @@
.endif
.endm
+#if __riscv_xlen >= 64
+func ff_vp8_luma_dc_wht_rvv, zve64x
+ vsetivli zero, 1, e64, m1, ta, ma
+ vlseg4e64.v v4, (a1)
+ vsetivli zero, 4, e16, mf2, ta, ma
+ vwadd.vv v1, v5, v6
+ addi t1, sp, -48
+ vwadd.vv v0, v4, v7
+ addi t2, sp, -32
+ vwsub.vv v2, v5, v6
+ addi t3, sp, -16
+ vwsub.vv v3, v4, v7
+ addi sp, sp, -64
+ vsetvli zero, zero, e32, m1, ta, ma
+ vadd.vv v4, v0, v1
+ vadd.vv v5, v3, v2
+ vse32.v v4, (sp)
+ vsub.vv v6, v0, v1
+ vse32.v v5, (t1)
+ vsub.vv v7, v3, v2
+ vse32.v v6, (t2)
+ vse32.v v7, (t3)
+ vlseg4e32.v v4, (sp)
+ vadd.vv v0, v4, v7
+ sd zero, (a1)
+ vadd.vv v1, v5, v6
+ sd zero, 8(a1)
+ vsub.vv v2, v5, v6
+ sd zero, 16(a1)
+ vsub.vv v3, v4, v7
+ sd zero, 24(a1)
+ vadd.vi v0, v0, 3 # rounding mode not supported, do it manually
+ li t0, 4 * 16 * 2
+ vadd.vi v3, v3, 3
+ addi t1, a0, 16 * 2
+ vadd.vv v4, v0, v1
+ addi t2, a0, 16 * 2 * 2
+ vadd.vv v5, v3, v2
+ addi t3, a0, 16 * 2 * 3
+ vsub.vv v6, v0, v1
+ vsub.vv v7, v3, v2
+ vsetvli zero, zero, e16, mf2, ta, ma
+ vnsra.wi v0, v4, 3
+ addi sp, sp, 64
+ vnsra.wi v1, v5, 3
+ vsse16.v v0, (a0), t0
+ vnsra.wi v2, v6, 3
+ vsse16.v v1, (t1), t0
+ vnsra.wi v3, v7, 3
+ vsse16.v v2, (t2), t0
+ vsse16.v v3, (t3), t0
+ ret
+endfunc
+#endif
+
.macro vp8_idct_dc_add
vlse32.v v0, (a0), a2
lh a5, 0(a1)