fusion_cases.py

Go to the documentation of this file.

"""Representative fused-runtime workload iterators for API tests (mirrors benchmarks fusion_case_selection)."""

from __future__ import annotations

from squander.partitioning.noisy_planner import build_phase3_continuity_partition_descriptor_set

from .continuity import build_phase2_continuity_vqe
from .runtime import build_initial_parameters
from .workloads import iter_microcase_descriptor_sets, iter_structured_descriptor_sets

CONTINUITY_QUBITS = (4, 6, 8, 10)


def iter_fusion_continuity_cases():
    for qbit_num in CONTINUITY_QUBITS:
        vqe, hamiltonian, topology = build_phase2_continuity_vqe(qbit_num)
        descriptor_set = build_phase3_continuity_partition_descriptor_set(vqe)
        parameters = build_initial_parameters(vqe.get_Parameter_Num())
        metadata = {
            "case_name": f"phase2_xxz_hea_q{qbit_num}_continuity",
            "workload_id": f"phase2_xxz_hea_q{qbit_num}_continuity",
            "case_kind": "continuity",
            "qbit_num": qbit_num,
            "topology": topology,
            "continuity_energy_real": float(vqe.Optimization_Problem(parameters)),
        }
        yield metadata, descriptor_set, parameters, hamiltonian


def iter_fusion_microcase_cases():
    for metadata, descriptor_set in iter_microcase_descriptor_sets():
        case_metadata = dict(metadata)
        case_metadata["workload_id"] = case_metadata.get("workload_id", case_metadata["case_name"])
        case_metadata["case_kind"] = "microcase"
        yield case_metadata, descriptor_set, build_initial_parameters(
            descriptor_set.parameter_count
        )


def iter_fusion_structured_cases():
    for metadata, descriptor_set in iter_structured_descriptor_sets():
        case_metadata = dict(metadata)
        case_metadata["case_kind"] = "structured_family"
        yield case_metadata, descriptor_set, build_initial_parameters(
            descriptor_set.parameter_count
        )