U %d @szddlZddlZddlmZddlZddlmZeej dZ e dddZ e e dd d Zd ad d ZddZedS)N)Path) module_utilslib)rcCs(tjdkrdnd}t|d|}|S)NntZpydso.)osname_LIB_DIR)rsuffixpathr 9/tmp/pip-unpacked-wheel-lbdmvq91/torchaudio/_extension.py _get_lib_path sr)rreturncCs0t|}|sdStj|tj|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_libraryclasses)rr r r r _load_libs   rFc CstrdStjjstdz tdWn,tk rT}ztd|W5d}~XYnXddl }tjj tjj dkrtjj ddadS)Nzftorchaudio is not compiled with FFmpeg integration. Please set USE_FFMPEG=1 when compiling torchaudio.Zlibtorchaudio_ffmpegz6FFmpeg libraries are not found. Please install FFmpeg.rT) _FFMPEG_INITIALIZEDrr torchaudioZis_ffmpeg_available RuntimeErrorrOSError ImportErrorZtorchaudio._torchaudio_ffmpegZ ffmpeg_initZffmpeg_get_log_levelZffmpeg_set_log_level)errrr r r _init_ffmpeg;s   rcCsPtdstddStdddlm}z tWntk rJYnXdS)Nztorchaudio._torchaudioz*torchaudio C++ extension is not available.Z libtorchaudior _torchaudio) _mod_utilsZis_module_availablewarningswarnrrrr Exceptionrr r r_init_extensionSs    r$)rr!pathlibrrZtorchaudio._internalrr __file__parentr strrboolrrrr$r r r rs  '