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)argskwargsrrG/home/ubuntu/.local/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)Nxformers)r)rr r rget_xformers_operators rc@s8eZdZUeed<eed<eed<edefddZdS) BaseOperatorOPERATORNAMEOPERATOR_CATEGORYreturncCs$|jdust|jdddkrdSdS)N__name__rFT)rrclsr r r is_available s zBaseOperator.is_availableN) r __module__ __qualname__r__annotations__str classmethodboolr r r r rrs rOPERATORS_REGISTRYFUNC_TO_XFORMERS_OPERATORClsTrrcCst||t|j<|SN)r'appendr(rrr r rregister_operator1s  r,untyped_storagexcCs t|Sr*)_GET_TENSOR_STORAGEdata_ptr)r.r r r_get_storage_base?s r1)typingrrrrrrr$rrrr'r#r(r)r,rTensorr/storageintr1r r r rs