diff --git a/gradlib/gradlib/fp8_gemm_tuner.py b/gradlib/gradlib/fp8_gemm_tuner.py index 0624a2645edff..61df1933f8658 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 import hipbsolidxgemm import pandas as pd +import torch import torch.nn.functional as F hipbsolidxgemm.hipb_create_extension()