Fixes for Python 3.10

Author: mdboom

cuda_core/cuda/core/system/_device.pyx

@@ -5,7 +5,11 @@
 from libc.stdint cimport intptr_t, uint64_t
 from libc.math cimport ceil
 
-from enum import StrEnum
+import sys
+if sys.version_info >= (3, 11):
+    from enum import StrEnum
+else:
+    from backports.strenum import StrEnum
 from multiprocessing import cpu_count
 from typing import Iterable
 import warnings

cuda_core/cuda/core/system/_system_events.pyx

@@ -5,7 +5,11 @@
 
 from libc.stdint cimport intptr_t
 
-from enum import StrEnum
+import sys
+if sys.version_info >= (3, 11):
+    from enum import StrEnum
+else:
+    from backports.strenum import StrEnum
 
 from cuda.bindings import nvml
 

cuda_core/pixi.toml

@@ -186,6 +186,9 @@ numpy = "*"
 cuda-bindings = "*"
 cuda-pathfinder = "*"
 
+[package.target.'python_version < "3.11"'.run-dependencies]
+"backports.strenum" = "*"
+
 [target.linux.tasks.build-cython-tests]
 cmd = ["$PIXI_PROJECT_ROOT/tests/cython/build_tests.sh"]
 

cuda_core/pyproject.toml

@@ -50,6 +50,7 @@ classifiers = [
 dependencies = [
     "cuda-pathfinder >=1.4.2",
     "numpy",
+    "backports.strenum; python_version < '3.11'",
 ]
 
 [project.optional-dependencies]