← Files Runpod (Official)ARCHIVED FILE

skills/flash/evals/cpu-gpu-pipeline.eval.md

1.03 KB · Oct 5, 2026 · 18:17 UTC

↓ Download file

# 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