Zigzag Decoding with AVX-512

(zeux.io)

78 points | by luu 4 days ago ago

13 comments

  • flohofwoe 2 hours ago

    Worth mentioning that MeshOptimizer (https://github.com/zeux/meshoptimizer) has become one of a handful 'hidden champion' pillar libraries that probably carries half of the gaming industry.

    Basically the curl of asset pipelines ;)

    https://github.com/zeux/meshoptimizer/discussions/986

  • MaskRay 4 hours ago

    While we could utilize zigzag encoding (i>>31) ^ (i<<1) to convert SLEB128-encoded type/addend to use ULEB128 instead, the generate code is inferior to or on par with SLEB128 for one-byte encodings on x86, AArch64, and RISC-V. Haven't tried wider values - but zigzag encoding is likely slower as well

    // One-byte case for SLEB128 int64_t from_signext(uint64_t v) { return v < 64 ? v - 128 : v; }

    // One-byte case for ULEB128 with zig-zag encoding int64_t from_zigzag(uint64_t z) { return (z >> 1) ^ -(z & 1); }

  • londons_explore 3 hours ago

    This sort of analysis is great.

    Now why can't compilers do this sort of thing automatically?

    Almost any problem seems to be possible to speed up 1000x in AVX512+days of thought compared to the naive version written in a python loop. If we could automate that whole process for big codebases the performance gains could be huge.

    • cmovq 3 hours ago

      Compilers can’t really, in a meaningful way, change the layout of your data in memory. And you do need to think about your memory layout to get any benefit from SIMD. You’ll notice a lot of compiler auto vectorization insert many instructions just to shuffle data around to get to a usable layout, which negates much of the benefit.

      • fulafel 2 hours ago

        Depends on the programming language. A good question is why we don't have more optimizable languages in mainstream use.

        • inigyou 12 minutes ago

          FORTRAN is used for a lot of numerical algorithms - today! installed on your computer right now in some library! - because it optimizes better than C because it doesn't have pointers.

          • fc417fc802 a few seconds ago

            As I understand it most of the difference can be made up by adding the restrict qualifier to everything in C.

        • flohofwoe 2 hours ago

          Are there any programming languages which change the data layout beyond naively sorting struct members by alignment? (which at best helps with reducing padding bytes but can be either good or bad for performance, depending on the code which accesses the data).

          • londons_explore 33 minutes ago

            One simple optimization is to change arrays of struts into struts of arrays. To my knowledge, nothing even makes those changes, despite them being safe and having a huge potential performance benefit.

            • flohofwoe 26 minutes ago

              Now that you mention it... :)

              Zig has MultiArrayList in the stdlib which does the SoA transform via comptime:

              https://ziglang.org/documentation/master/std/#std.multi_arra...

              Zig also sorts struct members by size/alignment, but has two escape hatches ('extern struct' which is for C compatibility, and 'packed struct' which offers an explicit bit-by-bit memory layout).

              AFAIK Odin and Jai offer the SoA transform as specialized language features, e.g. in Odin:

              https://odin-lang.org/docs/overview/#soa-data-types

              I'd still always want such data layout transforms as an explicit language feature though, not the compiler making this decision for me.

    • flohofwoe 2 hours ago

      > 1000x in AVX512+days of thought compared to the naive version written in a python loop

      Out of this 1000x speedup you get 100x by just not using python though ;)

      Also IIRC the main problem specifically with AVX512 was that mainstream CPUs simply didn't have it, so a smart compiler won't be of much use when the output code only runs on a handful devices.

    • diamondlovesyou 3 hours ago

      > Now why can't compilers do this sort of thing automatically?

      They do - they just can't assume GFNI instructions are present unless you explicitly say so: https://godbolt.org/z/eYasbKsse

    • lifthrasiir 2 hours ago

      Intents matter. Compilers can't see through your skull to infer your intents and thus behave very conservatively unless you override that behavior somehow. This inference, alas, also takes (much) time, so compilers have to balance the compilation time with quality of intents guessed as well. (This is why we can't exactly use LLMs in mainstream compilers, by the way.) So go and make a programming language that preserves your intents by every means; but making it practical would be very difficult.