v1.1.1 release (#61)

[Bug Fix] Fixed an issue in older cudnn versions where heurisitcs would return SUCCESS even if the number of heuristics results were zero.
NVIDIA · Feb 27, 2024 · fd27d76 · fd27d76
1 parent c29d609
commit fd27d76
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 3 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -1,6 +1,6 @@
 cmake_minimum_required(VERSION 3.17)
 
-project(cudnn_frontend VERSION 1.1.0)
+project(cudnn_frontend VERSION 1.1.1)
 
 option(CUDNN_FRONTEND_BUILD_SAMPLES "Defines if samples are built or not." ON)
 option(CUDNN_FRONTEND_BUILD_UNIT_TESTS "Defines if unittests are built or not." OFF)

diff --git a/include/cudnn_frontend.h b/include/cudnn_frontend.h
@@ -125,7 +125,7 @@
 
 #define CUDNN_FRONTEND_MAJOR_VERSION 1
 #define CUDNN_FRONTEND_MINOR_VERSION 1
-#define CUDNN_FRONTEND_PATCH_VERSION 0
+#define CUDNN_FRONTEND_PATCH_VERSION 1
 #define CUDNN_FRONTEND_VERSION \
     ((CUDNN_FRONTEND_MAJOR_VERSION * 10000) + (CUDNN_FRONTEND_MINOR_VERSION * 100) + CUDNN_FRONTEND_PATCH_VERSION)
 

diff --git a/include/cudnn_frontend_Heuristics.h b/include/cudnn_frontend_Heuristics.h
@@ -96,7 +96,11 @@ class EngineHeuristics_v8 : public BackendDescriptor {
                                           count,
                                           &result,
                                           heuristic_results_.data());
+#if (CUDNN_VERSION >= 8907)
         if (status != CUDNN_STATUS_SUCCESS || result < 1) {
+#else
+        if (status != CUDNN_STATUS_SUCCESS) {
+#endif
             set_error_and_throw_exception(
                 this, status, "CUDNN_BACKEND_ENGINEHEUR_DESCRIPTOR: GetAttribute CUDNN_ATTR_ENGINEHEUR_RESULTS Failed");
         };

diff --git a/setup.py b/setup.py
@@ -88,7 +88,7 @@ def build_extension(self, ext: CMakeExtension) -> None:
 # logic and declaration, and simpler if you include description/version in a file.
 setup(
     name="cudnn",
-    version="1.1.0",
+    version="1.1.1",
     author="",
     author_email="",
     description="cudnn_frontend python package",