U yh`@sTUddlZddlmZddlZddlmZdaeeed<ejddZ ddZ dS) N)Optional)ContentStoreReaderLOAD_TENSOR_READERccs.tdks tt|ddaz dVW5daXdS)NF)cache)rAssertionErrorr)locrJ/var/www/html/venv/lib/python3.8/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 BackendSelectcSstdkr"ddlm}|||||Sddlm}tj||d}t||ksbt|d|t| |kst| d||j |kst|j d||j |kr|||d}|SdS)Nr) rand_strided) clone_input)devicez != )dtype) rZtorch._dynamo.testingr Ztorch._dynamo.utilsr Z read_tensorlistsizerstrider r)namerrrr r r rrrr load_tensor_factory"s  &&  z1register_debug_prims..load_tensor_factory)torchlibrarydefineimpl)rrrr register_debug_primss  r) contextlibtypingrrZtorch.utils._content_storerr__annotations__contextmanagerr rrrrr s