Code

Refactor custom FPx cast#363

Code

Merged

msaroufim merged 11 commits intopytorch:mainpytorch/ao:mainfrom gau-nernst:custom_fpxgau-nernst/ao:custom_fpxCopy head branch name to clipboardJun 17, 2024

+239-348

Commits

Commits on Jun 14, 2024

• refactor custom fp cast

  gau-nernst

  committed
  47f7bc1

• add dequant

  gau-nernst

  committed
  da17611

• small formating

  gau-nernst

  committed
  3345740

Commits on Jun 15, 2024

• compile with fullgraph=True

  gau-nernst

  committed
  2690b92

• add fullgraph=true

  gau-nernst

  committed
  8aa0146

• undo

  gau-nernst

  committed
  be77632

• add another version

  gau-nernst

  committed
  95f4582

• fast path for mbits=1

  gau-nernst

  committed
  dcd5a05

• Merge branch 'pytorch:main' into custom_fpx

  gau-nernst

  authored
  f61ff05

Commits on Jun 17, 2024

• Merge branch 'pytorch:main' into custom_fpx

  gau-nernst

  authored
  4ad065f

• add back docstring

  gau-nernst

  committed
  bd64efc