o
    HñhG
  ã                   @   s2   d dl mZ dejdeej deddfdd„ZdS )é    NÚmoduleÚignored_modulesÚuse_orig_paramsÚreturnc                 C   s(   |   ¡ D ]}||vr	 d|_||_qdS )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)ÚmodulesÚ_is_fsdp_managed_moduleÚ_fsdp_use_orig_params)r   r   r   Ú	submodule© r
   úX/var/www/vscode/kcb/lib/python3.10/site-packages/torch/distributed/fsdp/_dynamo_utils.pyÚ_annotate_modules_for_dynamo   s   
€ãr   )Útorch.nnÚnnÚModuleÚsetÚboolr   r
   r
   r
   r   Ú<module>   s    ÿþýü