Skip to content

Commit 9679e0e

Browse files
Fix type in querying handle from Stream argument
1 parent b89c95f commit 9679e0e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

cuda_core/cuda/core/experimental/_module.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -228,14 +228,14 @@ def max_potential_cluster_size(self, config: LaunchConfig, stream: Optional[Stre
228228
""" "int: The maximum cluster size that can be launched for this kernel and launch configuration"""
229229
drv_cfg = _to_native_launch_config(config)
230230
if stream is not None:
231-
drv_cfg.hStream = stream._handle
231+
drv_cfg.hStream = stream.handle
232232
return handle_return(driver.cuOccupancyMaxPotentialClusterSize(self._handle, drv_cfg))
233233

234234
def max_active_clusters(self, config: LaunchConfig, stream: Optional[Stream] = None) -> int:
235235
""" "int: The maximum number of clusters that could co-exist on the target device"""
236236
drv_cfg = _to_native_launch_config(config)
237237
if stream is not None:
238-
drv_cfg.hStream = stream._handle
238+
drv_cfg.hStream = stream.handle
239239
return handle_return(driver.cuOccupancyMaxActiveClusters(self._handle, drv_cfg))
240240

241241

0 commit comments

Comments
 (0)