o Zha@sTUddlZddlmZddlZddlmZdaeeed<ejddZ ddZ dS) N)Optional)ContentStoreReaderLOAD_TENSOR_READERccs2tdusJt|ddazdVWdadSdaw)NF)cache)rr)locrG/var/www/auris/lib/python3.10/site-packages/torch/_prims/debug_prims.pyload_tensor_reader s   r cCs(tjddtjdddd}dS)Nzdebugprims::load_tensorzR(str name, int[] size, int[] stride, *, ScalarType dtype, Device device) -> TensorZ BackendSelectcSstdurddlm}|||||Sddlm}tj||d}t||ks1J|d|t||ksDJ|d||j |ksSJ|j d||j |kr^|||d}|S)Nr) rand_strided) clone_input)devicez != )dtype) rZtorch._dynamo.testingr Ztorch._dynamo.utilsr Z read_tensorlistsizestrider r )namerrr r r r rrrrload_tensor_factory#s  &&  z1register_debug_prims..load_tensor_factory)torchZlibraryZdefineimpl)rrrrregister_debug_primss  r) contextlibtypingrrZtorch.utils._content_storerr__annotations__contextmanagerr rrrrrs