o Y۷iV@s ddlZddlmZmZmZddlmZmZmZm Z m Z ddZ ddZ dd Z eejejeje eejejeje eejejeje eejd d Ze ejd d ZeejejddZeejejddZeejejeejejddZdS)N)typestypingcgutils) lower_cast lower_builtinlower_getattr_genericimpl_ret_untrackedlower_setattr_genericcCtjSN)rtrue_bitcontextbuildersigargsrI/home/ubuntu/vllm_env/lib/python3.10/site-packages/numba/core/optional.pyalways_return_true_impl rcCr r )r false_bitr rrralways_return_false_implrrc Csj|j\}}|\}}|tjkr||}}||}}|}|} |||| } |t|| j} t|||j | S)z/ Check if an Optional value is invalid ) rrnone make_helpernot_r as_bool_bitvalidr return_type) rrrrltyrtylvalrvalopt_typeopt_valoptresrrroptional_is_nones    r&cCs2|j}|||||}|||}||||||S)z? Optional.__getattr__ => redirect to the wrapped type. )typecast get_getattr)rrtypvalueattr inner_typevalimprrroptional_getattr.s r0c CsR|j\}}|\}}|j} ||||| }t|j| |} ||| } | |||fS)z? Optional.__setattr__ => redirect to the wrapped type. )rr'r(r signaturer get_setattr) rrrrr,basetyvaltytargetr. target_typenewsigr/rrroptional_setattr9s  r8c Cs|j|||d}t||j}|||}||W\}} |tj|_|||j|j|j|_Wdn1s:wY| tj |_t |jj|_Wdn1sXwYWd| SWd| S1stwY| S)a The handling of optional->optional cast must be special cased for correct propagation of None value. Given type T and U. casting of T? to U? (? denotes optional) should always succeed. If the from-value is None, the None value the casted value (U?) should be None; otherwise, the from-value is casted to U. This is different from casting T? to U, which requires the from-value must not be None. r+N) rrrrif_elser r(datar'rget_null_value _getvalue) rrfromtytotyr.optvalvalidbit outoptvalis_valid is_not_validrrroptional_to_optionalHs.        rEcCs:|tjkr |||jS|||||j}|||j|Sr )rrmake_optional_noner'r(make_optional_value)rrr>r?r.rrrany_to_optionalgs rHcCs|j|||d}t||j}|j||ddd|jf}|j|t |fWdn1s3wY| ||j |j|S)Nr9F)likelyzexpected %s, got None) rrrrif_thenrr' call_convreturn_user_exc TypeErrorr(r;)rrr>r?r.r@rAmsgrrroptional_to_anyps rO)operator numba.corerrrnumba.core.imputilsrrrrr rrr&is_rOptionalr0r8rEAnyrHBooleanrOrrrrs(