Code

Bitpacking#291

Code

Merged

msaroufim merged 32 commits intopytorch:mainpytorch/ao:mainfrom vayuda:bitpackingvayuda/ao:bitpackingCopy head branch name to clipboardMay 30, 2024

+171

Commits

Commits on May 23, 2024

• init additions

  vayuda

  committed
  37e4d2d

Commits on May 25, 2024

• extended pack/unpack

  vayuda

  committed
  b37b529

• Merge branch 'pytorch:main' into uint4-improvements

  vayuda

  authored
  f834885

Commits on May 26, 2024

• pack/unpack from n to m dtypes

  vayuda

  committed
  33845fe

• works with torch.compile, but not optimized

  vayuda

  committed
  3e7ca9b

• works on gpu

  vayuda

  committed
  88fe113

• Merge branch 'pytorch:main' into uint4-improvements

  vayuda

  authored
  3e1813a

Commits on May 28, 2024

• added row-wise bitpack

  vayuda

  committed
  80b9a41

• Merge branch 'pytorch:main' into uint4-improvements

  vayuda

  authored
  036334c

• Merge branch 'uint4-improvements' of https://github.com/JayakumarPawan/ao into uint4-improvements

  vayuda

  committed
  6a02cc1

Commits on May 29, 2024

• Merge branch 'pytorch:main' into uint4-improvements

  vayuda

  authored
  61d3666

• restructured into prototype/

  vayuda

  committed
  47d9c92

• Merge branch 'uint4-improvements' of https://github.com/vayuda/ao into uint4-improvements

  vayuda

  committed
  5bdef89

• revert nuclear fix

  vayuda

  committed
  8d1ea34

• removed temp log

  vayuda

  committed
  6e1a7d6

• removed trinary stuff from this branch

  vayuda

  committed
  46e39fd

• moved tests to tests/

  vayuda

  committed
  dcada3e

• updated tests to skip if cuda DNE

  vayuda

  committed
  a5dd25d

• added full_graph=True to compile

  vayuda

  committed
  5ec3deb

• Apply suggestions from code review

  msaroufim

  authored
  6e9a738

• fixed test skipping

  vayuda

  committed
  189677d

• Merge branch 'bitpacking' of https://github.com/vayuda/ao into bitpacking

  vayuda

  committed
  4e5b8a5

• Merge branch 'main' into bitpacking

  vayuda

  authored
  98cf146

• added import for has_triton

  vayuda

  committed
  c8055b2

• Merge branch 'bitpacking' of https://github.com/vayuda/ao into bitpacking

  vayuda

  committed
  45b990c

• Merge branch 'pytorch:main' into bitpacking

  vayuda

  authored
  74f03e2

• added support for any device type

  vayuda

  committed
  ae91c6c

• Merge branch 'bitpacking' of https://github.com/vayuda/ao into bitpacking

  vayuda

  committed
  e1253b5

• Merge branch 'pytorch:main' into bitpacking

  vayuda

  authored
  f6758e9

• fix gpu tests

  vayuda

  committed
  68179e8

• fix gpu tests

  vayuda

  committed
  4ce1295

• Merge branch 'bitpacking' of https://github.com/vayuda/ao into bitpacking

  vayuda

  committed
  b01550f