jpUddlZddlmZddlZddlmZdaeeed<ejdZ dZ dS)N)Optional)ContentStoreReaderLOAD_TENSOR_READERc#ZKtJt|da dVdadS#dawxYw)NF)cache)rr)locs IC:\Projects\Historical\venv\Lib\site-packages\torch/_prims/debug_prims.pyload_tensor_readerr sU  % % %,Cu===" !T!!!!s&*ctjddtjddd}dS)Nzdebugprims::load_tensorzR(str name, int[] size, int[] stride, *, ScalarType dtype, Device device) -> Tensor BackendSelectctddlm}|||||Sddlm}t||}t ||ks!J|d|t ||ks!J|d||j |ksJ|j d||j |kr |||}|S)Nr) rand_strided) clone_input)devicez != )dtype) rtorch._dynamo.testingrtorch._dynamo.utilsr read_tensorlistsizestriderr)namerrrrrrrs r load_tensor_factoryz1register_debug_prims..load_tensor_factory#s+  % : : : : : :<feV<< < 7 7 7 7 7 7#..tF.CCA>>T)))affhh+B+BD+B+B))) ##v---!((**/J/J&/J/J---8v%%%!('@'@'@'@%%%w%K///H)torchlibrarydefineimpl)rs r register_debug_primsr s\ M!\  ]1?CCDCr) contextlibtypingrrtorch.utils._content_storerr__annotations__contextmanagerr r rr r's 99999948H/0777  " " " r