U yhr @s6ddlmZddlmZejeejedddZdS))SetN)moduleignored_modulesuse_orig_paramscCs&|D]}||krd|_||_qdS)z Annotates the submodules in ``module`` 's tree, except those in ``ignored_modules``, indicating that the submodules are FSDP-managed and saving the ``use_orig_params`` setting passed to the FSDP constructor. TN)modulesZ_is_fsdp_managed_moduleZ_fsdp_use_orig_params)rrr submodulerV/var/www/html/venv/lib/python3.8/site-packages/torch/distributed/fsdp/_dynamo_utils.py_annotate_modules_for_dynamos r )typingrZtorch.nnnnModuleboolr rrrr s