Skip to content

Commit

Permalink
RVV: ShuffleChannel, fp16& bf16, make C906 happy
Browse files Browse the repository at this point in the history
  • Loading branch information
thelastlin committed Aug 29, 2023
1 parent b03cd19 commit 33a5741
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/layer/riscv/shufflechannel_riscv.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -988,13 +988,13 @@ int ShuffleChannel_riscv::forward_bf16s_fp16s(const Mat& bottom_blob, Mat& top_b
vbool2_t _mask = vlm_v_b2(bitmask, vl);
vuint16m8_t _idx_init = viota_m_u16m8(_mask, vl);
vuint16m8_t _idx = vadd_vx_u16m8(_idx_init, (_group - 1) * elempack, vl);
#endif
for (int shift = _group - 2; shift >= 0; shift--)
{
vuint16m8_t _idx_lower = vadd_vx_u16m8(_idx_init, shift * elempack, vl);
_idx = vslideup_vx_u16m8(vundefined_u16m8(), _idx, 1, vl);
_idx = vmerge_vvm_u16m8(_mask, _idx, _idx_lower, vl);
}
#endif

for (int q = 0; q < channels_per_group; q++)
{
Expand Down

0 comments on commit 33a5741

Please sign in to comment.