PyTorch PR fixes static specialization for FSDP modules

ezyang · x · 2026-08-26

This PR addresses an issue where compiling FSDP2-wrapped models generates excessive symbolic kernel args, destroying Inductor fusion. The solution statically specializes int/tuple attrs of FSDP-managed modules under skipfsdpguards.

Original post →

More from Infra

Infra channel →