mirror of
https://git.ffmpeg.org/ffmpeg.git
synced 2024-12-25 08:42:39 +00:00
lavu/floatdsp: RISC-V V butterflies_float
This commit is contained in:
parent
f4ea45040f
commit
47ce9735cc
@ -33,6 +33,7 @@ void ff_vector_fmul_scalar_rvv(float *dst, const float *src, float mul,
|
|||||||
int len);
|
int len);
|
||||||
void ff_vector_fmul_add_rvv(float *dst, const float *src0, const float *src1,
|
void ff_vector_fmul_add_rvv(float *dst, const float *src0, const float *src1,
|
||||||
const float *src2, int len);
|
const float *src2, int len);
|
||||||
|
void ff_butterflies_float_rvv(float *v1, float *v2, int len);
|
||||||
|
|
||||||
void ff_vector_dmul_rvv(double *dst, const double *src0, const double *src1,
|
void ff_vector_dmul_rvv(double *dst, const double *src0, const double *src1,
|
||||||
int len);
|
int len);
|
||||||
@ -51,6 +52,7 @@ av_cold void ff_float_dsp_init_riscv(AVFloatDSPContext *fdsp)
|
|||||||
fdsp->vector_fmac_scalar = ff_vector_fmac_scalar_rvv;
|
fdsp->vector_fmac_scalar = ff_vector_fmac_scalar_rvv;
|
||||||
fdsp->vector_fmul_scalar = ff_vector_fmul_scalar_rvv;
|
fdsp->vector_fmul_scalar = ff_vector_fmul_scalar_rvv;
|
||||||
fdsp->vector_fmul_add = ff_vector_fmul_add_rvv;
|
fdsp->vector_fmul_add = ff_vector_fmul_add_rvv;
|
||||||
|
fdsp->butterflies_float = ff_butterflies_float_rvv;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (flags & AV_CPU_FLAG_RVV_F64) {
|
if (flags & AV_CPU_FLAG_RVV_F64) {
|
||||||
|
@ -93,6 +93,24 @@ func ff_vector_fmul_add_rvv, zve32f
|
|||||||
ret
|
ret
|
||||||
endfunc
|
endfunc
|
||||||
|
|
||||||
|
// (a0) = (a0) + (a1), (a1) = (a0) - (a1) [0..a2-1]
|
||||||
|
func ff_butterflies_float_rvv, zve32f
|
||||||
|
1:
|
||||||
|
vsetvli t0, a2, e32, m1, ta, ma
|
||||||
|
vle32.v v16, (a0)
|
||||||
|
sub a2, a2, t0
|
||||||
|
vle32.v v24, (a1)
|
||||||
|
vfadd.vv v0, v16, v24
|
||||||
|
vfsub.vv v8, v16, v24
|
||||||
|
vse32.v v0, (a0)
|
||||||
|
sh2add a0, t0, a0
|
||||||
|
vse32.v v8, (a1)
|
||||||
|
sh2add a1, t0, a1
|
||||||
|
bnez a2, 1b
|
||||||
|
|
||||||
|
ret
|
||||||
|
endfunc
|
||||||
|
|
||||||
// (a0) = (a1) * (a2) [0..a3-1]
|
// (a0) = (a1) * (a2) [0..a3-1]
|
||||||
func ff_vector_dmul_rvv, zve64d
|
func ff_vector_dmul_rvv, zve64d
|
||||||
1:
|
1:
|
||||||
|
Loading…
Reference in New Issue
Block a user