Update fp8_gemm_tuner.py exchange import torch and hipbsolidxgemm (#46)

* 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 * run isort on fp9_gemm_tuner.py * add # isort: split * fix yapf --------- Co-authored-by: charlifu <[email protected]>
ROCm · Jun 11, 2024 · ff24102 · ff24102
1 parent d254de7
commit ff24102
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  # isort: split
 import hipbsolidxgemm
 import pandas as pd
-import torch
 import torch.nn.functional as F
 
 hipbsolidxgemm.hipb_create_extension()