§
    2@Äjs“  ã                   ó¦  — d Z ddlZddlZddlZddlmZ ddlmZ ddlZ	ddl
Z
ddlmZmZmZ ddlmZ ddlmZmZmZmZ ddlmZmZmZmZmZmZ ddlmZmZm Z m!Z! dd	l"m#Z# dd
l$m%Z% ddl&m'Z'm(Z( ddlm)Z) ddlmZ*  ed¦  «        Z+ G d„ de,¦  «        Z- G d„ de,¦  «        Z. G d„ d edd¦  «        ¦  «        Z/ G d„ d edd¦  «        ¦  «        Z0 G d„ d edd¦  «        ¦  «        Z1d0d„Z2 ej3        ej3         ej4        ej3        ¦  «        ej3         ej4        ej3        ¦  «        ¦  «        Z5d„ Z6d„ Z7d „ Z8d!„ Z9d"„ Z:d#„ Z;d$„ Z< G d%„ d&e,¦  «        Z=d'„ Z>d(„ Z?	 d1d*„Z@d1d+„ZAe+ B                    e
jC        ejD        ¦  «        d,„ ¦   «         ZEd-„ ZF eF ejG        ¦   «         e+jB        ¦  «         e(d.„ ¦   «         ZH e'e	jI        ¦  «        d/„ ¦   «         ZJdS )2z3
Implementation of functions in the Numpy package.
é    N)Ú
namedtuple)ÚarrayobjÚufunc_dbÚnumpy_support)Úparse_signature)ÚRegistryÚimpl_ret_new_refÚforce_error_modelÚimpl_ret_borrowed)ÚtypingÚtypesÚutilsÚcgutilsÚcallconvÚconfig)Úufunc_find_matching_loopÚselect_array_wrapperÚ
from_dtypeÚ_ufunc_loop_sig)Ú_getitem_array_generic)Únpydecl)ÚoverloadÚ	intrinsic)Úerrors)r   Únpyimplc                   ó   — e Zd Zd„ Zd„ ZdS )Ú_ScalarIndexingHelperc                 ó   — d S ©N© )ÚselfÚloop_indicesÚnames      úAC:\Projects\Historical\venv\Lib\site-packages\numba/np/npyimpl.pyÚupdate_indicesz$_ScalarIndexingHelper.update_indices.   ó   € Øˆó    c                 ó   — d S r   r    ©r!   s    r$   Ú	as_valuesz_ScalarIndexingHelper.as_values1   r&   r'   N©Ú__name__Ú
__module__Ú__qualname__r%   r*   r    r'   r$   r   r   -   s2   € € € € € ðð ð ðð ð ð ð r'   r   c                   ó@   — e Zd ZdZd„ Zd„ Zd„ Zd„ Zed„ ¦   «         Z	dS )Ú_ScalarHelpera\  Helper class to handle scalar arguments (and result).
    Note that store_data is only used when generating code for
    a scalar ufunc and to write the output value.

    For loading, the value is directly used without having any
    kind of indexing nor memory backing it up. This is the use
    for input arguments.

    For storing, a variable is created in the stack where the
    value will be written.

    Note that it is not supported (as it is unneeded for our
    current use-cases) reading back a stored value. This class
    will always "load" the original value it got at its creation.
    c                 óX  — || _         || _        || _        || _        |                     t
          j        ¦  «        }t          j        |d¦  «        g| _	        |t
          j
        k    r|                     |¦  «        nt          j        d¦  «        }t          j        ||¦  «        | _        d S ©Né   )ÚcontextÚbuilderÚvalÚ	base_typeÚget_value_typer   ÚintpÚirÚConstantÚshapeÚbooleanÚget_data_typeÚIntTyper   Úalloca_onceÚ_ptr)r!   ÚctxtÚbldr6   ÚtyÚintptyÚltys          r$   Ú__init__z_ScalarHelper.__init__E   s�   € ØˆŒØˆŒØˆŒØˆŒØ×$Ò$¥U¤ZÑ0Ô0ˆÝ”k &¨!Ñ,Ô,Ð-ˆŒ
à(*­e¬mÒ(;Ð(;ˆd× Ò  Ñ$Ô$Ð$ÅÄÈAÁÄˆÝÔ'¨¨SÑ1Ô1ˆŒ	ˆ	ˆ	r'   c                 ó   — t          ¦   «         S r   )r   r)   s    r$   Úcreate_iter_indicesz!_ScalarHelper.create_iter_indicesP   s   € Ý$Ñ&Ô&Ð&r'   c                 ó   — | j         S r   )r6   ©r!   Úindicess     r$   Ú	load_dataz_ScalarHelper.load_dataS   s	   € ØŒxˆr'   c                 óF   — | j                              || j        ¦  «         d S r   )r5   ÚstorerA   )r!   rL   r6   s      r$   Ú
store_dataz_ScalarHelper.store_dataV   s"   € ØŒ×Ò˜3 ¤	Ñ*Ô*Ð*Ð*Ð*r'   c                 ó@   — | j                              | j        ¦  «        S r   )r5   ÚloadrA   r)   s    r$   Ú
return_valz_ScalarHelper.return_valY   s   € àŒ|× Ò  ¤Ñ+Ô+Ð+r'   N)
r,   r-   r.   Ú__doc__rG   rI   rM   rP   ÚpropertyrS   r    r'   r$   r0   r0   5   su   € € € € € ðð ð	2ð 	2ð 	2ð'ð 'ð 'ðð ð ð+ð +ð +ð ð,ð ,ñ „Xð,ð ,ð ,r'   r0   c                   ó   — e Zd Zd„ Zd„ ZdS )Ú_ArrayIndexingHelperc                 ó"  — | j         j        }| j         j                             t          j        ¦  «        }t          j        t          j        |j	        ¦  «        d¦  «        }|t          |¦  «        t          | j        ¦  «        z
  d …         }t          || j        | j         j        ¦  «        D ]`\  }}}	|                     d|	|¦  «        }
|                     |
¦  «        5  |                     ||¦  «         d d d ¦  «         n# 1 swxY w Y   Œad S )Nr3   ú>)Úarrayr5   r4   r8   r   r9   r:   r;   r?   ÚwidthÚlenrL   Úzipr<   Úicmp_unsignedÚif_thenrO   )r!   r"   r#   rC   rE   ÚONErL   ÚsrcÚdstÚdimÚconds              r$   r%   z#_ArrayIndexingHelper.update_indices`   s1  € ØŒjÔ ˆØ”Ô#×2Ò2µ5´:Ñ>Ô>ˆÝŒk�"œ* V¤\Ñ2Ô2°AÑ6Ô6ˆð
 �s <Ñ0Ô0µ3°t´|Ñ3DÔ3DÑDÐEÐEÔFˆÝ  ¨$¬,¸¼
Ô8HÑIÔIð 	$ð 	$‰MˆC��cØ×$Ò$ S¨#¨sÑ3Ô3ˆDØ—’˜TÑ"Ô"ð $ð $Ø—	’	˜#˜sÑ#Ô#Ð#ð$ð $ð $ñ $ô $ð $ð $ð $ð $ð $ð $øøøð $ð $ð $ð $øð	$ð 	$s   Ã DÄD	Ä
D	c                 óB   ‡— | j         j        Šˆfd„| j        D ¦   «         S )zø
        The indexing helper is built using alloca for each value, so it
        actually contains pointers to the actual indices to load. Note
        that update_indices assumes the same. This method returns the
        indices as values
        c                 ó:   •— g | ]}‰                      |¦  «        ‘ŒS r    ©rR   )Ú.0ÚindexrC   s     €r$   ú
