Sorry, something went wrong.
|
@umar456 Wouldn't it be valid to lookup only versions same or earlier to the one current build is built with ? diff --git a/src/backend/cuda/cudnnModule.cpp b/src/backend/cuda/cudnnModule.cpp
index b76b0c65f..c1c6ab0b1 100644
--- a/src/backend/cuda/cudnnModule.cpp
+++ b/src/backend/cuda/cudnnModule.cpp
@@ -27,7 +27,10 @@ namespace cuda {
// clang-format off
// Latest version from each minor releases are enlisted below
constexpr std::array<common::Version, 11> cudnnVersions = {
+#if CUDNN_VERSION >= 8000
make_tuple(8, 0, 1),
+#endif
+#if CUDNN_VERSION >= 7000
make_tuple(7, 6, 5),
make_tuple(7, 5, 1),
make_tuple(7, 4, 2),
@@ -35,8 +38,13 @@ constexpr std::array<common::Version, 11> cudnnVersions = {
make_tuple(7, 2, 1),
make_tuple(7, 1, 4),
make_tuple(7, 0, 5),
+#endif
+#if CUDNN_VERSION >= 6000
make_tuple(6, 0, 21),
+#endif
+#if CUDNN_VERSION >= 5000
make_tuple(5, 1, 10),
+#endif
make_tuple(4, 0, 7)
};
// clang-format on
|
Sorry, something went wrong.
Description
Fixes: #2935
Changes to Users
None
Checklist