diff --git a/CMakeLists.txt b/CMakeLists.txt index a5a8feb8..645f37dc 100644 --- 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 index 0f0d5a66..57ae8c89 100644 --- 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 index e404951b..d4c984b6 100644 --- 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 index 335e35c8..d3ea3a0b 100644 --- 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",