<listcomp>z2_ArrayIndexingHelper.as_values.<locals>.<listcomp>v   s#   ø€ Ð:Ð:Ð: E�—’˜‘”Ð:Ð:Ð:r'   )rZ   r5   rL   )r!   rC   s    @r$   r*   z_ArrayIndexingHelper.as_valuesn   s+   ø€ ð ŒjÔ ˆØ:Ð:Ð:Ð:¨T¬\Ð:Ñ:Ô:Ð:r'   Nr+   r    r'   r$   rW   rW   ^   s2   € € € € € ð$ð $ð $ð;ð ;ð ;ð ;ð ;r'   rW   )rZ   rL   c                   ó*   — e Zd ZdZd„ Zd„ Zd„ Zd„ ZdS )Ú_ArrayHelperz­Helper class to handle array arguments/result.
    It provides methods to generate code loading/storing specific
    items as well as support code for handling indices.
    c                 ó¶  — | j                              t          j        ¦  «        }t	          j        t	          j        |j        ¦  «        d¦  «        }g }t          | j	        ¦  «        D ]c}t          j        | j        t	          j        |j        ¦  «        ¦  «        }| j                             ||¦  «         |                     |¦  «         Œdt          | |¦  «        S ©Nr   )r4   r8   r   r9   r:   r;   r?   r[   ÚrangeÚndimr   r@   r5   rO   ÚappendrW   ©r!   rE   ÚZEROrL   ÚiÚxs         r$   rI   z _ArrayHelper.create_iter_indices�   s±   € Ø”×,Ò,­U¬ZÑ8Ô8ˆÝŒ{�2œ: f¤lÑ3Ô3°QÑ7Ô7ˆàˆÝ�t”yÑ!Ô!ð 	ð 	ˆAÝÔ# D¤Lµ"´*¸V¼\Ñ2JÔ2JÑKÔKˆAØŒL×Ò˜t QÑ'Ô'Ð'Ø�NŠN˜1ÑÔÐÐÝ# D¨'Ñ2Ô2Ð2r'   c           	      ót   — t          j        | j        | j        | j        | j        | j        | j        |¬¦  «        S ©N)Údatar<   ÚstridesÚlayoutÚinds)r   Úget_item_pointer2r4   r5   rx   r<   ry   rz   rK   s     r$   Ú_load_effective_addressz$_ArrayHelper._load_effective_addressŒ   s<   € ÝÔ(¨¬Ø)-¬Ø.2¬iØ/3¬zØ15´Ø04´Ø.5ð7ñ 7ô 7ð 	7r'   c                 ó�   — | j         j        | j                 }|                      |¦  «        }|                     | j        |¦  «        S r   )r4   Údata_model_managerr7   r}   Úload_from_data_pointerr5   )r!   rL   ÚmodelÚptrs       r$   rM   z_ArrayHelper.load_data•   s>   € Ø”Ô/°´Ô?ˆØ×*Ò*¨7Ñ3Ô3ˆØ×+Ò+¨D¬L¸#Ñ>Ô>Ð>r'   c                 óö   — | j         }| j        }|                     || j        |¦  «        }|                     | j        ¦  «        |j        k    sJ ‚|                     ||                      |¦  «        ¦  «         d S r   )r4   r5   Úget_value_as_datar7   r>   ÚtyperO   r}   )r!   rL   ÚvalueÚctxrC   Ústore_values         r$   rP   z_ArrayHelper.store_dataš   su   € ØŒlˆØŒlˆØ×+Ò+¨C°´ÀÑGÔGˆØ× Ò  ¤Ñ0Ô0°KÔ4DÒDÐDÐDÐDØ�	Š	�+˜t×;Ò;¸GÑDÔDÑEÔEÐEÐEÐEr'   N)r,   r-   r.   rT   rI   r}   rM   rP   r    r'   r$   rl   rl   y   s_   € € € € € ðð ð	3ð 	3ð 	3ð7ð 7ð 7ð?ð ?ð ?ð
Fð Fð Fð Fð Fr'   rl   )	r4   r5   r<   ry   rx   rz   r7   rp   rS   c                   ó:   — e Zd ZdZd„ Zd„ Zd„ Zd„ Zdd defd„Z	d	S )
Ú_ArrayGUHelperz÷Helper class to handle array arguments/result.
    It provides methods to generate code loading/storing specific
    items as well as support code for handling indices.

    Contrary to _ArrayHelper, this class can create a view to a subarray
    c                 óÐ  — | j                              t          j        ¦  «        }t	          j        t	          j        |j        ¦  «        d¦  «        }g }t          | j	        | j
        j	        z
  ¦  «        D ]c}t          j        | j        t	          j        |j        ¦  «        ¦  «        }| j                             ||¦  «         |                     |¦  «         Œdt!          | |¦  «        S rn   )r4   r8   r   r9   r:   r;   r?   r[   ro   rp   Úinner_arr_tyr   r@   r5   rO   rq   rW   rr   s         r$   rI   z"_ArrayGUHelper.create_iter_indices¬   s¼   € Ø”×,Ò,­U¬ZÑ8Ô8ˆÝŒ{�2œ: f¤lÑ3Ô3°QÑ7Ô7ˆàˆÝ�t”y 4Ô#4Ô#9Ñ9Ñ:Ô:ð 	ð 	ˆAÝÔ# D¤Lµ"´*¸V¼\Ñ2JÔ2JÑKÔKˆAØŒL×Ò˜t QÑ'Ô'Ð'Ø�NŠN˜1ÑÔÐÐÝ# D¨'Ñ2Ô2Ð2r'   c           	      ó  — | j         }| j        }t          j        | j        | j        | j        ¦  «        } |                     |¦  «        ||| j        ¦  «        }t          j
        |||j        | j        | j        | j        |¬¦  «        S rw   )r4   r5   r   ÚArrayr7   rp   rz   Ú
make_arrayrx   r   r|   r<   ry   )r!   rL   r4   r5   Úarr_tyÚarrs         r$   r}   z&_ArrayGUHelper._load_effective_address·   s‚   € Ø”,ˆØ”,ˆÝ”˜Tœ^¨T¬Y¸¼ÑDÔDˆØ(ˆg× Ò  Ñ(Ô(¨°'¸4¼9ÑEÔEˆåÔ(¨Ø)0Ø.1¬hØ/3¬zØ15´Ø04´Ø.5ð7ñ 7ô 7ð 	7r'   c           	      óº  — | j         | j        }}| j        j        dk    rD| j        r=|j        | j                 }|                      |¦  «        }|                     ||¦  «        S | j        j        dk    �r\| j        �sT| 	                    t          j        ¦  «        } |d¦  «        }t          j        | j        | j        | j        ¦  «        }t          j        | j        d| j        ¦  «        }	 |t          j        ||¦  «        ¦  «        }
 | j                              |¦  «        ||| j        ¦  «        } | j                              |	¦  «        ||¦  «        }t          j        ||                      |¦  «        t'          j        ||g¦  «        t'          j        ||
g¦  «        |j        |j        |j        ¬¦  «         |                     ¦   «         S t          j        f| j        | j        j        z
  z  }t          j        | j        | j        | j        ¦  «        } | j                              |¦  «        ||| j        ¦  «        }t5          ||| j        ||||¦  «        }|S )Nr   r3   )rx   r<   ry   ÚitemsizeÚmeminfoÚparent)r4   r5   rŒ   rp   Úis_input_argr   r7   r}   r€   r8   r   r9   rŽ   rz   r   Úget_itemsizer�   rx   Úpopulate_arrayr   Ú
pack_arrayr“   r”   r•   Ú	_getvalueÚint64r   )r!   rL   r4   r5   r�   r‚   rE   ÚoneÚfromtyÚtotyr“   Úarr_fromÚarr_toÚindex_typesÚarrtyr‘   Úress                    r$   rM   z_ArrayGUHelper.load_dataÅ   s6  € Øœ<¨¬�ˆàÔÔ! QÒ&Ð&¨4Ô+<Ð&àÔ.¨t¬~Ô>ˆEØ×.Ò.¨wÑ7Ô7ˆCØ×/Ò/°¸Ñ=Ô=Ð=ØÔÔ# qÒ(Ñ(°Ô1BÑ(ð ×+Ò+­E¬JÑ7Ô7ˆFØ�&˜‘)”)ˆCå”[ ¤°´¸D¼KÑHÔHˆFÝ”;˜tœ~¨q°$´+Ñ>Ô>ˆDØ�v�hÔ3°G¸VÑDÔDÑEÔEˆHð 7�t”|×.Ò.¨vÑ6Ô6°wØ7>Ø7;´yñBô BˆHð 3�T”\×,Ò,¨TÑ2Ô2°7¸GÑDÔDˆFÝÔ#ØØ×1Ò1°'Ñ:Ô:ÝÔ(¨°3°%Ñ8Ô8ÝÔ*¨7°X°JÑ?Ô?Ø!Ô*Ø Ô(Ø”ð(ñ (ô (ð (ð ×#Ò#Ñ%Ô%Ð%õ !œ;˜.¨D¬I¸Ô8IÔ8NÑ,NÑOˆKÝ”K ¤°´	¸4¼;ÑGÔGˆEØ0�$”,×)Ò)¨%Ñ0Ô0°¸'À4Ä9ÑMÔMˆCÝ(¨°'Ø)-Ô):¸EÀ3Ø)4°gñ?ô ?ˆCð
 ˆJr'   c           	      óä  ‡— | j         j        Šˆfd„}| j        | j        }}t	          j        t	          j        t          j        t          |¦  «        ¦  «        t	          j        t          j        t          | j	        ¦  «        ¦  «        ¦  «        }| 
                    ||j        d         |¦  «        | 
                    ||j        d         | j	        ¦  «        f}|                     ||||¦  «         d S )Nc                 óÆ   •— t          |¦  «        ‰z
  }|t          | ¦  «        k    }|s't          |¦  «        D ]}|||         | |         k    z  }Œ|rt          d¦  «        ‚d S )Nz&Loop and array shapes are incompatible©r\   ro   Ú
