precommit

Author: jimmyzho

flashinfer/artifacts.py

@@ -86,6 +86,7 @@ class ArtifactPath:
     The paths are generated in cubin publishing script logs (accessible by codeowners).
     When compiling new cubins for backend directories, update the corresponding path.
     """
+
     TRTLLM_GEN_FMHA: str = "463def7494c9fc6792b5aa5b5beef34025e247ac/fmha/trtllm-gen/"
     TRTLLM_GEN_BMM: str = (
         "56fea80cb22f8b2ef2a2c6a822a075fb20b36803/batched_gemm-074aec4-cc00b23"
@@ -117,6 +118,7 @@ class CheckSumHash:
     The sha256 hashes are generated in cubin publishing script logs (accessible by codeowners).
     When updating the ArtifactPath for backend directories, update the corresponding hash.
     """
+
     TRTLLM_GEN_FMHA: str = (
         "639c534614e9fdf5a9cfa91f7ea8f53989613019c0e1f8b755f461e1fcc7546f"
     )

flashinfer/jit/attention/modules.py

@@ -28,7 +28,7 @@
     sm90a_nvcc_flags,
     current_compilation_context,
 )
-from ...jit.cubin_loader import get_cubin, get_meta_hash,
+from ...jit.cubin_loader import get_cubin, get_meta_hash
 from ..utils import (
     dtype_map,
     filename_safe_dtype_map,

flashinfer/jit/cubin_loader.py

@@ -140,11 +140,12 @@ def get_meta_hash(checksums_bytes: bytes) -> str:
     """
     Parse the checksums.txt file and get the hash of corresponding flashinferMetaInfo.h file
     """
-    for line in checksums_bytes.splitlines():
+    checksums_lines = checksums_bytes.decode("utf-8").splitlines()
+    for line in checksums_lines:
         sha256, filename = line.strip().split()
         if ".h" in filename:
             return sha256
-    raise ValueError(f"Invalid checksums.txt, no flashinferMetaInfo.h found")
+    raise ValueError("Invalid checksums.txt, no flashinferMetaInfo.h found")
 
 
 def verify_cubin(cubin_path: str, expected_sha256: str) -> bool:

flashinfer/jit/fused_moe.py

@@ -185,7 +185,7 @@ def gen_trtllm_gen_fused_moe_sm100_module() -> JitSpec:
     checksum = get_cubin(checksum_path, CheckSumHash.TRTLLM_GEN_BMM)
     assert checksum, f"Failed to get checksums.txt from {checksum_path}"
     meta_hash = get_meta_hash(checksum)
-    
+
     # use `get_cubin` to get "flashinferMetaInfo.h"
     metainfo = get_cubin(
         f"{include_path}/{header_name}.h",