Spaces:
Sleeping
Sleeping
File size: 1,639 Bytes
07fb183 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 | """Verify Wave-4 CUDA kernel extraction preserved kernel bytes exactly."""
from __future__ import annotations
import hashlib
from pathlib import Path
ROOT = Path(__file__).resolve().parents[1]
PAIRS = (
(
ROOT / "src" / "playground" / "cuda" / "recurrent.cu",
ROOT / "src" / "acceleration" / "cuda" / "recurrent" / "kernel.cu",
),
(
ROOT / "src" / "playground" / "cuda" / "builder_synapses.cu",
ROOT / "src" / "acceleration" / "cuda" / "plasticity" / "kernel.cu",
),
)
def sha256(path: Path) -> str:
return hashlib.sha256(path.read_bytes()).hexdigest()
def kernel_mismatches() -> list[str]:
mismatches: list[str] = []
for legacy, canonical in PAIRS:
if not legacy.is_file() or not canonical.is_file():
mismatches.append(
f"missing kernel pair: {legacy.relative_to(ROOT)} / "
f"{canonical.relative_to(ROOT)}"
)
continue
if legacy.read_bytes() != canonical.read_bytes():
mismatches.append(
f"kernel changed during extraction: "
f"{legacy.relative_to(ROOT)}={sha256(legacy)} "
f"{canonical.relative_to(ROOT)}={sha256(canonical)}"
)
return mismatches
def main() -> int:
mismatches = kernel_mismatches()
if mismatches:
raise SystemExit("\n".join(mismatches))
for legacy, canonical in PAIRS:
print(
f"kernel freeze PASS: {canonical.relative_to(ROOT)} "
f"sha256={sha256(canonical)}"
)
return 0
if __name__ == "__main__":
raise SystemExit(main())
|