Skip to content

Commit ff7a0f6

Browse files
Fix mypy error in import statement used in cutlass_gemm example
1 parent b2b1c72 commit ff7a0f6

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

python/examples/cutlass_gemm.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,12 +19,11 @@
1919

2020
import cuda.bindings.driver as driver
2121
import cuda.core.experimental as core
22+
import cuda.nvbench as nvbench
2223
import cupy as cp
2324
import cutlass
2425
import numpy as np
2526

26-
import nvbench
27-
2827

2928
def as_bindings_Stream(cs: nvbench.CudaStream) -> driver.CUstream:
3029
return driver.CUstream(cs.addressof())

0 commit comments

Comments
 (0)