在 RISC-V RVV 0.7.1 中屏蔽 CSR SpMV 的单个行

问：

编辑：我已经将我的问题重新表述为更有成效的东西，并将在下面提供答案。这个问题的旧版本仍然在下面。

我正在RVV 0.7.1中为CSR格式实现优化的SpMV内核。在 C 语言中，SpMV 可以这样实现：

for (int i = 0; i < numRows; i++)
    for (int j = rowPtr[i]; j < rowPtr[i+1]; j++)
       y[i] += val[j] * x[colIdx[j]];

矢量化最初很简单，通过加载 ，使用它来执行 的索引加载，加载和乘以两者。在此之后，必须在每一行中执行约简和 （），将其中的所有乘积相加并将它们存储在 的相应索引中。挑战在于从 创建掩码，以便只有属于给定行的元素在每次缩减时处于活动状态。coldIdxxvalv{f}redsum.vsyrowPtr

我最初的想法（应该提供有关原始问题的上下文）是使用 中的索引将 1 放在向量寄存器的相应元素中，然后使用 和 元素移位创建一个包含每个元素所属行的寄存器。然而，这显然是不可行的。rowPtrviota

如何构建所需的蒙版？

老问题

我正在尝试使用 RISC-V 矢量扩展 （0.7.1） 来创建基于索引数组的掩码寄存器。例如，采用以下索引数组 idx array。目标是使用其中存在的索引将 1 放入相应的掩码元素中，如下面的掩码所示：

idx array | 0 2 3 4 4 6 8
elem idx  | 0 1 2 3 4 5 6 7 8
mask      | 1 0 1 1 1 0 1 0 1

起初我错误地查看了 vrgather.vx 指令，但它并没有完全按照我想要的方式进行，因为它填充了所有元素。我希望代码也非常高性能，所以如果可能的话，它主要由向量指令组成，那将是理想的。如果有人能为我指出正确的方向，我将不胜感激。