o ei@shUddlZddlmZmZddlZddlmZdaedBed<ej de dedfdd Z d d d Z dS) N) GeneratorSequence)ContentStoreReaderLOAD_TENSOR_READERlocreturn)NNNccs6tdur tdt|ddazdVWdadSdaw)Nz!LOAD_TENSOR_READER is already setF)cache)rAssertionErrorr)rr b/var/www/addictedbytheproject.nl/epg/venv/lib/python3.10/site-packages/torch/_prims/debug_prims.pyload_tensor_reader s r c CsPtjddtjdddtdttdttdtjdtjd tj f d d }dS) Nzdebugprims::load_tensorzR(str name, int[] size, int[] stride, *, ScalarType dtype, Device device) -> Tensor BackendSelectnamesizestridedtypedevicercSstdurddlm}|||||Sddlm}tj||d}t||kr1t|d|t| |krDt| d||j |krSt|j d||j |kr^|||d}|S)Nr) rand_strided) clone_input)rz != )r) rtorch._dynamo.testingrtorch._dynamo.utilsr read_tensorlistrr rrr)rrrrrrrrr r r load_tensor_factory#s     z1register_debug_prims..load_tensor_factory) torchlibrarydefineimplstrrintrrTensor)rr r r register_debug_primss$ r")rN) contextlibcollections.abcrrrtorch.utils._content_storerr__annotations__contextmanagerrr r"r r r r s