Question

对于uni赋值，我需要编写一个函数来计算汇编中字符串中的空格数（由指针和索引定义）。需要使用pcmpeqb（即使用SSE寄存器），以及使用popcnt和pmovmskb的提示。我的基本方法是以16字节块的形式处理字符串，将每个块加载到%xmm8并将其与初始化为包含16个空格的%xmm9进行比较。但是，我需要特别以某种方式处理最后一个块。

我的第一个想法是使用旋转指令删除字符串末尾的垃圾。（该字符串保证在结束后分配一些额外的空间以防止段错误，但那里的数据可能不应该用于比较。）我偶然发现PSRLDQ但它似乎不接受非立即论证。（或者至少拒绝了我扔的东西。）所以我的问题是：我如何删除SSE寄存器的最后X个字节，而不是将其中的一半归零，或者逐字逐句地删除？（据我所知，大多数可用的操作都是这样做的。）

我的代码（modulo样板文件）目前看起来像这样 - 有问题的位在标签_last:后面的末尾：

    # === Arguments ===
    # %rdi - char *input
    # %rsi - size_t count
    # === Temporaries ===
    # %rdx - how many chars to process in final run
    # %rcx - how many characters were "read" already
    # %r8 - pop count of last iteration
    # %r9
    # %r11
    # === SSE Temporaries ===
    # %xmm8 - the chunk of the string being processed
    # %xmm9 - 16 spaces

    xor %rcx, %rcx
    xor %rax, %rax
    movdqu _spaces(%rip), %xmm9

_loop:
    # set %rdx to number of characters left to process
    mov %rsi, %rdx
    sub %rcx, %rdx

    # we've reached the end of the string
    cmp %rdx, %rsi
    jge _end

    movdqu (%rdi, %rcx), %xmm8 # load chunk of string to process
    add $16, %rcx

    # less than 16 characters to process
    cmp $16, %rdx
    jg _last

_compare: #compare %xmm8 with spaces and add count of spaces to %eax
    pcmpeqb %xmm9, %xmm8
    pmovmskb %xmm8, %r8d
    popcntl %r8d, %r8d
    add %r8d, %eax
    jmp _loop

_last: # last part of string, less than 16 chars
    sub $16, %rdx
    neg %rdx
    # I need to delete possible garbage after the last chars
    psrldq %edx, %xmm8 
    jmp _compare

_end:
    ret

（控制流程可能还有错误，但我稍后会处理。）

Answer 1

不要试图“删除”SSE寄存器中的额外字节。相反，在比较并执行PMOVMSKB之后，只需屏蔽结果掩码中与额外字节对应的位。这是矢量化中非常标准的方法;而不是跳过箍来获得你想要的数据，处理所有的东西，然后清理你以后不想要的东西。

Answer 2

我认为最简单的解决方案是在SSE寄存器中使用全部16个字符，但屏蔽pmovmskb之后的位。请注意，像您一样使用16字节加载是不安全的，因为您可能会跨越到无法访问的页面。

如何在SSE寄存器末尾“删除”字节？

2 个答案: