Skip to content

Commit 9e3758d

Browse files
authored
Update mx README.md (#2778)
change B200 verbiage
1 parent 0347f35 commit 9e3758d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchao/prototype/mx_formats/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ in native PyTorch. We are currently in prototype and are actively working on op
1414

1515
## Training e2e benchmarks on NVIDIA B200
1616

17-
- Single-node training on 8xB200 GPUs limited to 750W, batch size 1, sequence length 8192, steps 100, `torch.compile`, FSDP2, per-op SAC
17+
- Single-node training on 8x power limited B200 GPUs, batch size 1, sequence length 8192, steps 100, `torch.compile`, FSDP2, per-op SAC
1818
- pytorch version: `2.9.0.dev20250815+cu128`, torchao version: `0.13.0+gite4e681be6`, torchtitan commit: `6fc499f6f5b32151a799188be2208cfb09faed30`
1919

2020
| Model | Scaling | Peak Memory (GB) | Median tokens/second | Speedup over baseline

0 commit comments

Comments
 (0)