o i@sddlZddZdS)NcCs`d}|D]$}tj|}r(tj|}|jdusJ|j|t||Sqtd|)z rsqrt = libdevice_find("rsqrt") is a triton-version-friendly way to say from triton.language.extra.libdevice import rsqrt )ztriton.language.extra.libdevicez$triton.language.extra.cuda.libdeviceztriton.language.mathztriton.language.libdeviceNz#Could not find a library to import ) importlibutil find_specmodule_from_specloader exec_modulegetattr ImportError)namelocslocspecmodulerM/home/ubuntu/.local/lib/python3.10/site-packages/xformers/triton/importing.pylibdevice_find s  r)importlib.utilrrrrrrs