华为计算微信公众号
昇腾AI开发者公众号
华为计算微博
华为计算今日头条
两个向量Compare之后的mask是连续排布的, 我现在想用GatherMask根据mask从原向量中提取对应位置的数, 因为源数据是int32的, 一次repeat只能处理64个数, 消耗64bit的mask, 只有8bytes, 不够一个block, 所以repeat stride不知道该怎么设置, 有没有更好的实现方式?
我要发帖子
两个向量Compare之后的mask是连续排布的, 我现在想用GatherMask根据mask从原向量中提取对应位置的数, 因为源数据是int32的, 一次repeat只能处理64个数, 消耗64bit的mask, 只有8bytes, 不够一个block, 所以repeat stride不知道该怎么设置, 有没有更好的实现方式?