o ְi @sddlZddlZdddZdS)NcCs@|dur tj}|j}g}i}t|D]E}t||tjr[||}t |}||vr9d |||}d}n|||<|}d}d t t |j} t |j} ||| | |jj|gqd} d} d} d}tt|D]<}||}| t|dkr~t|d} | t|dkrt|d} | t|dkrt|d} |dr|t|d7}qjt|dkrtd| }td| }td| }d |d |d |d }td |d t|d tt|D]9}||}td |dd |t|dd|dd |t|dd |dd |t|dd |dqtd |dS)aPrint the CuPy arrays in the given dictionary. Prints out the name, shape, bytes and type of all of the ndarrays present in `vardict`. If there is no dictionary passed in or `vardict` is None then returns CuPy arrays in the globals() dictionary (all CuPy arrays in the namespace). Args: vardict : (None or dict) A dictionary possibly containing ndarrays. Default is globals() if `None` specified .. admonition:: Example >>> a = cupy.arange(10) >>> b = cupy.ones(20) >>> cupy.who() Name Shape Bytes Type =========================================================== a 10 80 int64 b 20 160 float64 Upper bound on total bytes = 240 >>> d = {'x': cupy.arange(2.0), ... 'y': cupy.arange(3.0), 'txt': 'Some str', ... 'idx':5} >>> cupy.who(d) Name Shape Bytes Type =========================================================== x 2 16 float64 y 3 24 float64 Upper bound on total bytes = 40 Nz{} ({})rz x  zName {} Shape {} Bytes {} Type z{} {} =z{} {} {} {} {} {} {}z' Upper bound on total bytes = {})sys _getframef_back f_globalssortedkeys isinstancecupyndarrayidformatjoinmapstrshapenbytesappenddtypenamerangelenintmaxprint)vardictframestacachervaridvnamestroriginalshapestrbytestrmaxnamemaxshapemaxbyte totalbyteskvalsp1sp2sp3prvalr6B/home/ubuntu/.local/lib/python3.10/site-packages/cupy/_misc/who.pywhost*           r8)N)r rr8r6r6r6r7s