/jUddlZddlmZmZddlZddlmZdaedzed<ej de dedfdZ d d Z dS) N) GeneratorSequence)ContentStoreReaderLOAD_TENSOR_READERlocreturn)NNNc#tKttdt|da dVdadS#dawxYw)Nz!LOAD_TENSOR_READER is already setF)cache)rAssertionErrorr)rs ]/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/torch/_prims/debug_prims.pyload_tensor_readerr s^%@AAA,Cu===" !T!!!!s37c 8tjddtjdddtdt t dt t dtjdtjd tj f d }dS) Nzdebugprims::load_tensorzR(str name, int[] size, int[] stride, *, ScalarType dtype, Device device) -> Tensor BackendSelectnamesizestridedtypedevicerc$tddlm}|||||Sddlm}t||}t ||kr&t|d|t | |kr&t| d||j |krt|j d||j |kr |||}|S)Nr) rand_strided) clone_input)rz != )r) rtorch._dynamo.testingrtorch._dynamo.utilsr read_tensorlistrr rrr)rrrrrrrrs r load_tensor_factoryz1register_debug_prims..load_tensor_factory#s-  % : : : : : :<feV<< < 7 7 7 7 7 7#..tF.CCAAFFHH~~%%$%<%%>f%>%>???w%K///H) torchlibrarydefineimplstrrintrrTensor)rs r register_debug_primsr&s M!\  ]1?CCsm {     DCr)rN) contextlibcollections.abcrrrtorch.utils._content_storerr__annotations__contextmanagerr#r r&rr r-s//////// 99999915&-444 "C"I.>$?"""""""""""r