o
    s::jí  ã                   @   sd   d dl Z d dlmZmZ dd„ Zdd„ Zdd„ Ze ee¡Ze ee¡Z	e 
ee	g¡Ze e¡ZdS )	é    N)Ú_fusion_utilsÚpatternc                 C   s    d||   |t d¡ ¡d   S ©Ng      à?é   g      ð?©ÚErfÚmathÚsqrt©ÚopÚx© r   ún/home/nk/hobo-godmode/plappi-mvp/.venv/lib/python3.10/site-packages/onnxscript/rewriter/ort_fusions/erfgelu.pyÚerf_gelu_pattern_1	   s    
r   c                 C   s    |d|   |t d¡ ¡d   S r   r   r
   r   r   r   Úerf_gelu_pattern_2   s    r   c                 C   s   | j |ddS )Nzcom.microsoft)Ú_domain)ÚGelur
   r   r   r   Úgelu   s   r   )r   Úonnxscript.rewriterr   r   r   r   r   ÚRewriteRuleÚrule1Úrule2ÚRewriteRuleSetÚrulesÚapply_fusion_rulesÚfuse_erfgelur   r   r   r   Ú<module>   s   