+ j):^RIt^RIt^RIHt^RIHtRRltR#)N)CustomClassBase)is_traceable_wrapper_subclasscnV^8dQhR\PR\\P,/#)treturn)torchTensorsetUntypedStorage)formats"f/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/distributed/_tools/common_utils.py __annotate__rs)#%#%ELL#%S1E1E-F#%c@V.p\4p\V4^8EdVP4p\V4'dVP 4wrEVFep\ W64;;\ PRc;ewpVPV4K;;\Rc;ewKNp\R\V4 24h K\VR4'g,\P!R\V4 2\^R7KVP!VP#44EKV#)z Recursively extracts untyped storages from a tensor or its subclasses. Args: t (torch.Tensor): The tensor to extract storages from. Returns: Set[torch.UntypedStorage]: A set of untyped storages. z(expected Tensor or CustomClassBase, got untyped_storagezEExpected a tensor or a traceable wrapper-subclass of tensor, but got )category stacklevel)r lenpopr__tensor_flatten__getattrr r appendrAssertionErrortypehasattrwarningswarn UserWarningaddr) runflattened_tensorsflattened_tensor_storagesobjattrs_attrv unexpecteds & rget_untyped_storagesr*s# # ! "Q &!%%' ( - ---/HEc(,,+2215-**+#,FtJGWFXY3 122 [\`ad\e[fg(  *--c.A.A.CD $$r)rr torch._custom_class_basertorch.utils._python_dispatchrr*rrrr-s 4F#%r