try setting layout_opt_default=0

tinglvv · tinglvv · commit 1b53ad00739f · 2025-05-12T15:29:09.000-07:00
diff --git a/torch/_inductor/config.py b/torch/_inductor/config.py
@@ -476,7 +476,7 @@ def use_autoheuristic(name: str) -> bool:
 )
 
 # Disabled by default on ROCm, opt-in if model utilises NHWC convolutions
-layout_opt_default = "1" if not torch.version.hip else "0"
+layout_opt_default = "0"
 layout_optimization = (
     os.environ.get("TORCHINDUCTOR_LAYOUT_OPTIMIZATION", layout_opt_default) == "1"
 )

Original file line number	Diff line number	Diff line change
`@@ -476,7 +476,7 @@ def use_autoheuristic(name: str) -> bool:`
`476`	`476`	`)`
`477`	`477`
`478`	`478`	`# Disabled by default on ROCm, opt-in if model utilises NHWC convolutions`
`479`		`-layout_opt_default = "1" if not torch.version.hip else "0"`
	`479`	`+layout_opt_default = "0"`
`480`	`480`	`layout_optimization = (`
`481`	`481`	`os.environ.get("TORCHINDUCTOR_LAYOUT_OPTIMIZATION", layout_opt_default) == "1"`
`482`	`482`	`)`