From 35136f556498c26733f7c6ce085ba02793ecaefe Mon Sep 17 00:00:00 2001 From: Ruoxi Date: Wed, 30 Apr 2025 12:29:33 -0700 Subject: [PATCH] Fix wrong detection of python version for `use_rmm`. (#2224) --- python/cutlass/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/cutlass/__init__.py b/python/cutlass/__init__.py index cd4b4617..583de7dd 100644 --- a/python/cutlass/__init__.py +++ b/python/cutlass/__init__.py @@ -99,7 +99,7 @@ this = sys.modules[__name__] this.logger = logging.getLogger(__name__) # RMM is only supported for Python 3.9+ -if (sys.version_info.major == 3 and sys.version_info.major > 8) or sys.version_info.major > 3: +if (sys.version_info.major == 3 and sys.version_info.minor > 8) or sys.version_info.major > 3: try: import rmm this.use_rmm = True