ValueError)Ú
loop_shapeÚarray_shapeÚ	remainingÚ_raisert   Ú
inner_ndims        €r$   Ú
raise_implz._ArrayGUHelper.guard_shape.<locals>.raise_implô   sˆ   ø€ õ
 ˜KÑ(Ô(¨:Ñ5ˆIØ¥# j¡/¤/Ò1ˆFØð @Ý˜yÑ)Ô)ð @ð @�AØ˜{¨1œ~°¸A´Ò>Ñ?�F�FØð Kõ
 !Ð!IÑJÔJÐJðKð Kr'   r   r3   )rŒ   rp   r4   r5   r   ÚnoneÚUniTupler9   r\   r<   Ú
make_tupleÚargsÚcompile_internal)r!   Ú	loopshaper­   r4   r5   ÚsigÚtupr¬   s          @r$   Úguard_shapez_ArrayGUHelper.guard_shapeò   sÞ   ø€ ØÔ&Ô+ˆ
ð	Kð 	Kð 	Kð 	Kð 	Kð"  œ<¨¬�ˆÝŒjÝŒN�5œ:¥s¨9¡~¤~Ñ6Ô6ÝŒN�5œ:¥s¨4¬:¡¤Ñ7Ô7ñ
ô 
ˆð ×!Ò! '¨3¬8°A¬;¸	ÑBÔBØ×!Ò! '¨3¬8°A¬;¸¼
ÑCÔCðEˆà× Ò  ¨*°c¸3Ñ?Ô?Ð?Ð?Ð?r'   ÚotherÚndimsc           	      óà  ‡— ˆfd„}| j         | j        }}t          j        t          j        t          j        t          | j        ¦  «        ¦  «        t          j        t          j        t          |j        ¦  «        ¦  «        ¦  «        }|                     ||j	        d         | j        ¦  «        |                     ||j	        d         |j        ¦  «        f}| 
                    ||||¦  «         d S )Nc                 óÔ   •— d}t          | ¦  «        ‰z
  t          |¦  «        ‰z
  }}t          ‰¦  «        D ]}|| ||z            |||z            k    z  }Œ|sd}t          |¦  «        ‚d S )NTzŽOperand has a mismatch in one of its core dimensions. Please, check if all arguments to a @guvectorize function have the same core dimensions.r¦   )Ú
self_shapeÚother_shapeÚsameÚaÚbrt   Úmsgr¸   s          €r$   r­   z8_ArrayGUHelper.guard_match_core_dims.<locals>.raise_impl  s‹   ø€ ØˆDÝ�z‘?”? UÑ*­C°Ñ,<Ô,<¸uÑ,DˆqˆAÝ˜5‘\”\ð @ð @�Ø˜
 1 q¡5Ô)¨[¸¸Q¹Ô-?Ò?Ñ?��Øð 
&ðA�õ ! ‘o”oÐ%ð
&ð 
&r'   r   r3   )r4   r5   r   r®   r¯   r9   r\   r<   r°   r±   r²   )r!   r·   r¸   r­   r4   r5   r´   rµ   s     `     r$   Úguard_match_core_dimsz$_ArrayGUHelper.guard_match_core_dims  sÔ   ø€ ð	&ð 	&ð 	&ð 	&ð 	&ð"  œ<¨¬�ˆÝŒjÝŒN�5œ:¥s¨4¬:¡¤Ñ7Ô7ÝŒN�5œ:¥s¨5¬;Ñ'7Ô'7Ñ8Ô8ñ
ô 
ˆð ×!Ò! '¨3¬8°A¬;¸¼
ÑCÔCØ×!Ò! '¨3¬8°A¬;¸¼ÑDÔDðGˆà× Ò  ¨*°c¸3Ñ?Ô?Ð?Ð?Ð?r'   N)
r,   r-   r.   rT   rI   r}   rM   r¶   ÚintrÁ   r    r'   r$   rŠ   rŠ   ¢   s‰   € € € € € ðð ð	3ð 	3ð 	3ð7ð 7ð 7ð+ð +ð +ðZ@ð @ð @ð8 @Ð+;ð  @ÀCð  @ð  @ð  @ð  @ð  @ð  @r'   rŠ   )
r4   r5   r<   ry   rx   rz   r7   rp   rŒ   r–   úinput operandc                 óà  — t          |t          j        ¦  «        r!|}|j        }|                      ||||¦  «        }t          |t          j        ¦  «        r‹ |                      |¦  «        | ||¦  «        }t          j        ||j	        |j
        ¦  «        }t          j        ||j        |j
        ¦  «        }t          | ||||j        |j        |j        |j
        |¦	  «	        S t          j        |¦  «        t          j        t          j        hz  v st          |t&          j        j        ¦  «        rt-          | |||¦  «        S t/          d                     |t3          |¦  «        ¦  «        ¦  «        ‚)zîreturns an instance of the appropriate Helper (either
    _ScalarHelper or _ArrayHelper) class to handle the argument.
    using the polymorphic interface of the Helper classes, scalar
    and array cases can be handled with the same codezunsupported type for {0}: {1})Ú
isinstancer   ÚOptionalr…   ÚcastÚArrayCompatibler�   r   Úunpack_tupler<   rp   ry   rl   rx   rz   ÚdtypeÚ	unliteralÚnumber_domainr=   Ú	npy_typesÚdatetimeÚ_NPDatetimeBaser0   ÚNotImplementedErrorÚformatÚstr)	rB   rC   ÚinpÚtyinpÚwhereÚotyÚaryr<   ry   s	            r$   Ú_prepare_argumentrØ   1  sO  € õ �%�œÑ(Ô(ð .ØˆØ”
ˆØ�iŠi˜˜S # uÑ-Ô-ˆõ �%�Ô.Ñ/Ô/ð /Ø(�$—/’/ %Ñ(Ô(¨¨s°CÑ8Ô8ˆÝÔ& s¨C¬I°u´zÑBÔBˆÝÔ& s¨C¬K¸¼ÑDÔDˆÝ˜D # u¨g°s´xØ!œL¨%¬+°u´zÀ3ñHô Hð 	Hå
Œ/˜%Ñ
 Ô
 ¥EÔ$7½5¼=¸/Ñ$IÐ
IÐ
IÝ˜�yÔ1ÔAÑBÔBð Jå˜T 3¨¨UÑ3Ô3Ð3å!Ð"A×"HÒ"HÈÝ"% e¡*¤*ñ#.ô #.ñ /ô /ð 	/r'   c                 ó¸   — | |k    rdS d}|| z
  }|| k     rD||         }||         }|dk    r||k    r|dk    r|dz    S n|dk    r|||<   |dz  }|dz  }|| k     °D|S )a¡  Low-level utility function used in calculating a shape for
    an implicit output array.  This function assumes that the
    destination shape is an LLVM pointer to a C-style array that was
    already initialized to a size of one along all axes.

    Returns an integer value:
    >= 1  :  Succeeded.  Return value should equal the number of dimensions in
             the destination shape.
    0     :  Failed to broadcast because source shape is larger than the
             destination shape (this case should be weeded out at type
             checking).
    < 0   :  Failed to broadcast onto destination axis, at axis number ==
             -(return_value + 1).
    r   r3   r    )Úsrc_ndimÚ	src_shapeÚ	dest_ndimÚ
