Skip to content

Commit 1eb5902

Browse files
committed
Update README.md for mx_formats build from source (#2934)
Update README.md
1 parent 22638cd commit 1eb5902

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

torchao/prototype/mx_formats/README.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,8 @@ including [downloading a tokenizer](https://github.com/pytorch/torchtitan?tab=re
3636
- mxfp8_cublas: `TORCHTITAN_ROOT=<path> MX_RECIPE="mxfp8_cublas" ./benchmarks/float8/training/llama3.sh`
3737
- mxfp8_cublas_rceil: `TORCHTITAN_ROOT=<path> MX_RECIPE="mxfp8_cublas_rceil" ./benchmarks/float8/training/llama3.sh`
3838

39+
> :warning: For now you need to build `torchao` from source for optimal training performance. See https://github.com/pytorch/ao/issues/2932 for details.
40+
3941
# User API
4042

4143
## MX training

0 commit comments

Comments
 (0)