From e14bbec38c5efeed7bc153e5bafe5cf53e10071c Mon Sep 17 00:00:00 2001 From: Guillaume Fraux Date: Wed, 23 Jul 2025 12:02:50 +0200 Subject: [PATCH] Also look for libgomp in the user site-package if enabled --- python/metatomic_torch/metatomic/torch/_extensions.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/python/metatomic_torch/metatomic/torch/_extensions.py b/python/metatomic_torch/metatomic/torch/_extensions.py index 781580d6..9ef95830 100644 --- a/python/metatomic_torch/metatomic/torch/_extensions.py +++ b/python/metatomic_torch/metatomic/torch/_extensions.py @@ -68,7 +68,11 @@ def _find_openmp_dep(search_dir): if sys.platform.startswith("linux"): libs_list = [] - for prefix in site.getsitepackages(): + site_packages = site.getsitepackages() + if site.ENABLE_USER_SITE: + site_packages.append(site.getusersitepackages()) + + for prefix in site_packages: libs_dir = os.path.join(prefix, search_dir) if os.path.exists(libs_dir): libs_list = glob.glob(os.path.join(libs_dir, "libgomp-*.so*"))