DŽgalUddlZddlmZddlZddlmZdaeeed<ejdZ dZ y)N)Optional)ContentStoreReaderLOAD_TENSOR_READERc#RKtJt|da dday#dawxYww)NF)cache)rr)locs `/home/mcse/projects/flask_80/flask-venv/lib/python3.12/site-packages/torch/_prims/debug_prims.pyload_tensor_readerr s7  %% %,Cu=" !Ts' '$'ctjjddtjjddd}y)Nzdebugprims::load_tensorzR(str name, int[] size, int[] stride, *, ScalarType dtype, Device device) -> Tensor BackendSelectctddlm}|||||Sddlm}tj ||}t |j|k(sJ|jd|t |j|k(sJ|jd||j|k(sJ|jd||j|k7r |||}|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#..tF.CA>T) BaffhZtD6+B B) #v- J!((*T&/J J-88v% @!((4x'@ @%ww%/H)torchlibrarydefineimpl)rs r register_debug_primsr sA MM!\  ]]1?CDr) contextlibtypingrrtorch.utils._content_storerr__annotations__contextmanagerr r rr r'sC 948H/07  " " r