dest_shapeÚ	src_indexÚ
dest_indexÚsrc_dim_sizeÚdest_dim_sizes           r$   Ú_broadcast_ontorâ   O  s²   € ð �)ÒÐð ˆqàˆ	Ø Ñ)ˆ
Ø˜(Ò"Ð"Ø$ YÔ/ˆLØ& zÔ2ˆMð  Ò!Ð!ð   =Ò0Ð0°\ÀQÒ5FÐ5FØ'¨!™^Ð,Ð,øØ Ò"Ð"à)5�
˜:Ñ&Ø˜‰NˆIØ˜!‰OˆJð ˜(Ò"Ð"ð  Ðr'   c                 óÌ  ‡ ‡‡— d„ |D ¦   «         }‰                       t          j        ¦  «        }ˆ fd„} |d¦  «        } |d¦  «        }t          j        ‰||j        d¦  «        }	 ||j        ¦  «        }
t          j        ‰||j        d¦  «        Št          ˆˆfd„t          |j        ¦  «        D ¦   «         ¦  «        }|D ]}‰                     ||¦  «         Œt          |¦  «        D �]9\  }}t          |d¦  «        sŒ ||j        ¦  «        }t          |j        ¦  «        D ]7}‰                     |j        |         t          j        ‰|	|¦  «        ¦  «         Œ8‰                      ‰t          t          ||	|
‰g¦  «        }t          j        ‰‰                     d	||¦  «        ¦  «        5  d
|fz  }t$          j                             dd¦  «        }|�|d|j        |j        fz  z  }‰ j                             ‰t2          |f¦  «         ddd¦  «         n# 1 swxY w Y   �Œ;|j        }t          ˆfd„|D ¦   «         ¦  «        }t7          j        ‰ ‰||¦  «        }t;          |¦  «        }||         }	 ‰                      d |||¦  «        ¦  «        }||         j        |                      ¦   «         f} |‰|¦  «        }n?# tB          $ r2 |j"        t          j#        j"        k    r‚ |                      ¦   «         }Y nw xY w|j        }t          j$        ‰|j        |¦  «        }t          j$        ‰|j%        |¦  «        }tM          ‰ ‰|||j'        |j(        |j)        ||¦	  «	        S )zªUtility function to handle allocation of an implicit output array
    given the target context, builder, output array type, and a list of
    _ArrayHelper instances.
    c                 óT   — g | ]%}t          |t          j        ¦  «        r|j        n|‘Œ&S r    )rÅ   r   rÆ   r…   ©rh   ru   s     r$   rj   z _build_array.<locals>.<listcomp>}  s?   € ð )ð )ð )Øõ (¨­5¬>Ñ:Ô:ÐA�1”6�6Àð )ð )ð )r'   c                 óD   •— ‰                      t          j        | ¦  «        S r   )Úget_constantr   r9   )r6   r4   s    €r$   Úmake_intp_constz%_build_array.<locals>.make_intp_const�  s   ø€ Ø×#Ò#¥E¤J°Ñ4Ô4Ð4r'   r   r3   rÛ   rÝ   c              3   óD   •K  — | ]}t          j        ‰‰|¦  «        V — Œd S r   )r   Úgep_inbounds)rh   ri   r5   rÝ   s     €€r$   ú	<genexpr>z_build_array.<locals>.<genexpr>Œ  sN   øè è € ð @ð @Ø!&õ %Ô1°'¸:ÀuÑMÔMð @ð @ð @ð @ð @ð @r'   rp   ú<z/unable to broadcast argument %d to output arrayÚlocNz
File "%s", line %d, c              3   óB   •K  — | ]}‰                      |¦  «        V — Œd S r   rg   )rh   Údest_shape_addrr5   s     €r$   rë   z_build_array.<locals>.<genexpr>­  sK   øè è € ð Dð DØ.ð #Ÿ<š<¨Ñ8Ô8ð Dð Dð Dð Dð Dð Dr'   Ú__array_wrap__)*r8   r   r9   r   r@   rp   Útuplero   rO   Ú	enumerateÚhasattrr<   rê   r²   râ   Ú_broadcast_onto_sigÚif_unlikelyÚicmp_signedr   Úloc_infoÚgetÚfilenameÚlineÚ	call_convÚreturn_user_excr§   Úas_arrayr   Ú_empty_nd_implr   Úget_functionrS   rš   rÐ   Úarray_priorityrŽ   rÉ   ry   rl   rx   rz   rÊ   ) r4   r5   Úarray_tyÚinput_typesÚinputsÚintp_tyrè   rs   r`   rÛ   rÜ   Údest_shape_addrsrï   Ú
arg_numberÚargÚarg_ndimri   Ú
arg_resultrÀ   rí   Úreal_array_tyÚdest_shape_tupÚ	array_valÚarray_wrapper_indexÚarray_wrapper_tyÚ
array_wrapÚ	wrap_argsÚout_valrp   r<   ry   rÝ   s    ``                             @r$   Ú_build_arrayr  w  s€  øøø€ ð)ð )Ø'ð)ñ )ô )€Kð ×$Ò$¥U¤ZÑ0Ô0€Gð5ð 5ð 5ð 5ð 5ð ˆ?˜1ÑÔ€DØ
ˆ/˜!Ñ
Ô
€CåÔ# G¨W°h´mØ$/ñ1ô 1€Ià� ¤Ñ.Ô.€IÝÔ$ W¨g°x´}Ø%1ñ3ô 3€Jåð @ð @ð @ð @ð @Ý*/°´Ñ*>Ô*>ð@ñ @ô @ñ @ô @Ðð ,ð ,ð ,ˆØ�Š�c˜?Ñ+Ô+Ð+Ð+õ
 % VÑ,Ô,ð Kñ K‰ˆ
�CÝ�s˜FÑ#Ô#ð 	ØØ"�? 3¤8Ñ,Ô,ˆÝ˜3œ8‘_”_ð 	Kð 	KˆEØ�MŠM˜#œ) EÔ*Ý!Ô.¨w¸	À5ÑIÔIñKô Kð Kð Kà×-Ò-Ø•_Õ&9Ø�y )¨ZÐ8ñ:ô :ˆ
õ Ô  Ø!(×!4Ò!4°S¸*ÀcÑ!JÔ!JñLô Lð 		Kð 		KàCØðGñ ˆCõ ”/×%Ò% e¨TÑ2Ô2ˆCØˆØÐ/°3´<ÀÄÐ2JÑJÑJ�àÔ×-Ò-¨gµzÀCÀ6ÑJÔJÐJð		Kð 		Kð 		Kñ 		Kô 		Kð 		Kð 		Kð 		Kð 		Kð 		Kð 		Køøøð 		Kð 		Kð 		Kð 		Kùð Ô%€Måð Dð Dð Dð DØ2BðDñ Dô Dñ Dô D€NåÔ'¨°¸-Ø(6ñ8ô 8€Iõ /¨{Ñ;Ô;ÐØ"Ð#6Ô7Ðð1à×)Ò)Ð*:Ø*2¨(Ð3CÀ]Ñ*SÔ*SñUô Uˆ
ð Ð/Ô0Ô;¸Y×=PÒ=PÑ=RÔ=RÐSˆ	Ø�*˜W iÑ0Ô0ˆˆøõ ð (ð (ð (ð Ô*­e¬kÔ.HÒHÐHØØ×%Ò%Ñ'Ô'ˆˆˆð(øøøð Œ=€DÝÔ" 7¨I¬O¸TÑBÔB€EÝÔ" 7¨IÔ,=¸tÑDÔD€GÝ˜ ¨%°¸)¼.Ø œ¨¬¸Øñ!ô !ð !s%   Æ-AHÈH	ÈH	É6 K Ë9L Ë?L c                 óP   — | j         dk    r|j        gS t          |j        ¦  «        S r2   )ÚnoutÚreturn_typeÚlist)Úufuncr´   s     r$   Ú_unpack_output_typesr  Ì  s(   € Ø„z�Q‚€Ø”Ð Ð å�C”OÑ$Ô$Ð$r'   c                 óH   — | j         dk    r|gS t          j        ||¦  «        S r2   )r  r   rÉ   )r  r5   Úvaluess      r$   Ú_unpack_output_valuesr  Ó  s'   € Ø„z�Q‚€ØˆxˆåÔ# G¨VÑ4Ô4Ð4r'   c                 óV   — | j         dk    r|d         S |                     |||¦  «        S )Nr3   r   )r  r°   )r  r4   r5   Útypr  s        r$   Ú_pack_output_valuesr  Ú  s/   € Ø„z�Q‚€Ø�aŒyÐà×!Ò! '¨3°Ñ7Ô7Ð7r'   c           
      óò  ‡ ‡— ˆˆ fd„t          ||j        ¦  «        D ¦   «         }t          |¦  «        |j        k     r;t	          d                     |j        |j        t          |¦  «        ¦  «        ¦  «        ‚t          t          ||¦  «        ¦  «        D ]Ò\  }}|j        |z   t          |¦  «        k    r�t          |t          j        ¦  «        rt          ‰ ‰||j        |¦  «        }	n8t          ‰ ‰t          j        ‰                      |¦  «        d ¦  «        |¦  «        }	|                     |	¦  «         Œ¡‰ j        r*‰ j                             ‰|||j        |z            ¦  «         ŒÓ|d |j        …         }
