From fe2beabf56acef7754bcc72ac2f6e0b8e50d421e Mon Sep 17 00:00:00 2001 From: Vasiliy Kuznetsov Date: Sun, 19 Oct 2025 08:26:51 -0400 Subject: [PATCH] mx_formats - add compile to training example --- torchao/prototype/mx_formats/README.md | 1 + 1 file changed, 1 insertion(+) diff --git a/torchao/prototype/mx_formats/README.md b/torchao/prototype/mx_formats/README.md index b5ba91326d..5323727a8b 100644 --- a/torchao/prototype/mx_formats/README.md +++ b/torchao/prototype/mx_formats/README.md @@ -92,6 +92,7 @@ config = MXLinearConfig( scale_calculation_mode=scale_calculation_mode, ) quantize_(m, config) +m = torch.compile(m, fullgraph=True) # training loop (not shown) ```