fix GemmTuner import

ROCm · Dec 11, 2024 · 769236d · 769236d
1 parent eb4d191
commit 769236d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()