diff options
author | Alfredo Tupone <tupone@gentoo.org> | 2023-04-24 10:06:19 +0200 |
---|---|---|
committer | Alfredo Tupone <tupone@gentoo.org> | 2023-04-24 10:35:48 +0200 |
commit | 4d6a404d63ba3914c98344efe64d9b631089e3e1 (patch) | |
tree | 27afb1abd91bf38b598ac2fe3d828d1f772e5b8d /sci-libs/caffe2 | |
parent | dev-libs/libffi: Stabilize 3.4.4-r1 sparc, #904936 (diff) | |
download | gentoo-4d6a404d63ba3914c98344efe64d9b631089e3e1.tar.gz gentoo-4d6a404d63ba3914c98344efe64d9b631089e3e1.tar.bz2 gentoo-4d6a404d63ba3914c98344efe64d9b631089e3e1.zip |
sci-libs/caffe2: fix cudnn include path
Closes: https://bugs.gentoo.org/904882
Signed-off-by: Alfredo Tupone <tupone@gentoo.org>
Diffstat (limited to 'sci-libs/caffe2')
-rw-r--r-- | sci-libs/caffe2/caffe2-2.0.0-r2.ebuild | 1 | ||||
-rw-r--r-- | sci-libs/caffe2/files/caffe2-2.0.0-cudnn_include_fix.patch | 12 |
2 files changed, 13 insertions, 0 deletions
diff --git a/sci-libs/caffe2/caffe2-2.0.0-r2.ebuild b/sci-libs/caffe2/caffe2-2.0.0-r2.ebuild index 38ce9cd3befa..7667c7bbee17 100644 --- a/sci-libs/caffe2/caffe2-2.0.0-r2.ebuild +++ b/sci-libs/caffe2/caffe2-2.0.0-r2.ebuild @@ -81,6 +81,7 @@ PATCHES=( "${FILESDIR}"/${PN}-1.12.0-glog-0.6.0.patch "${FILESDIR}"/${PN}-1.13.1-tensorpipe.patch "${FILESDIR}"/${P}-gcc13.patch + "${FILESDIR}"/${P}-cudnn_include_fix.patch ) src_prepare() { diff --git a/sci-libs/caffe2/files/caffe2-2.0.0-cudnn_include_fix.patch b/sci-libs/caffe2/files/caffe2-2.0.0-cudnn_include_fix.patch new file mode 100644 index 000000000000..ff64e4108087 --- /dev/null +++ b/sci-libs/caffe2/files/caffe2-2.0.0-cudnn_include_fix.patch @@ -0,0 +1,12 @@ +diff -uar pytorch-2.0.0/cmake/Dependencies.cmake pytorch-2.0.0orig/cmake/Dependencies.cmake +--- a/cmake/Dependencies.cmake 2023-04-23 09:43:20.767566047 -0400 ++++ b/cmake/Dependencies.cmake 2023-03-09 17:42:00.000000000 -0500 +@@ -1235,7 +1235,7 @@ + + # ---[ cuDNN + if(USE_CUDNN) +- set(CUDNN_FRONTEND_INCLUDE_DIR ${CMAKE_CURRENT_LIST_DIR}/../third_party/cudnn_frontend/include) ++ set(CUDNN_FRONTEND_INCLUDE_DIR /opt/cuda/include) + target_include_directories(torch::cudnn INTERFACE ${CUDNN_FRONTEND_INCLUDE_DIR}) + endif() + |