o )i@sUddlZddlZddlZddlmZddlmZmZmZmZzddl m Z Wn e y2dZ Ynwe dZ ejduZeed<d Zeed <d d Zed dZeddZdS)N)_cpp_lib) checkpointget_optimal_checkpoint_policylist_operatorsselective_checkpoint_wrapper) __version__z0.0.0Zxformers_has_cpp_libraryT_is_opensourcecsdfdd}|S)NcsdurS)Nr funcvaluer ]/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/xformers/__init__.py func_wrapper#sz"compute_once..func_wrapperr )r rr r r compute_once src Cstjdddkr dStjsdStjdddkrdStjddkr'dSzd dl}WdStt fyAt j d dd YdSw) NZXFORMERS_ENABLE_TRITON01TFZXFORMERS_FORCE_DISABLE_TRITONcuda)rrzJA matching Triton is not available, some optimizations will not be enabled)exc_info) osenvirongettorchrZ is_availableZget_device_capabilitytriton ImportErrorAttributeErrorloggerwarning)rr r r_is_triton_available,s$ r cCstjddS)NZxformers_pythonZDEF)rZlibraryLibraryr r r rget_python_libDsr")loggingrrrrrrrversionrr getLoggerrZ_cpp_library_load_exceptionr bool__annotations__r rr r"r r r rs$