diff options
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() + |