Spaces:
Running
Running
| """Submit the included real Triton example to a running GPU environment.""" | |
| import argparse | |
| from pathlib import Path | |
| from kernel_env import KernelAction, KernelEnv | |
| parser = argparse.ArgumentParser() | |
| parser.add_argument("--url", default="http://127.0.0.1:8000") | |
| args = parser.parse_args() | |
| with KernelEnv(base_url=args.url).sync() as env: | |
| initial = env.reset(seed=42, task_id="level1/19_ReLU") | |
| Path("task_prompt.txt").write_text(initial.observation.problem) | |
| source = Path(__file__).with_name("relu.py").read_text() | |
| result = env.step(KernelAction(code=source, language="triton")) | |
| print(result.observation.model_dump_json(indent=2)) | |
| if result.observation.correct is not True or "[mock]" in result.observation.feedback: | |
| raise SystemExit("Expected a correct result from the real GPU evaluator") | |