From 5e8c17c740d4ec205dfb5d23a9299c8f58f8d9e5 Mon Sep 17 00:00:00 2001
From: Oleksandr Pavlyk <21087696+oleksandr-pavlyk@users.noreply.github.com>
Date: Thu, 24 Jul 2025 10:33:08 -0500
Subject: [PATCH] Fix mypy error in import statement used in cutlass_gemm
 example

---
 python/examples/cutlass_gemm.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/python/examples/cutlass_gemm.py b/python/examples/cutlass_gemm.py
index 1675d0c..154bc16 100644
--- a/python/examples/cutlass_gemm.py
+++ b/python/examples/cutlass_gemm.py
@@ -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())