||j        d …         }t          |¦  «        |j        k    sJ ‚t-          d„ |D ¦   «         d„ |
D ¦   «         ¦  «        } |‰ ‰|¦  «        }‰                      t          j        ¦  «        }d„ |
D ¦   «         }|d         j        }d„ |
D ¦   «         }t          d„ |D ¦   «         ¦  «        }t          d	„ |D ¦   «         ¦  «        }||k    rd
}nd}t3          j        ‰|||¬¦  «        5 }g }t          t          ||
¦  «        ¦  «        D ]X\  }\  }}|                     ||¦  «         |                     |                     |                     ¦   «         ¦  «        ¦  «         ŒYt=          |‰ |j        |Ž ¦  «        }t          ||¦  «        D ]\  }}	|	                      ||¦  «         Œ	 d d d ¦  «         n# 1 swxY w Y   tC          |‰ ‰|j"        d„ |D ¦   «         ¦  «        }tG          ‰ ‰|j"        |¦  «        S )Nc                 ó:   •— g | ]\  }}t          ‰‰||¦  «        ‘ŒS r    )rØ   )rh   r  Útyargr5   r4   s      €€r$   rj   z&numpy_ufunc_kernel.<locals>.<listcomp>í  s;   ø€ ð 8ð 8ð 8Ù�S˜%õ # 7¨G°S¸%Ñ@Ô@ð 8ð 8ð 8r'   z+Not enough inputs to {}, expected {} got {}c                 ó   — g | ]	}|j         ‘Œ
S r    ©r7   ©rh   r¾   s     r$   rj   z&numpy_ufunc_kernel.<locals>.<listcomp>  s   € Ð&Ð&Ð&˜ˆŒÐ&Ð&Ð&r'   c                 ó   — g | ]	}|j         ‘Œ
S r    r#  r$  s     r$   rj   z&numpy_ufunc_kernel.<locals>.<listcomp>	  s   € Ð%Ð%Ð%˜ˆŒÐ%Ð%Ð%r'   c                 ó6   — g | ]}|                      ¦   «         ‘ŒS r    ©rI   ©rh   rÓ   s     r$   rj   z&numpy_ufunc_kernel.<locals>.<listcomp>  s$   € Ð;Ð;Ð;¨Sˆs×&Ò&Ñ(Ô(Ð;Ð;Ð;r'   r   c                 óF   — g | ]}t          |t          ¦  «        ¯|j        ‘ŒS r    )rÅ   rl   rz   r(  s     r$   rj   z&numpy_ufunc_kernel.<locals>.<listcomp>  s8   € ð 7ð 7ð 7 CÝ" 3­Ñ5Ô5ð7�S”Zð 7ð 7ð 7r'   c                 ó   — g | ]
}|d k    ¯|‘ŒS ©ÚCr    rå   s     r$   rj   z&numpy_ufunc_kernel.<locals>.<listcomp>  ó   € Ð=Ð=Ð=˜a°A¸²H°H˜°H°H°Hr'   c                 ó   — g | ]
}|d k    ¯|‘ŒS ©ÚFr    rå   s     r$   rj   z&numpy_ufunc_kernel.<locals>.<listcomp>  r-  r'   r0  r,  ©r9   Úorderc                 ó   — g | ]	}|j         ‘Œ
S r    )rS   )rh   Úos     r$   rj   z&numpy_ufunc_kernel.<locals>.<listcomp>-  s   € ÐHgÐHgÐHgÐZ[ÈÌÐHgÐHgÐHgr'   )$r]   r±   r\   ÚninÚRuntimeErrorrÑ   r,   rò   r  rÅ   r   rÈ   r  rØ   r:   r;   r8   rq   Ú
enable_nrtÚnrtÚincrefr  r   r9   r<   r   Ú	loop_nestr%   rM   r*   r  ÚgeneraterP   r  r  r	   )r4   r5   r´   r±   r  Úkernel_classÚ	argumentsÚout_iÚret_tyÚoutputr  ÚoutputsÚ	outer_sigÚkernelrE   rL   r³   Úinput_layoutsÚnum_c_layoutÚnum_f_layoutr2  r"   Úvals_inrt   ri   r  Úvals_outÚval_outÚouts   ``                           r$   Únumpy_ufunc_kernelrK  á  s	  øø€ ð8ð 8ð 8ð 8ð 8Ý#& t¨S¬XÑ#6Ô#6ð8ñ 8ô 8€Iõ ˆ9�~„~˜œ	Ò!Ð!ÝØ9ßŠV�E”N E¤I­s°9©~¬~Ñ>Ô>ñ@ô @ð 	@õ #Õ#7¸¸sÑ#CÔ#CÑDÔDð Ið I‰ˆˆvØŒ9�uÑ¥ I¡¤Ò.Ð.å˜&¥%Ô"7Ñ8Ô8ð OÝ% g¨w¸ÀÄÈ)ÑTÔT��å*Ø˜WÝ”K × 6Ò 6°vÑ >Ô >ÀÑEÔEÀvñOô O�ð ×Ò˜VÑ$Ô$Ð$Ð$ØÔð 	IàŒK×Ò˜w¨°°U´YÀÑ5FÔ0GÑHÔHÐHøà�z˜œ	�zÔ"€FØ˜œ	˜
˜
Ô#€GÝˆw‰<Œ<˜5œ:Ò%Ð%Ð%Ð%åØ&Ð&˜gÐ&Ñ&Ô&Ø%Ð%˜fÐ%Ñ%Ô%ñô €Ið ˆ\˜' 7¨IÑ6Ô6€FØ×#Ò#¥E¤JÑ/Ô/€Fà;Ð;°FÐ;Ñ;Ô;€Gð ˜”
Ô €Ið7ð 7¨6ð 7ñ 7ô 7€MåÐ=Ð= =Ð=Ñ=Ô=Ñ>Ô>€LÝÐ=Ð= =Ð=Ñ=Ô=Ñ>Ô>€Lð �lÒ"Ð"Øˆˆàˆå	Ô	˜7 I°FÀ%Ð	HÑ	HÔ	Hð 5ÈLØˆÝ(­¨W°fÑ)=Ô)=Ñ>Ô>ð 	=ð 	=‰OˆA‰|��sØ× Ò  ¨qÑ1Ô1Ð1Ø�NŠN˜3Ÿ=š=¨¯ªÑ):Ô):Ñ;Ô;Ñ<Ô<Ð<Ð<å(¨°¸¸¼È'Ð9RÑSÔSˆÝ" 8¨WÑ5Ô5ð 	5ð 	5‰OˆG�VØ×Ò˜l¨GÑ4Ô4Ð4Ð4ð	5ð5ð 5ð 5ñ 5ô 5ð 5ð 5ð 5ð 5ð 5ð 5øøøð 5ð 5ð 5ð 5õ ˜e W¨g°s´ÐHgÐHgÐ_fÐHgÑHgÔHgÑ
hÔ
h€CÝ˜G W¨c¬o¸sÑCÔCÐCs   É*B>L5Ì5L9Ì<L9c                 óÖ  — g }|j                              ¦   «         }|d         |d         z   }dg|j        z  dg|j        z  z   }t	          ||j        ||¦  «        D �]\  }	}
}}t          |
t          j        ¦  «        r¹ |  	                    |
¦  «        | ||	¦  «        }t          j        ||j        |
j        ¦  «        }t          j        ||j        |
j        ¦  «        }|
                     |¬¦  «        }|
j        }|
j        }|
j        }t%          | ||||	|||||¦
  «
        }|                     |¦  «         ŒÛt)          | ||	|
