diff --git a/gradlib/gradlib/fp8_gemm_tuner.py b/gradlib/gradlib/fp8_gemm_tuner.py index 61df1933f8658..225701939ec0a 100644 --- a/gradlib/gradlib/fp8_gemm_tuner.py +++ b/gradlib/gradlib/fp8_gemm_tuner.py @@ -4,9 +4,9 @@ import random from pathlib import Path +import torch # isort: split import hipbsolidxgemm import pandas as pd -import torch import torch.nn.functional as F hipbsolidxgemm.hipb_create_extension()