Function load_from_file

kernelbench/bridge.py:446–462

Load a single problem from a standalone .py file.

(
    path: str,
    level: int = 1,
    problem_id: int = 0,
)

def load_from_file(
    path: str,
    level: int = 1,
    problem_id: int = 0,
) -> KernelBenchProblem:
    """Load a single problem from a standalone .py file."""
    p = Path(path)
    if not p.exists():
        print(f"ERROR: File not found: {path}")
        sys.exit(1)
    source = p.read_text(encoding="utf-8")
    name = p.stem.replace("_", " ")
    prob = KernelBenchProblem(
        level=level, problem_id=problem_id, name=name, source_code=source,
    )
    prob.save_to_cache()
    return prob


# ---------------------------------------------------------------------------