from Hacker News

Using the most unhinged AVX-512 instruction to make fastest phrase search algo

by cmcollier on 1/23/25, 9:38 PM with 61 comments

  • by iamnotagenius on 1/26/25, 8:06 PM

    Imo the most "unhinged" cpus for AVX-512 are early batches of Alder Lakes which is the only cpu family that has nearly full coverage of all existing avx-512 subsets.
  • by nxobject on 1/26/25, 6:42 PM

    Spoiler if you don’t want to read through the (wonder but many) paragraphs of exposition: the instruction is `vp2intersectq k, zmm, zmm`.
  • by nine_k on 1/26/25, 11:47 PM

    What a post. It should have taken a week just to write it, never mind the amount of time it took to actually come up with all this stuff and overcome all the obstacles mentioned. What a dedication to improving the performance of phrase search.
  • by nemoniac on 1/27/25, 12:20 PM

    Fascinating blog post. Having said that, it may seem like nitpicking but I have to take issue with the point about recursion, which is often far too easily blamed for inefficiency.

    The blog post mentions it as one of the reasons for the inefficiency of the conventional algorithm.

    A glance at the algorithm shows that the recursion in question is a tail call. This means that any overhead can be readily eliminated using a technique known for nearly fifty years already.

    Steele, Guy Lewis (1977). "Debunking the "expensive procedure call" myth or, procedure call implementations considered harmful or, LAMBDA: The Ultimate GOTO". Proceedings of the 1977 annual conference on - ACM '77.

  • by yorwba on 1/27/25, 7:09 PM

    > Why are you merging up to one rare token at the beginning or at the end? Let’s consider that someone searched for C_0 R_1 C_2 C_3. If we don’t do this merge, we would end up searching for C_0, R_1, C_2 C_3, and this is bad. As established, intersecting common tokens is a problem, so it’s way better to search C_0 R_1, C_2 C_3. I learned this the hard way…

    But since R_1 C_2 C_3 is in the index as well, instead of searching for C_0 R_1, C_2 C_3 with a distance of 2, you can instead search for C_0 R_1, R_1 C_2 C_3 with a distance of 1 (overlapping), which hopefully means that the lists to intersect are smaller.

  • by jonstewart on 1/26/25, 8:12 PM

    The most unhinged AVX-512 instruction is GF2P8AFFINEQB.
  • by rkagerer on 1/27/25, 12:19 AM

    Is the first example under The genius idea heading missing entry #3 below?

      mary:
         docs:
           - 0:
               posns: [0, 8]
           - 1:
               posns: [2]
           - 3:
               posns: [1]
  • by ltbarcly3 on 1/27/25, 5:00 PM

    Very cool blog post, but the fastest phrase search would just use a suffix array, which would make any phrase search take double digit nanoseconds.
  • by csense on 1/29/25, 3:43 AM

    I'm trying to follow the example and getting lost.

    > Imagine the scenario where "mary had" occurs in the following positions: [1, 6] and "a" appears in the position [2], so "mary had a" occurs in the positions [2]

    Okay so basically this means "mary had a" ends at token position 2 (assuming "mary" is token position 0), and you're trying to create an efficient algorithm to do this backwards linking process.

    It's not entirely clear from the article what's pre-computed ahead of time (when documents are added / indexed by the system) and what's done on-the-fly (in response to a user's specific search query).

    Based on skimming the article it appears that you're doing this backwards linking process on the fly for a specific phrase the user enters (but I could be wrong about that).

    > allows us to decompose this value

    What value is being decomposed? It is not clear what "this value" refers to.

    > one representing the group and the other the value

    Ctrl+F is telling me that's the first occurrence of the word "group" in this post. What is the "group" being represented?

    > pos = group * 16 + value

    Okay so you're bit-packing two fields into pos. One of the things being packed is called "group," and it seems to be 16 bits. The other thing being packed is "value," and it seems to be 4 bits. So in total "pos" has 20 bits.

    > the maximum document length is 1048576 tokens

    It seems that "pos" simultaneously corresponds to a token position and our two-field bit-packed thing above. I can't figure out how these two things are in one-to-one correspondence.

    I stopped reading there, given my confusion so far it seems unlikely I'll really be able to really understand much of what follows.

    PS: I skimmed the article on roaring bitmaps. Seems they're basically bitmaps where each 1k (8192-bit) chunk has a storage format (dense, sparse, or RLE), and algorithms for quickly doing intersection, union, etc. with various optimized cases chosen roughly by the number of 1's. (Intersecting two dense bitmaps with say ~50% randomly distributed 1's you can't get faster than ANDing together the bit vectors. But if you're, say, intersecting a small sparse bitmap with ~5 1's against a big dense bitmap with ~4k 1's you can iterate over the 1's in the sparse bitmap checking whether each 1 is in the big bitmap.)

    So in my mind I'm basically just blackboxing roaring bitmaps as "java.util.BitSet or vector<bool> with some extra optimizations that kick in if your data has sections where most of the bits are the same".