1.1.2 release. (#65)

[Bug fix] Fixed an issue where the heuristic, when returning 0 results was throwing an error. This is considered a success as this can be appended with the fallback heuristics to build a valid plan.
NVIDIA · Feb 28, 2024 · 150798f · 150798f
1 parent fd27d76
commit 150798f
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 7 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.1)
+project(cudnn_frontend VERSION 1.1.2)
 
 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 1
+#define CUDNN_FRONTEND_PATCH_VERSION 2
 #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,11 +96,7 @@ 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.1",
+    version="1.1.2",
     author="",
     author_email="",
     description="cudnn_frontend python package",