
    {-j%                     (    d dl mZ ddlmZ dad ZdS )    )signature_safe_contextmanager   )ProgramFc                      t           dd            } dd}| |d}t          s4|                                D ]\  }}t          t          ||           dad S d S )	NFc              3      K   d V  d S )N )selfis_with_opts     X/var/www/html/banglarbhumi/venv/lib/python3.11/site-packages/paddle/pir/program_patch.py_lr_schedule_guardz0monkey_patch_program.<locals>._lr_schedule_guard   s      
 	    allc                     ddl m} ddlm} |5t	          ||j                  s t          dt          |           d          |
 |            }|                     ||          S )Nr   )core)global_scopezD`scope` should be None or `paddle.static.Scope'` type, but received .)	paddle.baser   paddle.base.executorr   
isinstance_Scope	TypeErrortype_state_dict)r	   modescoper   r   s        r   
state_dictz(monkey_patch_program.<locals>.state_dict   s    $$$$$$555555Zt{%C%CeW[\aWbWbeee   = LNNEe,,,r   )r   r   T)F)r   N)r   _already_patch_programitemssetattrr   )r   r   program_attrsattrvalues        r   monkey_patch_programr#      s    "   #"- - - - 1  M " &(..00 	* 	*KD%GT5))))!%	& &r   N)paddle.base.wrapped_decoratorr    r   r   r#   r   r   r   <module>r&      sK    H G G G G G       & & & & &r   