a hr@sdZddlZddlZddlZddlZddlmZddlZddlm Z e e Z ee jjdZedddZeed d d Zd d ZddZGdddejZddZddZdS)zModule to implement logics used for initializing extensions. The implementations here should be stateless. They should not depend on external state. Anything that depends on external state should happen in __init__.py N)Path)eval_envlib)rcCs(tjdkrdnd}t|d|}|S)NntZpydso.)osname_LIB_DIR)rsuffixpathr I/var/www/auris/lib/python3.9/site-packages/torchaudio/_extension/utils.py _get_lib_pathsr)rreturncCs$t|}|sdStj|dS)a Load extension module Note: In case `torchaudio` is deployed with `pex` format, the library file is not in a standard location. In this case, we expect that `libtorchaudio` is available somewhere in the search path of dynamic loading mechanism, so that importing `_torchaudio` will have library loader find and load `libtorchaudio`. This is the reason why the function should not raising an error when the library file is not found. Returns: bool: True if the library file is found AND the library loaded without failure. False if the library file is not found (like in the case where torchaudio is deployed with pex format, thus the shared library file is in a non-standard location.). If the library file is found but there is an issue loading the library, (such as missing dependency) then this function raises the exception as-is. Raises: Exception: If the library file is found, but there is an issue loading the library file, (when underlying `ctype.DLL` throws an exception), this function will pass the exception as-is, instead of catching it and returning bool. The expected case is `OSError` thrown by `ctype.DLL` when a dynamic dependency is not found. This behavior was chosen because the expected failure case is not recoverable. If a dependency is missing, then users have to install it. FT)rexiststorchopsZ load_library)rr r r r _load_libs  rcCsNtjdkrtdtdds$tdd}tj|s)r* __module__r-__name__r reprr+r r r__repr__|s  z_LazyImporter.__repr__cCs|t|jSr&)r.dirr*r4r r r__dir__sz_LazyImporter.__dir__cCs(|jdur$||_|j|jjdSr&)r*r)__dict__updater4r r rr.s  z_LazyImporter._import_oncecCs&z |Wnty YdS0dS)NFT)r. Exceptionr4r r r is_availables   z_LazyImporter.is_available) r2r1 __qualname____doc__r(r0r5r7r.r; __classcell__r r r,rr%ms  r%c CsLtjdddD]2}tj|rzt|WqtyDYq0qdS)NPATH;)renvirongetsplitr rZadd_dll_directoryr:)r r r r_init_dll_paths   rEcCsddl}|jj}|durtjjdurt|}|ddd|d}tjjd}|dd|d}||krt d|d|d|S) NrrzjDetected that PyTorch and TorchAudio were compiled with different CUDA versions. PyTorch has CUDA version z% whereas TorchAudio has CUDA version zJ. Please install the TorchAudio version that matches your PyTorch version.) Ztorchaudio.lib._torchaudiorZ _torchaudioZ cuda_versionrversionZcudastrrDr)Z torchaudiorI version_strZ ta_versionZ t_versionr r r_check_cuda_versions  rL)r=rloggingrtypespathlibrrZ!torchaudio._internal.module_utilsr getLoggerr2Z_LG__file__parentr rJrboolrrr$ ModuleTyper%rErLr r r rs    &)