NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-04-29 19:21:14 +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.bindings.driver as driver
import cuda.core.experimental as core import cuda.core.experimental as core
import cuda.nvbench as nvbench
import cupy as cp import cupy as cp
import cutlass import cutlass
import numpy as np import numpy as np
import nvbench
def as_bindings_Stream(cs: nvbench.CudaStream) -> driver.CUstream: def as_bindings_Stream(cs: nvbench.CudaStream) -> driver.CUstream:
return driver.CUstream(cs.addressof()) return driver.CUstream(cs.addressof())