mirror of
https://github.com/NVIDIA/nvbench.git
synced 2026-03-14 20:27:24 +00:00
Stub of __cuda_stream__ special method declare tuple[int, int] as return type
This is to indicate that special method always returns a pair of integers
This commit is contained in:
@@ -35,7 +35,7 @@ class CudaStream:
|
||||
----
|
||||
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
|
||||
from `cuda.core`. Returns a pair of integers:
|
||||
|
||||
Reference in New Issue
Block a user