NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-04-20 06:48:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix type annotations in cuda.nvbench, and in examples

Browse Source
This commit is contained in:
Oleksandr Pavlyk
2025-07-22 13:02:22 -05:00
parent 13ad115ca3
commit a535a1d173
6 changed files with 52 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/examples/cupy_extract.py
View File

@@ -5,7 +5,7 @@ import cupy as cp
def as_cp_ExternalStream(
cs: nvbench.CudaStream, dev_id: int = -1
cs: nvbench.CudaStream, dev_id: int | None = -1
) -> cp.cuda.ExternalStream:
h = cs.addressof()
return cp.cuda.ExternalStream(h, dev_id)