xla

[FSDP] Fix pre-hook check when module is only used in forward

#4670

Merged