target/riscv/vector_internals.c

   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 }