From 769236d976c06e5bcacf7bc467608e6f54d0f76c Mon Sep 17 00:00:00 2001
From: Rohan138 <Rohan.Potdar@amd.com>
Date: Wed, 11 Dec 2024 10:16:35 -0600
Subject: [PATCH] fix GemmTuner import

---
 gradlib/gemm_tuner.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/gradlib/gemm_tuner.py b/gradlib/gemm_tuner.py
index 0c0b42ba48ade..35345b2456848 100644
--- a/gradlib/gemm_tuner.py
+++ b/gradlib/gemm_tuner.py
@@ -7,7 +7,7 @@
 import pandas as pd
 
 import vllm._gradlib_C  # noqa: F401
-from gradlib.GemmTuner import GemmTuner
+from GemmTuner import GemmTuner
 
 torch.ops._gradlib_C.rocb_create_extension()
 torch.ops._gradlib_C.hipb_create_extension()