Skip to content

Commit ea67951

Browse files
Omar Atiecursoragent
authored andcommitted
fix(cuda.core): regenerate LaunchConfig stubs with stubgen-pyx
pre-commit.ci failed because the hand-merged .pyi did not match stubgen-pyx 0.2.19 output for cluster_scheduling_policy_preference. Co-authored-by: Cursor <cursoragent@cursor.com>
1 parent bca91b4 commit ea67951

1 file changed

Lines changed: 2 additions & 3 deletions

File tree

cuda_core/cuda/core/_launch_config.pyi

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ class LaunchConfig:
5151
shmem_size: int
5252
is_cooperative: bool
5353
programmatic_stream_serialization: bool
54-
cluster_scheduling_policy_preference: ClusterSchedulingPolicyType | None
54+
cluster_scheduling_policy_preference: object
5555

5656
def __init__(self, grid: int | tuple[int, ...] | None=None, cluster: int | tuple[int, ...] | None=None, block: int | tuple[int, ...] | None=None, shmem_size: int | None=None, is_cooperative: bool=False, programmatic_stream_serialization: bool=False, cluster_scheduling_policy_preference: ClusterSchedulingPolicyType | None=None) -> None:
5757
"""Initialize LaunchConfig with validation.
@@ -79,6 +79,7 @@ class LaunchConfig:
7979
def __eq__(self, other: object) -> bool: ...
8080
def __hash__(self) -> int: ...
8181

82+
def _validate_cluster_scheduling_policy_preference(value): ...
8283
def _to_native_launch_config(config: LaunchConfig) -> object:
8384
"""Convert LaunchConfig to native driver CUlaunchConfig.
8485
@@ -92,5 +93,3 @@ def _to_native_launch_config(config: LaunchConfig) -> object:
9293
driver.CUlaunchConfig
9394
Native CUDA driver launch configuration
9495
"""
95-
96-
def _validate_cluster_scheduling_policy_preference(value): ...

0 commit comments

Comments
 (0)