Code

Feat/quant/per block#2849

Code

Merged

laggui merged 28 commits intomaintracel-ai/burn:mainfrom feat/quant/per-blocktracel-ai/burn:feat/quant/per-blockCopy head branch name to clipboardMar 3, 2025

+2,275-635

Commits

Commits on Feb 27, 2025

• Refactor scheme to have quantization mode + add per-block variant

  laggui

  committed
  c51e348

  Copy full SHA for c51e348

• Add per-block calibration range

  laggui

  committed
  aa8bc1b

  Copy full SHA for aa8bc1b

• Add per-block qparams compute

  laggui

  committed
  cb1d36d

  Copy full SHA for cb1d36d

• Update book

  laggui

  committed
  1744e15

  Copy full SHA for 1744e15

• Add per-block quantization strategy

  laggui

  committed
  2f3c80d

  Copy full SHA for 2f3c80d

• Working per-block tests

  laggui

  committed
  fb029b5

  Copy full SHA for fb029b5

• Remove dead code

  laggui

  committed
  a833a49

  Copy full SHA for a833a49

• Quantize works

  laggui

  committed
  39c937c

  Copy full SHA for 39c937c

• Dequantize works

  laggui

  committed
  fc2c005

  Copy full SHA for fc2c005

• TODO

  laggui

  committed
  b1a3190

  Copy full SHA for b1a3190

• Add test with block_size > line_size

  laggui

  committed
  90bc2ee

  Copy full SHA for 90bc2ee

• Clean up

  laggui

  committed
  1a4bd87

  Copy full SHA for 1a4bd87

• Fix clippy + typos

  laggui

  committed
  2fe9015

  Copy full SHA for 2fe9015

• Fix no-std

  laggui

  committed
  008a1d0

  Copy full SHA for 008a1d0

• Add might_panic test attribute

  laggui

  committed
  ab9114f

  Copy full SHA for ab9114f

• Feature gate might_panic for no-std

  laggui

  committed
  1bcd205

  Copy full SHA for 1bcd205

• Add from/to data tests + make panic reason more specific

  laggui

  committed
  a6790c3

  Copy full SHA for a6790c3

• We forgot default std

  laggui

  committed
  55d62ef

  Copy full SHA for 55d62ef

Commits on Feb 28, 2025

• Merge branch 'main' into feat/quant/per-block

  laggui

  committed
  8097fe8

  Copy full SHA for 8097fe8

• WIP macos debug

  laggui

  committed
  a7bf68b

  Copy full SHA for a7bf68b

• More debug

  laggui

  committed
  3cc94ff

  Copy full SHA for 3cc94ff

• Debug

  laggui

  committed
  d50bff5

  Copy full SHA for d50bff5

• Fix precision issues

  laggui

  committed
  8ab78f8

  Copy full SHA for 8ab78f8

• Debug ci

  laggui

  committed
  fe0002d

  Copy full SHA for fe0002d

• More ci

  laggui

  committed
  d44dc0a

  Copy full SHA for d44dc0a

• Clean up ci

  laggui

  committed
  eda3e6a

  Copy full SHA for eda3e6a

• Remove todo comment

  laggui

  committed
  a40340a

  Copy full SHA for a40340a

• Remove comment

  laggui

  committed
  24d5857

  Copy full SHA for 24d5857