Fix wrong detection of python version for use_rmm. (#2224)

2025-04-30 12:29:33 -07:00
parent e5b810bed1
commit 35136f5564
1 changed files with 1 additions and 1 deletions
--- 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