002f9879b2
Replaces cherry-picked upstream_ref with complete source trees. xllm/ — Iluvatar official C++ inference engine (15MB, 1470 files) Complete: kernels → layers → models → runtime → scheduler → api Excluded: .git, binary images, third_party submodule checkouts ds_vllm/ — Iluvatar official vllm fork (8MB, 703 files) Included: csrc/ (ALL CUDA kernels), fused_moe/, qwen3_5 model, _custom_ops Excluded: tests, benchmarks, docs, examples (not needed for reference) Critical call chains now fully traceable: MoE: moe_topk_softmax_kernels.cuh → ixformer.h → fused_moe.cpp → layer GDN: qwen3_gated_delta_net_base.cpp → qwen3_5_gated_delta_net.cpp Attention: ixformer.h → xllm_paged_attention → attention.cpp
51 lines
1.3 KiB
Python
51 lines
1.3 KiB
Python
from __future__ import annotations
|
|
|
|
import hashlib
|
|
import json
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
from .manifest import KernelFamilyManifest
|
|
from .spec import KernelCompileSpec
|
|
from .toolchain import sha256_file
|
|
|
|
|
|
def compute_cache_key(
|
|
spec: KernelCompileSpec,
|
|
fingerprint: dict[str, Any],
|
|
dependency_files: list[str | Path],
|
|
) -> str:
|
|
payload = {
|
|
"spec": spec.cache_key_material(),
|
|
"fingerprint": fingerprint,
|
|
"dependencies": {
|
|
str(Path(path).resolve()): sha256_file(path) for path in dependency_files
|
|
},
|
|
}
|
|
encoded = json.dumps(payload, sort_keys=True, separators=(",", ":")).encode(
|
|
"utf-8"
|
|
)
|
|
return hashlib.sha256(encoded).hexdigest()
|
|
|
|
|
|
def is_cache_hit(
|
|
manifest_path: str | Path, variant_key: str, expected_cache_key: str
|
|
) -> bool:
|
|
path = Path(manifest_path)
|
|
if not path.is_file():
|
|
return False
|
|
|
|
try:
|
|
manifest = KernelFamilyManifest.read(path)
|
|
except Exception:
|
|
return False
|
|
|
|
variant = manifest.get_variant(variant_key)
|
|
if variant is None:
|
|
return False
|
|
|
|
if variant.cache_key != expected_cache_key:
|
|
return False
|
|
|
|
return Path(variant.generated_source).is_file() and Path(variant.compiled_binary).is_file()
|