
    sgr
                     ^    d dl mZ d dlmZ dej                  deej                     defdZy)    )SetNmoduleignored_modulesuse_orig_paramsc                 V    | j                         D ]  }||vs	 d|_        ||_         y)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   	submodules       W/var/www/html/venv/lib/python3.12/site-packages/torch/distributed/fsdp/_dynamo_utils.py_annotate_modules_for_dynamor      s;     ^^% >	O+* 15I- />I+;>    )typingr   torch.nnnnModuleboolr    r   r   <module>r      s6     '>II'>^'> '>r   