o װi@sUddlZddlmZmZmZddlZddlZejddZejddZ ejddZ ejd d Z Gd d d Z ee Z dZeded <e dde Ze ddeZe dde Ze dde ZdS)N)CallableFinalOptionalcCsB|jdvr|t|jS|jdvr|tj Std|NbiufzUnsupported type: )kindtypenumpyiinfomininf RuntimeErrordtyperR/home/ubuntu/.local/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@sPeZdZUejjjed<ejed<e ed<e ed<de de de ddfdd Z dS) _OpModefunc numpy_func idempotent identity_of func_namereturnNcCsDztt||_tt||_Wn tytdw||_||_dS)NzNo such function exists) getattrcupyrr rAttributeErrorrrr )selfr!rr rrr__init__-s   z_OpMode.__init__) __name__ __module__ __qualname__r$_core_kernelufunc__annotations__r boolrstrr'rrrrr's  rREPLICAminimumTmaximumaddFmultiply) functoolstypingrrrr r$ lru_cacherrrrrModer1r.MINMAXSUMPRODrrrrs&