Code

Add sparse marlin AQT layout#621

Code

Merged

jcaip merged 8 commits intopytorch:mainpytorch/ao:mainfrom Diogo-V:549-sparse-marlin-kernelDiogo-V/ao:549-sparse-marlin-kernelCopy head branch name to clipboardSep 6, 2024

+538-102

Commits

Commits on Sep 5, 2024

• feat: starting layout implementation

  Show description for fdad96f
  Diogo-V

  committed
  fdad96f

  Copy full SHA for fdad96f

• compile kind of working

  

  jcaip

  authored and

  Diogo-V

  committed
  b810707

  Copy full SHA for b810707

• fix: batching and layout outputs correct results

  Diogo-V

  committed
  b5eddd8

  Copy full SHA for b5eddd8

• fix: torch.compile

  Show description for cf5e286
  Diogo-V

  committed
  cf5e286

  Copy full SHA for cf5e286

• chore: cleanup

  Diogo-V

  committed
  ef17ed6

  Copy full SHA for ef17ed6

• chore: review

  Diogo-V

  committed
  ea70c74

  Copy full SHA for ea70c74

Commits on Sep 6, 2024

• chore: review v2

  Diogo-V

  committed
  153fd0b

  Copy full SHA for 153fd0b

• update benchmarks + README

  jcaip

  committed
  333a88f

  Copy full SHA for 333a88f