NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-03-14 20:27:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix mypy error in import statement used in cutlass_gemm example

Browse Source
This commit is contained in:
Oleksandr Pavlyk
2025-07-24 10:33:08 -05:00
parent b9554d7980
commit 5e8c17c740
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
python/examples/cutlass_gemm.py
View File

@@ -19,12 +19,11 @@ import sys
import cuda.bindings.driver as driver
import cuda.core.experimental as core
import cuda.nvbench as nvbench
import cupy as cp
import cutlass
import numpy as np
import nvbench
def as_bindings_Stream(cs: nvbench.CudaStream) -> driver.CUstream:
return driver.CUstream(cs.addressof())