From 21af71c140be915047091d035e47248ce5dd07b0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=BE=90=E5=B0=8F=E9=BE=99?= Date: Wed, 22 Jan 2025 09:06:56 +0800 Subject: [PATCH 1/2] =?UTF-8?q?=E8=A7=A3=E5=86=B3site-packages=E8=B7=AF?= =?UTF-8?q?=E5=BE=84=E9=94=99=E8=AF=AF=E9=97=AE=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- patch/npu.patch | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/patch/npu.patch b/patch/npu.patch index c54cd60..4f8f762 100644 --- a/patch/npu.patch +++ b/patch/npu.patch @@ -2645,8 +2645,8 @@ diff -Nur '--exclude=.git' apex/setup.py apex-develop/setup.py + if '--user' in sys.argv: + package_dir = site.getusersitepackages() + else: -+ py_version = f'{sys.version_info.major}.{sys.version_info.minor}' -+ package_dir = f'{sys.prefix}/lib/python{py_version}/site-packages' ++ import sysconfig ++ package_dir = sysconfig.get_paths()['purelib'] + return package_dir + + -- Gitee From accbf157d49b7c1217367f782fec3f20b9295be7 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=BE=90=E5=B0=8F=E9=BE=99?= Date: Wed, 22 Jan 2025 11:29:01 +0800 Subject: [PATCH 2/2] =?UTF-8?q?packages=20dir=E5=AE=9E=E9=99=85=E5=BA=94?= =?UTF-8?q?=E4=B8=BAtorch=E6=89=80=E5=9C=A8site-packages=E6=96=87=E4=BB=B6?= =?UTF-8?q?=E5=A4=B9=EF=BC=8C=E6=97=A0=E5=85=B3=E6=98=AF=E5=90=A6=E6=98=AF?= =?UTF-8?q?--user=E5=AE=89=E8=A3=85?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- patch/npu.patch | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/patch/npu.patch b/patch/npu.patch index 4f8f762..6271d92 100644 --- a/patch/npu.patch +++ b/patch/npu.patch @@ -2642,14 +2642,14 @@ diff -Nur '--exclude=.git' apex/setup.py apex-develop/setup.py +secure_link_args = ['-Wl,-z,now', '-Wl,-z,relro', '-Wl,-z,noexecstack', '-s'] + +def get_package_dir(): -+ if '--user' in sys.argv: -+ package_dir = site.getusersitepackages() -+ else: -+ import sysconfig -+ package_dir = sysconfig.get_paths()['purelib'] ++ torch_file_path = torch.__file__ ++ package_dir = os.path.dirname(os.path.dirname(torch_file_path)) + return package_dir + + ++ ++ ++ +def CppExtension(name, sources, *args, **kwargs): + r''' + Creates a :class:`setuptools.Extension` for C++. -- Gitee