mx_formats - add compile to training example (#3211)

vkuzo · web-flow · commit 35fbac9509d9 · 2025-10-19T08:33:20.000-07:00
diff --git 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)
 ```

Original file line number	Diff line number	Diff line change
`@@ -92,6 +92,7 @@ config = MXLinearConfig(`
`92`	`92`	`scale_calculation_mode=scale_calculation_mode,`
`93`	`93`	`)`
`94`	`94`	`quantize_(m, config)`
	`95`	`+m = torch.compile(m, fullgraph=True)`
`95`	`96`
`96`	`97`	`# training loop (not shown)`
`97`	`98`	```