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()