a d@sRddlZzddlmZWney.dZYn0ddZGdddZddZdS) N)_cudnncCsb|dkrttjjS|dkr(ttjjS|dkrrrrrr __getstate__#szUnserializable.__getstate__cCs d|_dSr r )rstaterrr __setstate__(szUnserializable.__setstate__N)__name__ __module__ __qualname__rrrrrrrrr sr c Csdttj}|r|nd}||vs6||durp|dkrLtd||<n$ttj|||tjtdd||<||}|S)NZdesc_rcuda)Zself_tydevice) strtorchrZcurrent_devicerr Z_cudnn_init_dropout_stateZuint8r)ZdropoutZtrainZ dropout_seedZ dropout_stateZdropout_desc_nameZ dropout_pZ dropout_tsrrrinit_dropout_state,s   r)Z torch.cudarZtorch._Cr ImportErrorr r rrrrrs