Update fp8_gemm_tuner.py exchange import torch and hipbsolidxgemm

ImportError: libc10.so: cannot open shared object file: No such file or directory https://stackoverflow.com/a/65710714
ROCm · Jun 11, 2024 · bd8d0a6 · bd8d0a6
1 parent d254de7
commit bd8d0a6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()