Code

Jcaip/llm bsr#1601

Code

Draft

jcaip wants to merge 18 commits intomainpytorch/ao:mainfrom jcaip/llm-bsrpytorch/ao:jcaip/llm-bsrCopy head branch name to clipboard

+1,509-367

Commits

Commits on Jan 21, 2025

• wip

  jcaip

  committed
  9b18913

  Copy full SHA for 9b18913

• cleaned up supermask

  jcaip

  committed
  19aa5d0

  Copy full SHA for 19aa5d0

• cleanup

  jcaip

  committed
  c1616c4

  Copy full SHA for c1616c4

Commits on Jan 22, 2025

• update

  jcaip

  committed
  2e78fc3

  Copy full SHA for 2e78fc3

Commits on Jan 23, 2025

• added padding to triton kernel

  jcaip

  committed
  bd3a3b1

  Copy full SHA for bd3a3b1

• wip

  jcaip

  committed
  44985d2

  Copy full SHA for 44985d2

Commits on Jan 24, 2025

• wip

  jcaip

  committed
  b32bdfb

  Copy full SHA for b32bdfb

• wip

  jcaip

  committed
  5e25b8b

  Copy full SHA for 5e25b8b

• added tests

  jcaip

  committed
  fe655a2

  Copy full SHA for fe655a2

• cleaned up BSR code

  jcaip

  committed
  b12df57

  Copy full SHA for b12df57

• update generate.py

  jcaip

  committed
  6df43e0

  Copy full SHA for 6df43e0

• wip

  jcaip

  committed
  d7fd295

  Copy full SHA for d7fd295

• wip

  jcaip

  committed
  13e230c

  Copy full SHA for 13e230c

• updated

  jcaip

  committed
  560198f

  Copy full SHA for 560198f

• moved file

  jcaip

  committed
  89f3ad0

  Copy full SHA for 89f3ad0

• big supermask refactor

  jcaip

  committed
  b414b49

  Copy full SHA for b414b49

Commits on Jan 28, 2025

• update

  jcaip

  committed
  1ff8aa0

  Copy full SHA for 1ff8aa0

• bsr triton updateS

  jcaip

  committed
  d503e5d

  Copy full SHA for d503e5d