| """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() |
|
|
