跳到正文
r/MachineLearning· /u/phoenix10701·· 6 小时前AI 评分54

Taco:跨平台 LLM 压缩工具,更换 GPU 或 CPU 后仍保持一致

Taco: cross-platform LLM compression that doesn't break when you change your GPU or CPU. [P]

AI 导读

Taco 是一个跨平台 LLM 压缩项目,目标是让压缩器在解码时算出与编码器完全相同的概率。作者指出,浮点数在不同硬件上存在差异,在 Mac 上编码的 float GPT-2 到 x86 上解码时。

正文
  1. The problem: an LLM compressor needs the decoder to compute exactly the same probabilities as the encoder. Floats differ across hardware; float GPT-2 encoded on a Mac decoded 0 of 4 windows on x86.
  2. The approach:
    • integer-only inference (int8 weights, fixed rounding, lookup tables);
    • an exact float32 → integer converter (mantissa × 2^exp);
    • conformance vectors.
  3. The proof: one hash across 7 backends (Mac CPU/Metal, Linux CPU/CUDA, Windows CPU/CUDA/Vulkan

https://github.com/tac0mania/taco

submitted by /u/phoenix10701
[link] [留言]

来源:r/MachineLearning · reddit.com