"""GPU benchmark entry: validate code identity before loading weights or writing data."""

import hashlib
import json
import os
import time
from pathlib import Path


def main():
    folder = Path(__file__).parent
    for name, expected in json.loads(os.environ["FLORA_EXPECTED_CODE"]).items():
        if Path(name).name != name or hashlib.sha256((folder / name).read_bytes()).hexdigest() != expected:
            raise RuntimeError("Flora image differs from the approved benchmark code")
    if os.environ.get("FLORA_EXECUTION") != "accelerated":
        raise RuntimeError("Benchmark must use the explicit accelerated profile")
    if os.environ.get("FLORA_VISION_FRAME_BATCHING") == "1":
        from flora_vision import verify_cuda_batching

        print(
            "FLORA_RUNTIME " + json.dumps({"event": "vision_sdpa_cuda_check", **verify_cuda_batching()}),
            flush=True,
        )
    os.environ["ANNOTATION_DEADLINE_EPOCH"] = str(time.time() + int(os.environ["FLORA_RUNTIME_SECONDS"]) - 60)
    from worker import run

    manifest = json.loads(os.environ["ANNOTATION_MANIFEST"])
    if manifest["provider_required"] or manifest["features"] != ["Subtasks"]:
        raise RuntimeError("Speed benchmark must be independent native Flora subtasks")
    run(manifest)


if __name__ == "__main__":
    main()