a ha@sTUddlZddlmZddlZddlmZdaeeed<ejddZ ddZ dS) N)Optional)ContentStoreReaderLOAD_TENSOR_READERccs2tdus Jt|ddazdVWdanda0dS)NF)cache)rr)locrF/var/www/auris/lib/python3.9/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 BackendSelectcSstdur"ddlm}|||||Sddlm}tj||d}t||ksbJ|d|t||ksJ|d||j |ksJ|j d||j |kr|||d}|SdS)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)torchlibraryZdefineimpl)rrrrregister_debug_primss  r) contextlibtypingrrZtorch.utils._content_storerr__annotations__contextmanagerr rrrrrs