i!UddlZddlmZmZddlZddlmZdaedzed<ejde dedfdZ d d Z y) N) GeneratorSequence)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 BackendSelectnamesizestridedtypedevicerctddlm}|||||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)rN) contextlibcollections.abcrrrtorch.utils._content_storerr__annotations__contextmanagerr"r r%rr r,sW/ 915&-4  "C "I.>$? " " r