o X۷i @sUddlmZddlmZddlZddlmZddlZddlZej ddZ ej ddZ ej d d Z ej d d Z Gd ddZedBZdZded<edde Zedde Zedde Zedde ZdS)) annotations)CallableN)FinalcCsB|jdvr|t|jS|jdvr|tj Std|NbiufzUnsupported type: )kindtypenumpyiinfomininf RuntimeErrordtyperT/home/ubuntu/vllm_env/lib/python3.10/site-packages/cupyx/distributed/array/_modes.py _min_value_of s  rcCs@|jdvr|t|jS|jdvr|tjStd|r)rr r r maxr rrrrr _max_value_ofs   rcC |dS)Nrr rrrr_zero_of  rcCr)Nrrrrr_one_of%rrc@s8eZdZUded<ded<ded<ded<dd dZdS)_OpModezcupy._core._kernel.ufuncfuncz numpy.ufunc numpy_funcbool idempotentr identity_of func_namestrreturnNonecCsDztt||_tt||_Wn tytdw||_||_dS)NzNo such function exists) getattrcupyrr rAttributeErrorrr r!)selfr"r r!rrr__init__0s   z_OpMode.__init__N)r"r#r rr!rr$r%)__name__ __module__ __qualname____annotations__r*rrrrr*s rz Final[None]REPLICAminimumTmaximumaddFmultiply) __future__rcollections.abcr functoolstypingrr r' lru_cacherrrrrModer/r.MINMAXSUMPRODrrrrs*