
    0Fie                    ,   d dl mZ d dlZd dlmZ d dlmZ d dlmZm	Z	 ej
                            d          Zej
                            d          Zd Zd	 Z e	d
ee            ed
          Z e                    e          ej                   dS )    )annotationsN)typename)pickle)ObjectDictSerializerregister_serialization_familycuda_serializecuda_deserializec                V   t          t          |                     }	 t                              t          |                     }n# t          $ r t          |          w xY w ||           \  }}|t          j        t          |                     ddt          |          z  d}||fS )Ncuda)F)
sub-headertype-serialized
serializercompression)	r   typer   dispatch	TypeErrorNotImplementedErrorr   dumpslen)x	type_namer   
sub_headerframesheaders         9lib/python3.11/site-packages/distributed/protocol/cuda.py
cuda_dumpsr      s    a!!I-''Q00 - - -!),,,- qJ !<Q00#f++-	 F 6>s   'A A c                    t          j        | d                   }t                              |          } || d         |          S )Nr   r   )r   loadsr	   r   )r   r   typr   s       r   
cuda_loadsr    !   sA    
,v/0
1
1C%%c**E5%v...    r   )
__future__r   dask
dask.utilsr   distributed.protocolr   distributed.protocol.serializer   r   utilsDispatchr   r	   r   r     cuda_object_with_dict_serializerregisterdictdeserialize r!   r   <module>r.      s   " " " " " "        ' ' ' ' ' '       
 $$%566:&&'9::   "/ / /  fj* = = = $8#7#?#?      $   @ L M M M M Mr!   