summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorAlfredo Tupone <tupone@gentoo.org>2023-04-24 10:06:19 +0200
committerAlfredo Tupone <tupone@gentoo.org>2023-04-24 10:35:48 +0200
commit4d6a404d63ba3914c98344efe64d9b631089e3e1 (patch)
tree27afb1abd91bf38b598ac2fe3d828d1f772e5b8d /sci-libs/caffe2
parentdev-libs/libffi: Stabilize 3.4.4-r1 sparc, #904936 (diff)
downloadgentoo-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.ebuild1
-rw-r--r--sci-libs/caffe2/files/caffe2-2.0.0-cudnn_include_fix.patch12
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()
+