← Files Runpod (Official)ARCHIVED FILE
skills/flash/evals/cpu-gpu-pipeline.eval.md
1.03 KB · Oct 5, 2026 · 18:17 UTC
# Build a CPU-preprocess then GPU-inference pipeline ## Prompt Using runpod-flash, build a two-stage pipeline: a CPU stage that cleans raw data with pandas, then a GPU stage that runs inference with torch. The CPU stage should use a compute CPU instance and the GPU stage an A100. Wire them together. ## Expected behavior The agent should: 1. Define a CPU `@Endpoint(cpu=CpuInstanceType.<type>, dependencies=["pandas"])` function 2. Define a GPU `@Endpoint(gpu=GpuGroup.AMPERE_80, dependencies=["torch"])` function 3. Import `pandas`/`torch` inside the respective functions 4. Chain them: `await infer(await preprocess(raw))` ## Assertions - CPU stage uses `cpu=CpuInstanceType.<member>` and does NOT set `gpu=` - GPU stage uses `gpu=GpuGroup.AMPERE_80` and does NOT set `cpu=` - `pandas` listed in the CPU stage `dependencies`, `torch` in the GPU stage `dependencies` - Imports are inside each decorated function - Stages are chained with `await` (the GPU call awaits the result of the CPU call) - Does NOT put `gpu=` and `cpu=` on the same Endpoint
SHA-256: 85cb8c6e02faaaff3e998d868f8b94b831aeb50069891f85fccef53275fec125