¦  «        }|                     |¦  «         �Œ || ||¦  «        }d„ |D ¦   «         }t+          d„ |D ¦   «         ¦  «        }t+          d„ |D ¦   «         ¦  «        }||k    rd	}nd
}||j        d …         }|                      t          j        ¦  «        }d„ |D ¦   «         }|d         j        |d         j        j        z
  }|d         j        d |…         }t3          |j        j        ¦  «        } t9          j        t	          t=          t+          |¦  «        ¦  «        | d         | d         z   ¦  «        d¬¦  «        D ]o\  \  }!}"\  }#}$||!         ||#         }&}%|"|$k    rNt?          d„ |%|&fD ¦   «         ¦  «        r3||!         ||#         }&}%|%                      |&t+          |"¦  «        ¦  «         Œp|d |j        …         D ],}	t          |	t$          ¦  «        r|	 !                    |¦  «         Œ-t          j"        ||||¬¦  «        5 }'g }(tG          t	          ||¦  «        ¦  «        D ]X\  })\  }*}	|* $                    |'|)¦  «         |(                     |	 %                    |* &                    ¦   «         ¦  «        ¦  «         ŒY |j'        |(Ž  d d d ¦  «         d S # 1 swxY w Y   d S )Nr   r3   TF)rp   c                 óF   — g | ]}t          |t          ¦  «        ¯|j        ‘ŒS r    )rÅ   rŠ   rz   )rh   r  s     r$   rj   z'numpy_gufunc_kernel.<locals>.<listcomp>K  s8   € ð 3ð 3ð 3˜cÝ˜S¥.Ñ1Ô1ð3ˆsŒzð 3ð 3ð 3r'   c                 ó   — g | ]
}|d k    ¯|‘ŒS r+  r    rå   s     r$   rj   z'numpy_gufunc_kernel.<locals>.<listcomp>M  ó   € Ð7Ð7Ð7˜a¨a°3ªh¨h˜¨h¨h¨hr'   c                 ó   — g | ]
}|d k    ¯|‘ŒS r/  r    rå   s     r$   rj   z'numpy_gufunc_kernel.<locals>.<listcomp>N  rO  r'   r0  r,  c                 ó6   — g | ]}|                      ¦   «         ‘ŒS r    r'  r(  s     r$   rj   z'numpy_gufunc_kernel.<locals>.<listcomp>[  s$   € Ð>Ð>Ð>¨Sˆs×&Ò&Ñ(Ô(Ð>Ð>Ð>r'   é   )Úrc              3   ó@   K  — | ]}t          |t          ¦  «        V — Œd S r   )rÅ   rŠ   rå   s     r$   rë   z&numpy_gufunc_kernel.<locals>.<genexpr>i  s,   è è € ÐJÐJ°a•J˜q¥.Ñ1Ô1ÐJÐJÐJÐJÐJÐJr'   r1  )(ÚdufuncÚexpected_ndimsr5  r  r]   r±   rÅ   r   rÈ   r�   r   rÉ   r<   rp   ry   Úcopyrz   rÊ   rŠ   rq   r0   r\   r8   r9   rŒ   r   Úgufunc_builderÚ	signatureÚ	itertoolsÚcombinationsro   ÚallrÁ   r¶   r:  rò   r%   rM   r*   r;  )+r4   r5   r´   r±   r  r<  r=  rV  Úis_inputr  rD   Úexp_ndimÚis_inpr‘   r<   ry   rŒ   rp   rz   r7   Úarray_helperÚscalar_helperrC  ÚlayoutsrE  rF  r2  rA  rE   rL   Úloopshape_ndimr³   Ú_sigÚidx_aÚsig_aÚidx_bÚsig_bÚarg_aÚarg_br"   rG  rt   ri   s+                                              r$   Únumpy_gufunc_kernelrk  1  s~  € Ø€IØ!Ô(×7Ò7Ñ9Ô9€NØ# AÔ&¨¸Ô):Ñ:€NØˆv˜œ	Ñ! U G¨e¬jÑ$8Ñ8€HÝ%(¨¨s¬x¸ÈÑ%RÔ%Rð ,ñ ,Ñ!ˆˆR�˜6Ý�b�%Ô/Ñ0Ô0ð 	,ð )�'×$Ò$ RÑ(Ô(¨°'¸3Ñ?Ô?ˆCÝÔ(¨°#´)¸R¼WÑEÔEˆEÝÔ*¨7°C´KÀÄÑIÔIˆGØŸ7š7¨˜7Ñ1Ô1ˆLØ”7ˆDØ”YˆFØœˆIÝ)¨'°7Ø*/°¸#Ø*0°)¸TØ*6¸ñ@ô @ˆLð ×Ò˜\Ñ*Ô*Ð*Ð*å)¨'°7¸CÀÑDÔDˆMØ×Ò˜]Ñ+Ô+Ð+Ñ+Øˆ\˜' 7¨CÑ0Ô0€Fð3ð 3 Yð 3ñ 3ô 3€GåÐ7Ð7 7Ð7Ñ7Ô7Ñ8Ô8€LÝÐ7Ð7 7Ð7Ñ7Ô7Ñ8Ô8€Lð �lÒ"Ð"Øˆˆàˆà˜œ	˜
˜
Ô#€GØ×#Ò#¥E¤JÑ/Ô/€FØ>Ð>°IÐ>Ñ>Ô>€GØ˜Q”Z”_ w¨q¤zÔ'>Ô'CÑC€NØ˜”
Ô Ð"2 NÐ"2Ô3€Iå˜5Ô/Ô9Ñ:Ô:€DÝ*3Ô*@Ý••c˜)‘n”nÑ%Ô%Ø�ŒG�d˜1”gÑñô àð+ñ +ô +ð ;ð ;Ñ&‰ˆ�™˜ ð ! Ô'¨°5Ô)9ˆuˆØ�EŠ>ˆ>ÝÐJÐJ¸EÀ5¸>ÐJÑJÔJÑJÔJð à$ UÔ+¨Y°uÔ-=�5ˆEØ×'Ò'¨­s°5©z¬zÑ:Ô:Ð:øà˜˜%œ)˜Ô$ð 'ð 'ˆÝ�c�>Ñ*Ô*ð 	'Ø�OŠO˜IÑ&Ô&Ð&øå	Ô	˜7Ø$Ø &Ø!&ð
(ñ 
(ô 
(ð 	"ð ,8ØˆÝ(­¨W°iÑ)@Ô)@ÑAÔAð 	=ð 	=‰OˆA‰|��sØ× Ò  ¨qÑ1Ô1Ð1Ø�NŠN˜3Ÿ=š=¨¯ªÑ):Ô):Ñ;Ô;Ñ<Ô<Ð<Ð<àˆŒ˜Ð!Ð!ð	"ð 	"ð 	"ñ 	"ô 	"ð 	"ð 	"ð 	"ð 	"ð 	"ð 	"ð 	"øøøð 	"ð 	"ð 	"ð 	"ð 	"ð 	"s   ÍBOÏO"Ï%O"c                   ó    — e Zd Zd„ Zd„ Zd„ ZdS )Ú_Kernelc                 ó0   — || _         || _        || _        d S r   )r4   r5   rB  )r!   r4   r5   rB  s       r$   rG   z_Kernel.__init__  s   € ØˆŒØˆŒØ"ˆŒˆˆr'   c                 ó(  — t          |t          j        ¦  «        rWt          |t          j        ¦  «        s=|j        }| j                             |d¦  «        } || j        | j        ||d¦  «        }|}| j                             | j        |||¦  «        S )zùNumpy uses cast semantics that are different from standard Python
        (for example, it does allow casting from complex to float).

        This method acts as a patch to context.cast so that it allows
        complex to real/int casts.

        Úreal)rÅ   r   ÚComplexÚunderlying_floatr4   Úget_getattrr5   rÇ   )r!   r6   r�   rž   ÚnewtyÚattrs         r$   rÇ   z_Kernel.cast„  s�   € õ �v�uœ}Ñ-Ô-ð 	Ý˜4¥¤Ñ/Ô/ð	ð Ô+ˆEØ”<×+Ò+¨F°FÑ;Ô;ˆDØ�$�t”| T¤\°6¸3ÀÑGÔGˆCØˆFð Œ|× Ò  ¤¨s°F¸DÑAÔAÐAr'   c                 óÆ  ‡ — ‰ j         }‰ j        }ˆ fd„t          ||j        |j        ¦  «        D ¦   «         }‰ j        j        rK‰ j        j                             t          j
        t          j
        gt          |j        ¦  «        z  ¦  «        }n*‰ j        j                             |j        |j        ¦  «        }‰ j        j        j        j        }t#          j        ||‰ j        j        j        ¦  «        }|j                             d¦  «         ‰ j        j                             ‰ j        ||j        |j        |¦  «        \  }}	‰                      |	|j        |j        ¦  «        S )Nc                 óF   •— g | ]\  }}}‰                      |||¦  «        ‘ŒS r    ©rÇ   ©rh   r6   ÚintyÚouttyr!   s       €r$   rj   z$_Kernel.generate.<locals>.<listcomp>›  s?   ø€ ð 6ð 6ð 6Ù)˜˜d Eð —Y’Y˜s D¨%Ñ0Ô0ð 6ð 6ð 6r'   Úalwaysinline)Ú	inner_sigrB  r]   r±   ÚcresÚ
