infra-bench/infra-bench-v1
20 AI-infra engineering tasks on AMD MI35X (5 categories: model deployment, profiling, kernel implementation, kernel tuning, debug triage)
Published 7/17/2026 by Xiake Sun
harbor run -d infra-bench/infra-bench-v1InfraBench
Github repository: InfraBench
20 real-world AI-infrastructure engineering tasks on AMD MI35X (CDNA4 / gfx950) GPUs, across five categories: model deployment, profiling analysis, kernel implementation, kernel tuning, and debug triage. Each task runs an agent in a Docker sandbox and is scored by a deterministic per-task verifier.
Built on Harbor framework via the harbor CLI.
Requirements
- AMD ROCm GPUs (
/dev/kfd+/dev/dri) and Docker harborCLI and an LLM-gateway-authenticated agent- Model weights at
/data/models(mounted read-only into model-dependent tasks)
export INFRABENCH_GPU_COUNT=8
harbor run -d infra-bench/infra-bench-v1 -a claude-code -m <model> -n 6
GPU Pool for Concurrency:
- Two tasks need 2 GPUs (
qr-rmsnorm-fusion,sglang-mmmu-ipc-crash); - the GPU-pool allocator is non-blocking, so keep
INFRA_PARALLEL ≤ INFRABENCH_GPU_COUNT - 2. - All concurrent trials must run on a single host (pool arbitration is filesystem
flock).
Results & scoring
- Output lands in
jobs/<timestamp>/.verifier/reward.txtis the score (1= pass,0= fail); - Use
harbor view jobsto browse trajectories andharbor analyze jobsto analyze failures.
| Task |
|---|
infra-bench/gemm-fp8-ptpc-quant |
infra-bench/cute-layout-composition |
infra-bench/flydsl-cdna4-preshuffle-gemm |
infra-bench/dwconv3d-occupancy |
infra-bench/gemma4-gemm-tuning |
infra-bench/engram-triton-kernel |
infra-bench/mem-bandwidth-bench |
infra-bench/hotspot-analysis-torch-profiler |
infra-bench/qr-rmsnorm-fusion |
infra-bench/triton-matmul-tuning |
infra-bench/gemma4-sglang-serving-opt |
infra-bench/hello-rocm |
infra-bench/llm-fp8-quantize |
infra-bench/pointnet2-hipify |
infra-bench/sglang-mmmu-ipc-crash |
infra-bench/sglang-sync-stall |
infra-bench/llvm-simple-constant-propagation |
infra-bench/paged-attention-hd256 |
infra-bench/vllm-aiter-debug |
infra-bench/gluon-a8w8-mfma-att |
Displaying 20 of 20 tasks