‹ BackHN Continuity

Thread

Fearless SIMD v1.0

318 points · 52 comments · verdagon

  1. modulovalue · · focus · HN ↗
    I’m currently working on adding better SIMD support to Dart (<a href="https:&#x2F;&#x2F;github.com&#x2F;dart-lang&#x2F;sdk&#x2F;issues&#x2F;64170" rel="nofollow">https:&#x2F;&#x2F;github.com&#x2F;dart-lang&#x2F;sdk&#x2F;issues&#x2F;64170) and I have a question for the author or others here.

    Does Rust or any other language support customizing the compiler so that interprocedural analyses can track custom subsets of, for example, doubles so that the compiler can choose the most efficient instruction sequence for example for min&#x2F;max? If we know a double is never NaN then we can emit only one instruction on x86, but have to emit one more on arm64. If we know a double is never zero and never NaN, we can emit a single instruction on both.

    This whole conversation between relaxed SIMD and deterministic SIMD seems to only exist because our compilers are not smart enough and&#x2F;or their whole program analyses don’t support any plugin-like capabilities.

    There are other examples where if we know a SIMD bitmask is canonical (all 1s per lane) then we can implement horizontal reductions more efficiently. This is very niche and I doubt that any language supports interprocedural analyses with such a rich domain, so it feels like a hole in the programming language space.

    1. athrowaway3z · · focus · HN ↗
      Last time i checked; no.

      But I suspect you&#x27;re overvaluing the potential savings. Knowing when a float is 0.0 or NaN beforehand is almost entirely impossible, except for the most trivial of cases - like when you first initialize a variable or first enter a loop. Everything after that is very hard or impossible with floats as they are.

      Those cases can be const folded at compile time.

      Those cases are never a measurable bottleneck.

      The closest thing I know of in the realm of the optimization you&#x27;re curious about is Rust NonZero* variants, but they&#x27;re used for enum compression afaik.

      1. modulovalue · · focus · HN ↗
        I’m not sure I agree on the impossible part, I feel like a sufficiently smart interprocedural analysis that also implements range analysis interprocedurally could prove a lot to where it becomes useful.

        I guess what I would like to see is SIMD libraries being able to confidently say nobody needs to use intrinsics (or differentiate between relaxed&#x2F;normal SIMD on the user API level) because the language + high level SIMD APIs are smart enough to choose the right implementation.

        IIRC IEEE min&#x2F;max with proper NaN handling needs 8 instructions on x86 vs 1 on arm64 I find it very sad that we apparently haven’t really solved that yet without forcing the user to use different APIs.

        1. orlp · · focus · HN ↗
          You can do it with just 3 instructions for IEEE 754-2019 minimumNumber (ignores NaN):

                  vminpd          ymm2, ymm1, ymm0
                  vcmpunordpd     ymm0, ymm0, ymm0
                  vblendvpd       ymm0, ymm2, ymm1, ymm0
          
          If you want proper IEEE 754-2019 minimum (propagate NaN, -0.0 &lt; +0.0, NaN bitpattern picked in the usual way) you can do it in 6:

                  vminpd          ymm1, ymm0, ymm1
                  vbroadcastsd    ymm2, qword ptr [rip + .LCPI0_0]
                  vandpd          ymm2, ymm0, ymm2
                  vorpd           ymm1, ymm2, ymm1
                  vcmpunordpd     ymm2, ymm0, ymm0
                  vblendvpd       ymm0, ymm1, ymm0, ymm2
          
          I personally find this a load of nonsense I don&#x27;t care about.

          If you want propagating NaNs but don&#x27;t care about signed zero or NaN payload&#x2F;sign, you can use

                  vminpd  ymm2, ymm0, ymm1
                  vminpd  ymm1, ymm1, ymm0
                  vorpd   ymm0, ymm1, ymm2
          
          What I do in Polars is a bit different, there for propagating NaNs I do

              if (self &lt; other) | self.is_nan() { self } else { other}
          
          this isn&#x27;t fully optimal on x86-64 but it&#x27;s fairly simple and autovectorizes decently on various platforms, here&#x27;s AVX2:

                  vcmpltpd        ymm2, ymm0, ymm1
                  vcmpunordpd     ymm3, ymm0, ymm0
                  vorpd           ymm2, ymm3, ymm2
                  vblendvpd       ymm0, ymm1, ymm0, ymm2
          1. dzaima · · focus · HN ↗
            That vminpd+vminpd+vorpd actually does handle signed zero properly! Screws up NaN payloads to the max though. Can be easily extended to canonicalize the NaN with 2 instrs + constant though of course. (which ends up at the same number of instrs as your proper impl (albeit with worse port distribution and latency), but you get to have a canonical NaN!)

            Hit upon <a href="https:&#x2F;&#x2F;github.com&#x2F;llvm&#x2F;llvm-project&#x2F;issues&#x2F;217376" rel="nofollow">https:&#x2F;&#x2F;github.com&#x2F;llvm&#x2F;llvm-project&#x2F;issues&#x2F;217376 while playing around with proper minimumnum, failing to SMT-verify whatever version of LLVM I had; did find a funky working 6-instr (+ constant) version though:

                vpandn      ymm2, ymm1, ymm0
                vpcmpeqd    ymm2, ymm2, 0x80000000 # whether ymm0 is -0 and ymm1 is +0 (or other cases that magically don&#x27;t cause issues)
                vcmpltpd    ymm3, ymm0, ymm1
                vcmpunordpd ymm3, ymm3, ymm1 # regular NaN-is-larger ymm0&lt;ymm1
                vpor        ymm2, ymm2, ymm3
                vpblendvb   ymm0, ymm1, ymm0, ymm2
            1. [deleted] · · focus · HN ↗

              [deleted]

Open on Hacker News to reply ↗

Unofficial Hacker News client; not affiliated with Y Combinator.