File size: 1,664 Bytes
3e5a997 | 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 | """Run the validated worker from Studio's image using its isolated CUDA environment."""
import hashlib
import json
import os
import tempfile
from pathlib import Path
def verify_runtime():
from flora_acceleration import require_fast_kernels
versions = require_fast_kernels()
if versions != json.loads(os.environ["FLORA_EXPECTED_KERNELS"]):
raise RuntimeError("Studio's CUDA dependency versions differ from the validated Flora runtime")
files = {
**json.loads(os.environ["FLORA_EXPECTED_CODE"]),
**json.loads(os.environ["FLORA_EXPECTED_RUNTIME_FILES"]),
}
for name, expected in files.items():
if (
Path(name).name != name
or hashlib.sha256(Path(__file__).with_name(name).read_bytes()).hexdigest() != expected
):
raise RuntimeError("Studio's Flora runtime differs from its validated source")
return versions
def main():
versions = verify_runtime()
if os.environ.get("FLORA_SMOKE_ONLY") == "1":
from flora_vision import verify_cuda_batching
with tempfile.TemporaryDirectory(dir="/workspace") as folder:
Path(folder, "writable").write_text("ok")
print(
json.dumps(
{
"event": "integrated_flora_smoke",
"kernels": versions,
"cuda_attention": verify_cuda_batching(),
"workspace_writable": True,
}
),
flush=True,
)
return
from flora_speed_entry import main as run_validated_worker
run_validated_worker()
if __name__ == "__main__":
main()
|