Code

Clean up FP6-LLM#304

Code

Merged

msaroufim merged 23 commits intopytorch:mainpytorch/ao:mainfrom gau-nernst:improve_fp6_llm_lineargau-nernst/ao:improve_fp6_llm_linearCopy head branch name to clipboardJun 9, 2024

+87-1,190

Commits

Commits on May 30, 2024

• override load from state dict

  gau-nernst

  committed
  7fabc8f

  Copy full SHA for 7fabc8f

Commits on May 31, 2024

• fix prefix

  gau-nernst

  committed
  1c08568

  Copy full SHA for 1c08568

Commits on Jun 1, 2024

• migrate to mx primitive

  gau-nernst

  committed
  d89e9da

  Copy full SHA for d89e9da

• remove unneeded code

  gau-nernst

  committed
  6f84293

  Copy full SHA for 6f84293

• comment out test

  gau-nernst

  committed
  571910b

  Copy full SHA for 571910b

• Merge branch 'pytorch:main' into improve_fp6_llm_linear

  gau-nernst

  authored
  73b8354

  Copy full SHA for 73b8354

Commits on Jun 2, 2024

• remove

  gau-nernst

  committed
  4e2964a

  Copy full SHA for 4e2964a

• add rounding test for f6_e3m2

  gau-nernst

  committed
  adefee8

  Copy full SHA for adefee8

• update tests

  gau-nernst

  committed
  f8268f0

  Copy full SHA for f8268f0

• remove openmp flag

  gau-nernst

  committed
  ebbff67

  Copy full SHA for ebbff67

• update benchmark script

  gau-nernst

  committed
  25e4be7

  Copy full SHA for 25e4be7

Commits on Jun 3, 2024

• test negative number

  gau-nernst

  committed
  21dfc60

  Copy full SHA for 21dfc60

• remove qtorch dep

  gau-nernst

  committed
  64e24f7

  Copy full SHA for 64e24f7

• fix type casting

  gau-nernst

  committed
  6d6f5dd

  Copy full SHA for 6d6f5dd

• add view

  gau-nernst

  committed
  474ebc2

  Copy full SHA for 474ebc2

Commits on Jun 4, 2024

• Merge branch 'main' into improve_fp6_llm_linear

  gau-nernst

  committed
  e64fbac

  Copy full SHA for e64fbac

• Merge branch 'main' into improve_fp6_llm_linear

  gau-nernst

  authored
  3231039

  Copy full SHA for 3231039

Commits on Jun 5, 2024

• Merge branch 'pytorch:main' into improve_fp6_llm_linear

  gau-nernst

  authored
  d7ec248

  Copy full SHA for d7ec248

Commits on Jun 7, 2024

• Merge branch 'pytorch:main' into improve_fp6_llm_linear

  gau-nernst

  authored
  86562c2

  Copy full SHA for 86562c2

Commits on Jun 9, 2024

• Merge branch 'main' into improve_fp6_llm_linear

  gau-nernst

  committed
  60c8e6a

  Copy full SHA for 60c8e6a

• fix strange pytest behavior

  gau-nernst

  committed
  509217c

  Copy full SHA for 509217c

• only skip tests requiring PyTorch 2.4

  gau-nernst

  committed
  11dcba3

  Copy full SHA for 11dcba3

• remove weight loading magic

  gau-nernst

  committed
  6f8e7e9

  Copy full SHA for 6f8e7e9