Lower green context resource handling to Cython

Author: leofang

cuda_core/cuda/core/_context.pyx

@@ -2,9 +2,13 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
+from __future__ import annotations
+
+from collections.abc import Sequence
 from dataclasses import dataclass
 
 from cuda.bindings cimport cydriver
+from cuda.core._device_resources import SMResource, WorkqueueResource
 from cuda.core._resource_handles cimport (
     ContextHandle,
     GreenCtxHandle,
@@ -21,6 +25,9 @@ from cuda.core._utils.cuda_utils cimport HANDLE_RETURN
 __all__ = ['Context', 'ContextOptions']
 
 
+DeviceResourcesT = SMResource | WorkqueueResource | Sequence[SMResource | WorkqueueResource]
+
+
 cdef class Context:
     """CUDA context wrapper.
 
@@ -100,12 +107,12 @@ cdef class Context:
 
 
 @dataclass
-class ContextOptions:
+cdef class ContextOptions:
     """Options for context creation.
 
     Attributes
     ----------
-    resources : Sequence[SMResource | WorkqueueResource], optional
+    resources : :obj:`~_context.DeviceResourcesT`
         Device resources used to create a green context.
     """
-    resources: object = None
+    resources: DeviceResourcesT

cuda_core/cuda/core/_device.pyx

@@ -7,19 +7,14 @@ from __future__ import annotations
 cimport cpython
 
 from cuda.bindings cimport cydriver
-from cuda.core._utils.cuda_utils cimport HANDLE_RETURN
+from cuda.core._utils.cuda_utils cimport check_or_create_options, HANDLE_RETURN
 from libc.stdlib cimport free, malloc
 
 import threading
 
 from cuda.core._context cimport Context
 from cuda.core._context import ContextOptions
 from cuda.core._device_resources cimport DeviceResources, SMResource, WorkqueueResource
-from cuda.core._device_resources import (
-    DeviceResources,
-    SMResource,
-    WorkqueueResource,
-)
 from cuda.core._event cimport Event as cyEvent
 from cuda.core._event import Event, EventOptions
 from cuda.core._memory._buffer cimport Buffer, MemoryResource
@@ -1309,7 +1304,7 @@ class Device:
         if options is None:
             raise NotImplementedError("WIP: https://github.com/NVIDIA/cuda-python/issues/189")
 
-        assert_type(options, ContextOptions)
+        options = check_or_create_options(ContextOptions, options, "Context options")
         if options.resources is None:
             raise NotImplementedError("WIP: https://github.com/NVIDIA/cuda-python/issues/189")
 

cuda_core/cuda/core/_device_resources.pxd

@@ -8,14 +8,18 @@ from cuda.bindings cimport cydriver
 cdef class SMResource:
     cdef:
         cydriver.CUdevResource _resource
+        unsigned int _sm_count
+        unsigned int _min_partition_size
+        unsigned int _coscheduled_alignment
+        unsigned int _flags
         bint _is_usable
         object __weakref__
 
     @staticmethod
-    cdef SMResource _from_dev_resource(cydriver.CUdevResource res)
+    cdef SMResource _from_dev_resource(cydriver.CUdevResource res, int device_id)
 
     @staticmethod
-    cdef SMResource _from_dry_run_resource(cydriver.CUdevResource res)
+    cdef SMResource _from_split_resource(cydriver.CUdevResource res, SMResource parent, bint is_usable)
 
 
 cdef class WorkqueueResource: