i@UddlZddlmZmZddlmZddlZddlmZda eee d<ejde dedfd Z d d Zy) N) GeneratorSequence)Optional)ContentStoreReaderLOAD_TENSOR_READERlocreturn)NNNc#RKtJt|da dday#dawxYww)NF)cache)rr)rs R/var/www/html/engine/venv/lib/python3.12/site-packages/torch/_prims/debug_prims.pyload_tensor_readerr s7  %% %,Cu=" !Ts' '$'c 2tjjddtjjdddtdt t dt t dtjdtjd tjf d }y) Nzdebugprims::load_tensorzR(str name, int[] size, int[] stride, *, ScalarType dtype, Device device) -> Tensor BackendSelectnamesizestridedtypedevicer ctddlm}|||||Sddlm}tj ||}t |j|k(sJ|jd|t |j|k(sJ|jd||j|k(sJ|jd||j|k7r |||}|S)Nr) rand_strided) clone_input)rz != )r) rtorch._dynamo.testingrtorch._dynamo.utilsr read_tensorlistrrrr)rrrrrrrrs 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) torchlibrarydefineimplstrrintrrTensor)rs r register_debug_primsr&s MM!\  ]]1?Csm {{     Dr)r N) contextlibcollections.abcrrtypingrrtorch.utils._content_storerr__annotations__contextmanagerr#r r&rr r.sZ/ 948H/07  "C "I.>$? " " r