diff options
Diffstat (limited to 'sci-libs/caffe2/files/caffe2-2.0.0-cudnn_include_fix.patch')
| -rw-r--r-- | sci-libs/caffe2/files/caffe2-2.0.0-cudnn_include_fix.patch | 12 |
1 files changed, 12 insertions, 0 deletions
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() + |