Class TestCompile

tests/sol_execbench/driver/test_problem_packager.py:186–235

class TestCompile:
    def test_returns_command_and_artifact_path(
        self, tmp_path, definition, workloads, cuda_solution, config
    ):
        pkg = _make_packager(tmp_path, definition, workloads, cuda_solution, config)
        cmd, artifact_path = pkg.compile()
        assert cmd == ["python", "build_ext.py"]
        assert artifact_path.endswith("benchmark_kernel.so")

    def test_build_ext_staged(
        self, tmp_path, definition, workloads, cuda_solution, config
    ):
        pkg = _make_packager(tmp_path, definition, workloads, cuda_solution, config)
        pkg.compile()
        build_ext = pkg.output_dir / "build_ext.py"
        assert build_ext.exists()
        ast.parse(build_ext.read_text())

    def test_gencode_injected_for_blackwell(
        self, tmp_path, definition, workloads, cuda_solution, config
    ):
        pkg = _make_packager(tmp_path, definition, workloads, cuda_solution, config)
        pkg.compile()
        sol = json.loads((pkg.output_dir / "solution.json").read_text())
        cuda_cflags = sol["spec"]["compile_options"]["cuda_cflags"]
        assert any("sm_100a" in f for f in cuda_cflags)

    def test_gencode_not_injected_when_explicit(
        self, tmp_path, definition, workloads, config
    ):
        sol_dict = {
            **_CUDA_SOLUTION_DICT,
            "spec": {
                **_CUDA_SOLUTION_DICT["spec"],
                "compile_options": {"cuda_cflags": ["-arch=sm_90"]},
            },
        }
        solution = Solution(**sol_dict)
        pkg = _make_packager(tmp_path, definition, workloads, solution, config)
        pkg.compile()
        sol = json.loads((pkg.output_dir / "solution.json").read_text())
        cuda_cflags = sol["spec"]["compile_options"]["cuda_cflags"]
        assert cuda_cflags == ["-arch=sm_90"]

    def test_asserts_on_python_solution(
        self, tmp_path, definition, workloads, python_solution, config
    ):
        pkg = _make_packager(tmp_path, definition, workloads, python_solution, config)
        with pytest.raises(AssertionError, match="C\\+\\+/CUDA"):
            pkg.compile()


# ── execute() ─────────────────────────────────────────────────────────────────