o
    Xi                     @   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   [/home/ubuntu/.local/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   