Load ModelNew from kernel.py.
()
| 132 | |
| 133 | |
| 134 | def load_kernel(): |
| 135 | """Load ModelNew from kernel.py.""" |
| 136 | if not KERNEL_PY.exists(): |
| 137 | print("ERROR: kernel.py not found.") |
| 138 | print(" Run: uv run kernelbench/bridge.py setup --level 1 --problem 1") |
| 139 | sys.exit(1) |
| 140 | mod = _load_module_from_path(KERNEL_PY, "_kb_kernel") |
| 141 | ModelNew = getattr(mod, "ModelNew", None) |
| 142 | if ModelNew is None: |
| 143 | print("ERROR: kernel.py must define a ModelNew class.") |
| 144 | sys.exit(1) |
| 145 | get_inputs = getattr(mod, "get_inputs", None) |
| 146 | get_init_inputs = getattr(mod, "get_init_inputs", None) |
| 147 | problem_meta = getattr(mod, "KERNELBENCH_PROBLEM", {}) |
| 148 | return ModelNew, get_inputs, get_init_inputs, problem_meta |
| 149 | |
| 150 | |
| 151 | def load_metadata() -> Dict[str, Any]: |
no test coverage detected