1 /* 2 * RISC-V Vector Extension Internals 3 * 4 * Copyright (c) 2020 T-Head Semiconductor Co., Ltd. All rights reserved. 5 * 6 * This program is free software; you can redistribute it and/or modify it 7 * under the terms and conditions of the GNU General Public License, 8 * version 2 or later, as published by the Free Software Foundation. 9 * 10 * This program is distributed in the hope it will be useful, but WITHOUT 11 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 12 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for 13 * more details. 14 * 15 * You should have received a copy of the GNU General Public License along with 16 * this program. If not, see <http://www.gnu.org/licenses/>. 17 */ 18 19 #include "vector_internals.h" 20 21 /* set agnostic elements to 1s */ 22 void vext_set_elems_1s(void *base, uint32_t is_agnostic, uint32_t cnt, 23 uint32_t tot) 24 { 25 if (is_agnostic == 0) { 26 /* policy undisturbed */ 27 return; 28 } 29 if (tot - cnt == 0) { 30 return ; 31 } 32 memset(base + cnt, -1, tot - cnt); 33 } 34 35 void do_vext_vv(void *vd, void *v0, void *vs1, void *vs2, 36 CPURISCVState *env, uint32_t desc, 37 opivv2_fn *fn, uint32_t esz) 38 { 39 uint32_t vm = vext_vm(desc); 40 uint32_t vl = env->vl; 41 uint32_t total_elems = vext_get_total_elems(env, desc, esz); 42 uint32_t vta = vext_vta(desc); 43 uint32_t vma = vext_vma(desc); 44 uint32_t i; 45 46 for (i = env->vstart; i < vl; i++) { 47 if (!vm && !vext_elem_mask(v0, i)) { 48 /* set masked-off elements to 1s */ 49 vext_set_elems_1s(vd, vma, i * esz, (i + 1) * esz); 50 continue; 51 } 52 fn(vd, vs1, vs2, i); 53 } 54 env->vstart = 0; 55 /* set tail elements to 1s */ 56 vext_set_elems_1s(vd, vta, vl * esz, total_elems * esz); 57 } 58 59 void do_vext_vx(void *vd, void *v0, target_long s1, void *vs2, 60 CPURISCVState *env, uint32_t desc, 61 opivx2_fn fn, uint32_t esz) 62 { 63 uint32_t vm = vext_vm(desc); 64 uint32_t vl = env->vl; 65 uint32_t total_elems = vext_get_total_elems(env, desc, esz); 66 uint32_t vta = vext_vta(desc); 67 uint32_t vma = vext_vma(desc); 68 uint32_t i; 69 70 for (i = env->vstart; i < vl; i++) { 71 if (!vm && !vext_elem_mask(v0, i)) { 72 /* set masked-off elements to 1s */ 73 vext_set_elems_1s(vd, vma, i * esz, (i + 1) * esz); 74 continue; 75 } 76 fn(vd, s1, vs2, i); 77 } 78 env->vstart = 0; 79 /* set tail elements to 1s */ 80 vext_set_elems_1s(vd, vta, vl * esz, total_elems * esz); 81 } 82