Debug load_lib

Author: pearu

src/torchaudio/_extension/__init__.py

@@ -21,13 +21,14 @@
 ]
 
 
-if os.name == "nt" and (3, 8) <= sys.version_info < (3, 9):
+if os.name == "nt":  # and (3, 8) <= sys.version_info < (3, 9):
     _init_dll_path()
 
 # When the extension module is built, we initialize it.
 # In case of an error, we do not catch the failure as it suggests there is something
 # wrong with the installation.
 _IS_TORCHAUDIO_EXT_AVAILABLE = _load_lib("_torchaudio")
+assert _IS_TORCHAUDIO_EXT_AVAILABLE
 _IS_ALIGN_AVAILABLE = False
 if _IS_TORCHAUDIO_EXT_AVAILABLE:
     if not _load_lib("libtorchaudio"):

src/torchaudio/_extension/utils.py

@@ -7,6 +7,7 @@
 import logging
 import os
 import types
+import warnings
 from pathlib import Path
 
 import torch
@@ -16,7 +17,8 @@
 
 
 def _get_lib_path(lib: str):
-    suffix = "pyd" if os.name == "nt" else "so"
+    suffix = ".pyd" if os.name == "nt" else ".so"
+    paths = _LIB_DIR.glob(f"{lib}*{suffix}")
     path = _LIB_DIR / f"{lib}.{suffix}"
     return path
 
@@ -52,10 +54,14 @@ def _load_lib(lib: str) -> bool:
             This behavior was chosen because the expected failure case is not recoverable.
             If a dependency is missing, then users have to install it.
     """
-    path = _get_lib_path(lib)
-    if not path.exists():
+    suffix = ".pyd" if os.name == "nt" else ".so"
+    paths = list(_LIB_DIR.glob(f"{lib}*{suffix}"))
+    print(f"_load_lib: {lib=} {paths=}")
+    if not paths:
         return False
-    torch.ops.load_library(path)
+    if len(paths) > 1:
+        warnings.warn(f"Expected a single file path to {lib}, got {paths=}")
+    torch.ops.load_library(paths[0])
     return True