check_binary_symbols: Change search path for libtorch

pytorch · Sep 4, 2024 · 3fd03e6 · 3fd03e6
1 parent 068f797
commit 3fd03e6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/check_binary_symbols.py b/test/check_binary_symbols.py
@@ -79,7 +79,7 @@ def check_lib_symbols_for_abi_correctness(lib: str, pre_cxx11_abi: bool = True)
 
 def main() -> None:
     if os.getenv("PACKAGE_TYPE") == "libtorch":
-       install_root = Path(__file__).parent.parent
+       install_root = Path(os.get_cwd())
     else:
        install_root = Path(distutils.sysconfig.get_python_lib()) / "torch"
     libtorch_cpu_path = install_root / "lib" / "libtorch_cpu.so"