fix GemmTuner import in gradlib (#319)

* fix GemmTuner import * fix GemmTuner import
ROCm · Dec 11, 2024 · e8e6ebf · e8e6ebf
1 parent eb4d191
commit e8e6ebf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gradlib/gemm_tuner.py b/gradlib/gemm_tuner.py
@@ -5,9 +5,9 @@
 
 import torch  # isort: split
 import pandas as pd
+from GemmTuner import GemmTuner
 
 import vllm._gradlib_C  # noqa: F401
-from gradlib.GemmTuner import GemmTuner
 
 torch.ops._gradlib_C.rocb_create_extension()
 torch.ops._gradlib_C.hipb_create_extension()