objectmoder4   rû   Úget_function_typer   Úpyobjectr\   r  r5   ÚblockÚfunctionÚmoduler   Úget_or_insert_functionÚfndescÚllvm_func_nameÚ
attributesÚaddÚcall_functionrÇ   )
r!   r±   ÚisigÚosigÚ	cast_argsÚ	func_typer„  Úentry_pointÚ_r£   s
   `         r$   r;  z_Kernel.generate˜  sO  ø€ ØŒ~ˆØŒ~ˆð6ð 6ð 6ð 6å˜˜tœy¨$¬)Ñ4Ô4ð6ñ 6ô 6ˆ	ð Œ9Ôð 	-ØœÔ.×@Ò@Ý”¥¤Ð 0µ3°t´y±>´>Ñ AñCô CˆIˆIð œÔ.×@Ò@ØÔ  $¤)ñ-ô -ˆIà”Ô#Ô,Ô3ˆÝÔ4Ø�IØŒIÔÔ+ñ-ô -ˆð 	Ô×"Ò" >Ñ2Ô2Ð2à”Ô'×5Ò5ØŒL˜+ tÔ'7¸¼Øñô ‰ˆˆ3ð �yŠy˜˜dÔ.°Ô0@ÑAÔAÐAr'   N)r,   r-   r.   rG   rÇ   r;  r    r'   r$   rm  rm  ~  sI   € € € € € ð#ð #ð #ð
Bð Bð Bð(Bð Bð Bð Bð Br'   rm  c                 ó6   ‡ ‡—  G ˆˆ fd„dt           ¦  «        Š‰S )a  Use the ufunc loop type information to select the code generation
    function from the table provided by the dict_of_kernels. The dict
    of kernels maps the loop identifier to a function with the
    following signature: (context, builder, signature, args).

    The loop type information has the form 'AB->C'. The letters to the
    left of '->' are the input types (specified as NumPy letter
    types).  The letters to the right of '->' are the output
    types. There must be 'ufunc.nin' letters to the left of '->', and
    'ufunc.nout' letters to the right.

    For example, a binary float loop resulting in a float, will have
    the following signature: 'ff->f'.

    A given ufunc implements many loops. The list of loops implemented
    for a given ufunc can be accessed using the 'types' attribute in
    the ufunc object. The NumPy machinery selects the first loop that
    fits a given calling signature (in our case, what we call the
    outer_sig). This logic is mimicked by 'ufunc_find_matching_loop'.
    c                   ó*   •‡ — e Zd Zˆˆ ˆfd„Zd„ Zˆ xZS )ú'_ufunc_db_function.<locals>._KernelImplc           
      ó¼  •— t          ‰| ¦  «                             |||¦  «         t          ‰|j        t	          t          ‰|¦  «        ¦  «        z   ¦  «        }|                     ‰¦  «                             |j        ¦  «        | _	        t          |j        |j        ¦  «        | _        | j	        €*d}t          |                     ‰j        |¦  «        ¦  «        ‚d S )Nz2Don't know how to lower ufunc '{0}' for loop '{1}')ÚsuperrG   r   r±   rñ   r  Úget_ufunc_inforø   Ú	ufunc_sigÚfnr   rA  r  r}  rÐ   rÑ   r,   )	r!   r4   r5   rB  ÚlooprÀ   Ú_KernelImplÚ	__class__r  s	         €€€r$   rG   z0_ufunc_db_function.<locals>._KernelImpl.__init__Ç  sÀ   ø€ Ý�+˜tÑ$Ô$×-Ò-¨g°wÀ	ÑJÔJÐJÝ+Ø�y”~­Õ.BÀ5È)Ñ.TÔ.TÑ(UÔ(UÑUñWô WˆDà×,Ò,¨UÑ3Ô3×7Ò7¸¼ÑGÔGˆDŒGÝ,¨T¬\¸4¼;ÑGÔGˆDŒNàŒwˆØJ�Ý)¨#¯*ª*°U´^ÀTÑ*JÔ*JÑKÔKÐKð ˆr'   c                 ó²  ‡ — ‰ j         }‰ j        }ˆ fd„t          ||j        |j        ¦  «        D ¦   «         }t	          ‰ j        d¦  «        5  ‰                      ‰ j        ‰ j        ||¦  «        }d d d ¦  «         n# 1 swxY w Y   ‰ j        j        }||j	                  
                    ‰ j        |¦  «        }‰                      ||j	        |j	        ¦  «        S )Nc                 óF   •— g | ]\  }}}‰                      |||¦  «        ‘ŒS r    rx  ry  s       €r$   rj   zD_ufunc_db_function.<locals>._KernelImpl.generate.<locals>.<listcomp>Ö  sE   ø€ ð Að Að AÙ-˜S $¨ð Ÿš 3¨¨eÑ4Ô4ð Að Að Ar'   Únumpy)r}  rB  r]   r±   r
   r4   r˜  r5   r   r  Úfrom_returnrÇ   )r!   r±   r‹  rŒ  r�  r£   Údmms   `      r$   r;  z0_ufunc_db_function.<locals>._KernelImpl.generateÒ  s1  ø€ Ø”>ˆDØ”>ˆDðAð Að Að AÝ14°T¸4¼9Ø59´Yñ2@ô 2@ðAñ Aô AˆIõ # 4¤<°Ñ9Ô9ð Kð KØ—g’g˜dœl¨D¬L¸$À	ÑJÔJ�ðKð Kð Kñ Kô Kð Kð Kð Kð Kð Kð Køøøð Kð Kð Kð Kà”,Ô1ˆCØ�dÔ&Ô'×3Ò3°D´LÀ#ÑFÔFˆCØ—9’9˜S $Ô"2°DÔ4DÑEÔEÐEs   Á#A;Á;A?ÂA?)r,   r-   r.   rG   r;  Ú__classcell__)r›  rš  r  s   @€€r$   rš  r“  Æ  s`   øø€ € € € € ð		Lð 		Lð 		Lð 		Lð 		Lð 		Lð 		Lð	Fð 	Fð 	Fð 	Fð 	Fð 	Fð 	Fr'   rš  )rm  )r  rš  s   `@r$   Ú_ufunc_db_functionr¢  °  sP   øø€ ð,Fð Fð Fð Fð Fð Fð Fð F•gñ Fô Fð Fð2 Ðr'   c                 óÀ   ‡ ‡— ˆˆ fd„}t           j        }|f‰ j        z  }t          ‰ j        dz   ¦  «        D ]&}t           j        f|z  }  |‰ g|¢|¢R Ž |¦  «         Œ'‰S )Nc                 ó,   •— t          | |||‰‰¦  «        S r   ©rK  ©r4   r5   r´   r±   rC  r  s       €€r$   Údo_ufuncz'register_ufunc_kernel.<locals>.do_ufuncæ  ó   ø€ Ý! '¨7°C¸¸uÀfÑMÔMÐMr'   r3   )r   ÚAnyr5  ro   r  rŽ   )r  rC  Úlowerr§  Ú_anyÚin_argsÚn_explicit_outÚout_argss   ``      r$   Úregister_ufunc_kernelr¯  å  s›   øø€ ðNð Nð Nð Nð Nð Nõ Œ9€DØˆg˜œ	Ñ!€Gõ   ¤
¨Q¡Ñ/Ô/ð 4ð 4ˆÝ”K�> NÑ2ˆØ)ˆˆˆeÐ)�gÐ) Ð)Ð)Ð)¨(Ñ3Ô3Ð3Ð3à€Mr'   Fc                 ó`   ‡‡— |rJ ‚ˆˆfd„}t           j        }  || |¦  «        |¦  «         d S )Nc                 ó,   •— t          | |||‰‰¦  «        S r   r¥  r¦  s       €€r$   Úlower_unary_operatorz<register_unary_operator_kernel.<locals>.lower_unary_operator÷  r¨  r'   )r   rŽ   )Úoperatorr  rC  rª  Úinplacer²  Ú	_arr_kinds    ``    r$   Úregister_unary_operator_kernelr¶  ô  s]   øø€ àÐÐˆ;ðNð Nð Nð Nð Nð Nå”€IØ€E€Eˆ(�IÑÔÐ3Ñ4Ô4Ð4Ð4Ð4r'   c                 óÀ   ‡‡— ˆˆfd„}ˆˆfd„}t           j        }t           j        }||f||f||fg}	|	D ])}
|s  || g|
¢R Ž |¦  «         Œ  || g|
¢R Ž |¦  «         Œ*d S )Nc                 ó,   •— t          | |||‰‰¦  «        S r   r¥  r¦  s       €€r$   Úlower_binary_operatorz>register_binary_operator_kernel.<locals>.lower_binary_operatorþ  r¨  r'   c                 ó´   •— t          |¦  «        |d         fz   }t          j        |j        g|j        |j        d         fz   ¢R Ž }t          | |||‰‰¦  «        S rn   )rñ   r   rY  r  r±   rK  r¦  s       €€r$   Úlower_inplace_operatorz?register_binary_operator_kernel.<locals>.lower_inplace_operator  s\   ø€ õ �T‰{Œ{˜d 1œg˜ZÑ'ˆÝÔ˜sœÐK°´¸S¼XÀa¼[¸NÑ1JÐKÐKÐKˆÝ! '¨7°C¸¸uÀfÑMÔMÐMr'   )r   r©  rŽ   )Úopr  rC  rª  r´  r¹  r»  r«  rµ  Úformal_sigsr´   s    ``        r$   Úregister_binary_operator_kernelr¾  ý  sÜ   øø€ ðNð Nð Nð Nð Nð NðNð Nð Nð Nð Nð Nõ Œ9€DÝ”€IØ˜yÐ)¨D°)Ð+<¸yÈ$Ð>OÐP€KØð 4ð 4ˆØð 	4ØˆEˆE�"ˆO�sˆOˆOˆOÐ1Ñ2Ô2Ð2Ð2àˆEˆE�"ˆO�sˆOˆOˆOÐ2Ñ3Ô3Ð3Ð3ð	4ð 4r'   c                 ód   —  G d„ dt           ¦  «        }t          | |||t          j        |¦  «        S )z¸Lowering function for +(array) expressions.  Defined here
    (numba.targets.npyimpl) since the remaining array-operator
    lowering functions are also registered in this module.
    c                   ó   — e Zd Zd„ ZdS )ú1array_positive_impl.<locals>._UnaryPositiveKernelc                 ó   — |\  }|S r   r    )r!   r±   r6   s      r$   r;  z:array_positive_impl.<locals>._UnaryPositiveKernel.generate  s   € Ø‰EˆSØˆJr'   N)r,   r-   r.   r;  r    r'   r$   Ú_UnaryPositiveKernelrÁ    s#   € € € € € ð	ð 	ð 	ð 	ð 	r'   rÃ  )rm  rK  ÚnpÚpositive)r4   r5   r´   r±   rÃ  s        r$   Úarray_positive_implrÆ    sO   € ðð ð ð ð �wñ ô ð õ
 ˜g w°°T½2¼;Ø2ñ4ô 4ð 4r'   c           	      óÚ  — i }| D ]%}t          |¦  «        }t          |||¦  «        ||<   Œ&t          j        j        t          j        j        fD ]ƒ}|                     ¦   «         D ]l\  }}t          t          |¦  «        }||         }|j	        dk    rt          ||||¦  «         Œ@|j	        dk    rt          ||||¦  «         Œ^t          d¦  «        ‚Œ„t          j        j        fD ]‡}|                     ¦   «         D ]p\  }}t          t          |¦  «        }||         }|j	        dk    rt          ||||d¬¦  «         ŒB|j	        dk    rt          ||||d¬¦  «         Œbt          d¦  «        ‚Œˆd S )Nr3   rR  z4There shouldn't be any non-unary or binary operatorsT)r´  )r¢  r¯  r   ÚNumpyRulesUnaryArrayOperatorÚ_op_mapÚNumpyRulesArrayOperatorÚitemsÚgetattrrÄ  r5  r¶  r¾  r6  ÚNumpyRulesInplaceArrayOperator)	Úufuncsrª  Úkernelsr  Údb_funcrÉ  r³  Ú
