Existe-t-il un moyen efficace d'obtenir le premier élément non nul dans un registre SIMD en utilisant les intrinsèques SIMD?

Question

Comme le titre lit, si un registre SIMD 256 bits est:

0 | 0 | 1 | 0 | 0 | 1 | 0 | 0 |

Comment puis-je obtenir efficacement l'index du premier élément non nul (à savoir l'indice 2 du premier 1)? Le moyen le plus simple est de conserver en mémoire et de vérifier un par un, mais cela peut coûter très cher. Y a-t-il des idées mignonnes pour le faire?

Answer 1

• PCMPEQB/W/D/Q contre un registre des zéros pour obtenir un vecteur d'éléments qui sont tout-1 pour les éléments zéro et tout à zéro pour les éléments nuls.
• PMOVMSKB pour activer le vecteur de tous les-uns ou tous nuls dans un nombre entier masque binaire
• inverti que (opérateur C ~, asm instruction NOT) pour obtenir 1s dans le bitmap pour les éléments qui étaient non nulle
• TZCNT ou BSF cet entier pour trouver le premier bit défini. Méfiez-vous du comportement de BSF si son entrée est entièrement nulle.

---

S'il n'y a qu'une seule possible valeur non nulle, PCMPEQ contre un vecteur de ce que vous n'avez pas besoin d'inverser ultérieurement. Si c'est le cas, pensez à stocker vos données dans un bitmap en premier lieu, pour réduire votre empreinte de cache par un facteur de 8. Ensuite, vous avez juste TZCNT 64 bits morceaux du tableau.

---

Oops, vient de remarquer l'étiquette intrinsèques. Le manuel de référence de l'instruction asm répertorie les intrinsèques C pertinents au bas de chaque entrée, et vous pouvez rechercher le finder intrinsèques d'Intel par asm mnemonic. (Voir le tag tag x86 pour les liens).