
    zj&                        U d dl mZ d dlZd dlmZmZmZ d dlmZ erd dl	m
Z
  ed          Z ed          Z e            Zded	<    e            Zded
<   ddZd Zd ZddZd ZddZddZd ZdS )    )annotationsN)TYPE_CHECKINGCallableTypeVar)	ParamSpec)CodeTypeTPzset[CodeType]NO_BREAKGRAPH_CODESNO_FALLBACK_CODESinputboolc                    | sJ d S N )r   s    c/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/paddle/jit/sot/psdb.pyassert_truer       s    LL5LL    c                 0    t          j        dg| R i | d S )Nz	[Dygraph])builtinsprint)argskwargss     r   r   r   $   s(    N;000000000r   c                     dd l } | j        j                            d           }t	          j                     | j        j                            |           d S )Nr   )paddle	frameworkcoreset_eval_framer   
breakpoint)r   olds     r   r   r   (   sS    MMM



.
.t
4
4C
((-----r   fnCallable[P, T]returnc                D    t                               | j                   | S r   )r   add__code__r!   s    r   check_no_breakgraphr(   0   s    BK(((Ir   c                     d S r   r   r   r   r   
breakgraphr*   5       Dr   c                D    t                               | j                   | S r   )r   r%   r&   r'   s    r   check_no_fallbackr-   9   s    "+&&&Ir   Fc                    d S r   r   )	recursives    r   fallbackr0   >   r+   r   c                     dS )NFr   r   r   r   in_sotr2   B   s    5r   )r   r   )r!   r"   r#   r"   )F)
__future__r   r   typingr   r   r   typing_extensionsr   typesr   r	   r
   setr   __annotations__r   r   r   r   r(   r*   r-   r0   r2   r   r   r   <module>r9      sO   # " " " " " "  3 3 3 3 3 3 3 3 3 3 ' ' ' ' ' ' GCLLIcNN%(SUU  * * * *#&355  ( ( ( (   1 1 1. . .   
	 	 	   
	 	 	 	    r   