File size: 923 Bytes
af83196 | 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 | # GPU Mode: RGB to Grayscale
Evolve a Triton kernel for RGB to Grayscale conversion using SkyDiscover.
**Formula:** `Y = 0.2989 * R + 0.5870 * G + 0.1140 * B`
## Quick Start
From the repo root:
```bash
uv run skydiscover-run \
benchmarks/gpu_mode/grayscale/initial_program.py \
benchmarks/gpu_mode/grayscale/evaluator.py \
-c benchmarks/gpu_mode/grayscale/config.yaml \
-s [your_algorithm] -i 50
```
## Scoring
- **Correctness:** Must pass all test cases (rtol/atol=1e-4 vs PyTorch reference)
- **Score:** `SCORE_SCALE / geom_mean_us` where `SCORE_SCALE = 3000.0`
- Higher is better (faster runtime = higher score)
## Modal Cloud GPU Support
```bash
GPUMODE_USE_MODAL=true GPUMODE_MODAL_GPU=H100 \
uv run skydiscover-run \
benchmarks/gpu_mode/grayscale/initial_program.py \
benchmarks/gpu_mode/grayscale/evaluator.py \
-c benchmarks/gpu_mode/grayscale/config.yaml \
-s [your_algorithm] -i 50
```
|