Code

Fix DDP with nf4#1684

Code

Open

jeromeku wants to merge 5 commits intopytorch:mainpytorch/ao:mainfrom jeromeku:nf4-ddp-fixjeromeku/ao:nf4-ddp-fixCopy head branch name to clipboard

+300-27

Commits

Commits on Feb 9, 2025

• implement aten.cat.default for nf4

  jeromeku

  committed
  fb05bf1

  Copy full SHA for fb05bf1

• add nf4 ddp tests

  jeromeku

  committed
  6e77adf

  Copy full SHA for 6e77adf

• run ruff

  jeromeku

  committed
  49a1ea2

  Copy full SHA for 49a1ea2

• add dtype check

  jeromeku

  committed
  8814571

  Copy full SHA for 8814571

• formatting

  jeromeku

  committed
  22bc211

  Copy full SHA for 22bc211