Запуск параметричного перебору

Версії пакетів

Код на цій сторінці розроблено з такими вимогами. Рекомендуємо використовувати ці версії або новіші.

qiskit-serverless~=0.38.0

Параметричний перебір працює як масив завдань Slurm. Викликай run() один раз для кожного набору вхідних даних, і завдання виконуються одночасно, кожне на власному вузлі. Використовуй цей самий шаблон для будь-якого набору незалежних вхідних даних, наприклад одне завдання для кожної молекули чи кожного бекенда. Завдання не можуть обмінюватися даними одне з одним, тому розділи роботу на незалежні вхідні дані.

Приклади на цій сторінці використовують робочий процес із Запуск керованого робочого процесу, який зчитує один кут, theta:

# Added by doQumentation — installs the packages this notebook needs if they are missing
import importlib.util

_needed = {"qiskit_ibm_catalog": "qiskit-ibm-catalog", "qiskit_serverless": "qiskit-serverless"}
_missing = [pip for module, pip in _needed.items()
            if importlib.util.find_spec(module) is None]
# One at a time, so a package that fails to install does not block the others
for _pip in _missing:
    %pip install -q {_pip}
if not _missing:
    print("\u2713 All required packages are installed")

# This cell is hidden from users. It creates the working directory.
from pathlib import Path

Path("./source_files").mkdir(exist_ok=True)

%%writefile ./source_files/my_workflow.py
import math

from qiskit_serverless import get_arguments, save_result

arguments = get_arguments()
theta = arguments["theta"]
if not 0 <= theta <= math.pi:
    raise ValueError(f"theta must be between 0 and pi, got {theta}")

print(f"Computing the expectation value for theta={theta}", flush=True)
save_result({"theta": theta, "expectation_value": math.cos(theta)})

Writing ./source_files/my_workflow.py

# This cell is hidden from users. It uploads the workflow under a title that
# other test notebooks do not use, and makes load("my-workflow") return it.
import logging

from qiskit_ibm_catalog import ManagedWorkflow, ManagedWorkflowClient

logging.getLogger("qiskit_ibm_runtime").setLevel(logging.ERROR)

_test_title = "docs-run-parameter-sweep"
ManagedWorkflowClient().upload(
    ManagedWorkflow(
        title=_test_title,
        entrypoint="my_workflow.py",
        working_dir="./source_files/",
    )
)
_load = ManagedWorkflowClient.load
ManagedWorkflowClient.load = lambda self, title, *a, **k: _load(
    self, _test_title if title == "my-workflow" else title, *a, **k
)

Запуск завдань

Отримай робочий процес за його назвою та викликай run() один раз для кожного входу. Збережи ідентифікатори завдань у файл: run_workflow_remotely.ipynb

import json

from qiskit_ibm_catalog import ManagedWorkflowClient

client = ManagedWorkflowClient()
my_workflow = client.load("my-workflow")

thetas = [0.1, 0.2, 0.3]

job_ids = {
    f"theta-{theta}": my_workflow.run(theta=theta).job_id for theta in thetas
}

with open("job_ids.json", "w") as f:
    json.dump(job_ids, f)

• Спочатку перевір кожен вхід. Якщо робочий процес має схему аргументів, викликай validate_arguments() для кожного входу перед циклом.

• Збережи ідентифікатори завдань у файл. Тоді ти зможеш зібрати результати з нової сесії.

• Давай файлам кожного запуску унікальну назву. Усі запуски спільно використовують файлове сховище робочого процесу. Дивись Рекомендація: давай файлам кожного запуску унікальну назву.

• Не перевищуй ліміт активних завдань. Керований робочий процес не ставить у чергу завдання понад ліміт, тож коли ти його досягнеш, запускай нові завдання лише після завершення попередніх.

Збір результатів

Отримай кожне завдання за його ідентифікатором і перехоплюй невдалі завдання, щоб одна помилка не зупиняла цикл: run_workflow_remotely.ipynb

from qiskit_serverless.exception import QiskitServerlessException

with open("job_ids.json") as f:
    job_ids = json.load(f)

results, errors = {}, {}
for run_name, sweep_job_id in job_ids.items():
    try:
        results[run_name] = client.job(sweep_job_id).result()
    except QiskitServerlessException as error:
        errors[run_name] = str(error)

print(f"{len(results)} jobs finished, {len(errors)} failed")

3 jobs finished, 0 failed

# This cell is hidden from users. It checks the sweep results.
assert len(results) == 3 and not errors, (results, errors)
assert all(results[f"theta-{t}"]["theta"] == t for t in thetas)

Щоб повторно виконати входи, які завершилися невдачею, виклич run() ще раз з тими самими аргументами.

Ліміт активних завдань

Одночасно може бути до 50 завдань, які стоять у черзі, ініціалізуються, виконуються або скасовуються. Цей ліміт враховує всі твої завдання, зокрема завдання інших робочих процесів. Якщо ти перевищиш цей ліміт, run() генерує QiskitServerlessException і не створює та не ставить завдання в чергу:

| Message: Http bad request.
| Code: 429
| Details: Active job limit reached. The maximum allowed is 50.

Наступні кроки

Рекомендації

• Керуй файлами між запусками, щоб ділитися вхідними файлами між запусками та давати файлам кожного запуску унікальну назву.
• Обери профіль обчислень, щоб запускати кожне завдання перебору на ресурсах, які відповідають його входу.