o )iw@sUddlmZmZmZmZmZddlZdedefddZdefddZ Gd d d Z ga eee e d <ia eeee fe d <ed ZdedefddZeejddZedur\ejjZdejdefddZdS))AnyDictListTypeTypeVarNlibrarynamec s>fdd}z tttjWSttfy|YSw)Ncstddd)NzNo such operator z::zC - did you forget to build xformers with `python setup.py develop`?) RuntimeError)argskwargsrr_/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/xformers/ops/common.pyno_such_operator sz&get_operator..no_such_operator)getattrtorchopsr AttributeError)rrrr r r get_operator s rcCs td|S)NZxformers)r)rr r rget_xformers_operators rc@s8eZdZUeed<eed<eed<edefddZdS) BaseOperatorOPERATORNAMEZOPERATOR_CATEGORYreturncCs$|jdust|jdddkrdSdS)N__name__rFT)rrclsr r r is_available s zBaseOperator.is_availableN) r __module__ __qualname__r__annotations__str classmethodboolrr r r rrs rOPERATORS_REGISTRYFUNC_TO_XFORMERS_OPERATORClsTrrcCst||t|j<|SN)r%appendr&rrr r rregister_operator1s  r*Zuntyped_storagexcCs t|Sr()_GET_TENSOR_STORAGEZdata_ptr)r+r r r_get_storage_base?s r-)typingrrrrrrr"rrrr%r!r&r'r*rZTensorr,Zstorageintr-r r r rs