ufunc_namerC  s	            r$   Úregister_ufuncsrÒ  %  sÍ  € Ø€GØð Fð FˆÝ$ UÑ+Ô+ˆÝ.¨u°g¸uÑEÔEˆ�‰ˆåÔ8Ô@ÝÔ3Ô;ðð [ð [ˆð %,§M¢M¡O¤Oð 	[ð 	[Ñ ˆH�jÝ�B 
Ñ+Ô+ˆEØ˜U”^ˆFØŒy˜AŠ~ˆ~Ý.¨x¸ÀÈÑNÔNÐNÐNØ”˜a’�Ý/°¸%ÀÈÑOÔOÐOÐOå"Ð#YÑZÔZÐZð	[õ Ô:ÔBð ð [ð [ˆà$+§M¢M¡O¤Oð 
	[ð 
	[Ñ ˆH�jÝ�B 
Ñ+Ô+ˆEØ˜U”^ˆFØŒy˜AŠ~ˆ~Ý.¨x¸ÀÈØ7;ð=ñ =ô =ð =ð =à”˜a’�Ý/°¸%ÀÈØ8<ð>ñ >ô >ð >ð >õ #Ð#YÑZÔZÐZð
	[ð[ð [r'   c                 óf  ‡— ˆfd„}t          ‰t          j        ¦  «        r1t          t	          j        ‰j        ¦  «        ¦  «        } ||¦  «        S t          ‰t          j        j        ¦  «        r@t          ‰j        ¦  «        }t          t	          j        |¦  «        ¦  «        } ||¦  «        S dS )zPGiven a string or NumberClass description *desc*, returns the dtype object.
    c                 óP   •— t          j        | ¦  «        } |‰¦  «        }d„ }||fS )Nc                 ó*   — |                       ¦   «         S r   )Úget_dummy_value)r4   r5   rY  r±   s       r$   Úcodegenz9_make_dtype_object.<locals>.from_nb_type.<locals>.codegenR  s   € ð ×*Ò*Ñ,Ô,Ð,r'   )r   ÚDType)Únb_typer  r´   r×  Údescs       €r$   Úfrom_nb_typez(_make_dtype_object.<locals>.from_nb_typeN  s<   ø€ Ý”k 'Ñ*Ô*ˆØˆk˜$ÑÔˆð	-ð 	-ð 	-ð
 �Gˆ|Ðr'   N)
rÅ   r   ÚLiteralr   rÄ  rÊ   Úliteral_valueÚ	functionsÚNumberClassrÒ   )Ú	typingctxrÚ  rÛ  rÙ  Úthestrs    `   r$   Ú_make_dtype_objectrâ  J  s±   ø€ ð	ð 	ð 	ð 	ð 	õ �$�œÑ&Ô&ð %å�RœX dÔ&8Ñ9Ô9Ñ:Ô:ˆØˆ|˜GÑ$Ô$Ð$Ý	�D�%œ/Ô5Ñ	6Ô	6ð %Ý�T”Z‘”ˆå�RœX fÑ-Ô-Ñ.Ô.ˆØˆ|˜GÑ$Ô$Ð$ð	%ð %r'   c                 ó°   — t          | t          j        t          j        j        f¦  «        rd„ }|S t          j        d                     | ¦  «        ¦  «        ‚)zKProvide an implementation so that numpy.dtype function can be lowered.
    c                 ó    — t          | ¦  «        S r   )râ  )rÊ   s    r$   Úimpznumpy_dtype.<locals>.imph  s   € Ý% eÑ,Ô,Ð,r'   zunknown dtype descriptor: {})rÅ   r   rÜ  rÞ  rß  r   ÚNumbaTypeErrorrÑ   )rÊ   rå  s     r$   Únumpy_dtyperç  c  s\   € õ �%�%œ-­¬Ô)DÐEÑFÔFð :ð	-ð 	-ð 	-àˆ
åÔ#Ø*×1Ò1°%Ñ8Ô8ñ:ô :ð 	:r'   )rÃ   )F)KrT   ÚmathÚsysrZ  Úcollectionsr   Úllvmlite.irr:   rž  rÄ  r³  Únumba.npr   r   r   Únumba.np.ufunc.sigparser   Únumba.core.imputilsr   r	   r
   r   Ú
numba.corer   r   r   r   r   r   Únumba.np.numpy_supportr   r   r   r   Únumba.np.arrayobjr   Únumba.core.typingr   Únumba.core.extendingr   r   r   rÍ   ÚregistryÚobjectr   r0   rW   rl   rŠ   rØ   r9   ÚCPointerrô   râ   r  r  r  r  rK  rk  rm  r¢  r¯  r¶  r¾  rª  ÚposrŽ   rÆ  rÒ  Ú
get_ufuncsrâ  rÊ   rç  r    r'   r$   ú<module>rù     s¾  ððð ð
 €€€Ø 
€
€
€
Ø Ð Ð Ð Ø "Ð "Ð "Ð "Ð "Ð "à Ð Ð Ð Ð Ð à Ð Ð Ð Ø €€€à 6Ð 6Ð 6Ð 6Ð 6Ð 6Ð 6Ð 6Ð 6Ð 6Ø 3Ð 3Ð 3Ð 3Ð 3Ð 3Ø bÐ bÐ bÐ bÐ bÐ bÐ bÐ bÐ bÐ bÐ bÐ bØ FÐ FÐ FÐ FÐ FÐ FÐ FÐ FÐ FÐ FÐ FÐ FÐ FÐ FÐ FÐ Fðð ð ð ð ð ð ð ð ð ð ð ð 5Ð 4Ð 4Ð 4Ð 4Ð 4Ø %Ð %Ð %Ð %Ð %Ð %Ø 4Ð 4Ð 4Ð 4Ð 4Ð 4Ð 4Ð 4à Ð Ð Ð Ð Ð Ø 'Ð 'Ð 'Ð 'Ð 'Ð 'àˆ8�IÑÔ€ðð ð ð ð ˜Fñ ô ð ð&,ð &,ð &,ð &,ð &,�Fñ &,ô &,ð &,ðR;ð ;ð ;ð ;ð ;˜:˜:Ð&<Ø&:ñ<ô <ñ ;ô ;ð ;ð6&Fð &Fð &Fð &Fð &F�:�:˜nð /=ñ >ô >ñ &Fô &Fð &FðRL@ð L@ð L@ð L@ð L@�Z�Z ð 1Qñ Rô Rñ L@ô L@ð L@ð^/ð /ð /ð /ð6 !�e”j ¤¨^¨U¬^¸E¼JÑ-GÔ-GØ %¤
¨N¨E¬N¸5¼:Ñ,FÔ,FñHô HÐ ð&ð &ð &ðPQ!ð Q!ð Q!ðj%ð %ð %ð5ð 5ð 5ð8ð 8ð 8ðMDð MDð MDð`I"ð I"ð I"ðZ/Bð /Bð /Bð /Bð /Bˆfñ /Bô /Bð /Bðd/ð /ð /ðjð ð ð  ,1ð5ð 5ð 5ð 5ð4ð 4ð 4ð 4ð2 
‡‚�”˜eœkÑ*Ô*ð4ð 4ñ +Ô*ð4ð[ð [ð [ðD €Ð#�Ô#Ñ%Ô% x¤~Ñ 6Ô 6Ð 6ð ð%ð %ñ „ð%ð0 
€ˆ"Œ(ÑÔð	:ð 	:ñ Ôð	:ð 	:ð 	:r'   