NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-04-30 11:41:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Stub of __cuda_stream__ special method declare tuple[int, int] as return type

Browse Source 
This is to indicate that special method always returns a pair of integers
This commit is contained in:
Oleksandr Pavlyk
2025-08-04 10:11:33 -05:00
parent a5e0a48f80
commit 73e18419b2
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/cuda/nvbench/__init__.pyi
View File

@@ -35,7 +35,7 @@ class CudaStream:
---- ----
The class is not directly constructible. The class is not directly constructible.
""" """
def __cuda_stream__(self) -> tuple[int]: def __cuda_stream__(self) -> tuple[int, int]:
""" """
Special method implement CUDA stream protocol Special method implement CUDA stream protocol
from `cuda.core`. Returns a pair of integers: from `cuda.core`. Returns a pair of integers: