diff --git a/gradlib/gemm_tuner.py b/gradlib/gemm_tuner.py index 35345b24568..182ced2b86d 100644 --- 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 GemmTuner import GemmTuner torch.ops._gradlib_C.rocb_create_extension() torch.ops._gradlib_C.hipb_create_extension()