diff --git a/patches/pip/pyg-lib.patch b/patches/pip/pyg-lib.patch index 6a50f69225c..2876d30a941 100644 --- a/patches/pip/pyg-lib.patch +++ b/patches/pip/pyg-lib.patch @@ -1,22 +1,22 @@ diff --git a/setup.py b/setup.py -index ab9d61d..bf3d56a 100644 +index 52399d2..542ca41 100644 --- a/setup.py +++ b/setup.py -@@ -65,6 +65,10 @@ class CMakeBuild(build_ext): - f'-DCMAKE_PREFIX_PATH={torch.utils.cmake_prefix_path}', - ] +@@ -97,6 +97,10 @@ class CMakeBuild(build_ext): + assert os.environ['TORCH_CUDA_ARCH_LIST'] is not None + print(f'TORCH_CUDA_ARCH_LIST={os.environ["TORCH_CUDA_ARCH_LIST"]}') + if CMakeBuild.check_env_flag('USE_CUDNN', "0"): + cmake_args.append('-DCAFFE2_USE_CUDNN=ON') + if CMakeBuild.check_env_flag('USE_CUFILE', "0"): + cmake_args.append('-DCAFFE2_USE_CUFILE=ON') if CMakeBuild.check_env_flag('USE_MKL_BLAS'): - include_dir = f"{sysconfig.get_path('data')}{os.sep}include" + include_dir = f'{sysconfig.get_path("data")}{os.sep}include' cmake_args.append(f'-DBLAS_INCLUDE_DIR={include_dir}') -@@ -138,6 +142,9 @@ setup( - 'test': test_requires, - 'dev': dev_requires, - }, +@@ -158,6 +162,9 @@ setup( + name='pyg_lib', + version=__version__, + install_requires=install_requires, + package_data={ + 'pyg_lib': ['csrc/**/*.h', 'csrc/**/*.hpp'] + }, diff --git a/pip/requirements.txt b/pip/requirements.txt index cb76947114a..1069619a851 100644 --- a/pip/requirements.txt +++ b/pip/requirements.txt @@ -305,8 +305,8 @@ pycurl==7.45.7 pydantic-core==2.41.5 pydot==4.0.1 pyflakes==3.4.0 -pyg-lib==0.6.0 -pyg-lib-cuda==0.6.0 +pyg-lib==0.7.0 +pyg-lib-cuda==0.7.0 Pygments==2.20.0 PyJWT==2.13.0 pylint==4.0.4 @@ -411,9 +411,9 @@ toolz==1.1.0 #e.g. for torch==2.10.0, torch-cuda and toch-rocm should also be 2.10.0 #Do the same for other torch-* packages e.g version of torch-scatter should be used for #torch-scatter-cuda and torch-scatter-rocm too. -torch==2.11.0 -torch-cuda==2.11.0 -torch-rocm==2.11.0 +torch==2.12.1 +torch-cuda==2.12.1 +torch-rocm==2.12.1 torch-cluster==1.6.3 torch-cluster-cuda==1.6.3 torch-cluster-rocm==1.6.3 diff --git a/pip/torch.file b/pip/torch.file index 10d5217817c..1879579460e 100644 --- a/pip/torch.file +++ b/pip/torch.file @@ -2,7 +2,6 @@ ## INCLUDE cpp-standard %define source0 git+https://github.com/pytorch/pytorch.git?obj=main/v%{realversion}&export=%{n}-%{realversion}&submodules=1&output=/%{n}-%{realversion}.tgz -Source96: https://github.com/pytorch/pytorch/commit/39565a7dcf8f93ea22cedeaa20088b24ff6d2634.patch Source97: pip/files/torch-build-env Source98: FindEigen3.cmake Source99: scram-tools.file/tools/eigen/env @@ -15,7 +14,6 @@ Requires: OpenBLAS zlib protobuf fmt py3-pybind11 py3-typing-extensions Requires: py3-filelock py3-fsspec py3-Jinja2 py3-networkx py3-sympy %define patchsrc0 cp %{_sourcedir}/FindEigen3.cmake cmake/Modules/ -%define patchsrc1 patch -p1 <%{_sourcedir}/39565a7dcf8f93ea22cedeaa20088b24ff6d2634.patch %define PipPreBuild %{expand: %(cat %{_sourcedir}/torch-build-env)} %define PipPostInstall \