o
    ‡ZŽ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)ÚmodulesZ_is_fsdp_managed_moduleZ_fsdp_use_orig_params)r   r   r   Ú	submodule© r   úS/var/www/auris/lib/python3.10/site-packages/torch/distributed/fsdp/_dynamo_utils.pyÚ_annotate_modules_for_dynamo   s   
€ăr
   )Ztorch.nnÚnnÚModuleÚsetÚboolr
   r   r   r   r	   Ú<module>   s    ˙ţýü