aboutsummaryrefslogtreecommitdiff
path: root/include/core/runtime.h
blob: cf0337198656d99a5c2faa120d5fa6b4b3649f14 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
#pragma once

#include <stdbool.h>
#include <stddef.h>

#include "core/circuit.h"
#include "maths/vector.h"

#define PSI_PARALLEL_THRESHOLD 8

struct PsiRuntimeConfig
{
    bool parallel;
    bool simd;
    bool batched;
    bool structure_aware;
    size_t parallel_threshold;
};

enum PsiRuntime
{
    PSI_RUNTIME_BASIC,
    PSI_RUNTIME_BASIC_MT,
    PSI_RUNTIME_BATCHED,
    PSI_RUNTIME_BATCHED_MT,
    PSI_RUNTIME_SIMD,
    PSI_RUNTIME_SIMD_MT,
    PSI_RUNTIME_STRUCTURE_AWARE,
    PSI_RUNTIME_STRUCTURE_AWARE_MT,
};

struct PsiRuntimeConfig psi_new_runtime_config(void);
struct PsiRuntimeConfig psi_optimal_runtime_config(void);
struct PsiRuntimeConfig psi_runtime_to_config(enum PsiRuntime runtime);

struct PsiVector psi_compute_runtime_config(struct PsiRuntimeConfig config, size_t num_qubits,
                                            const struct PsiGateOp* operations, size_t op_count);
struct PsiVector psi_compute_runtime(enum PsiRuntime runtime, size_t num_qubits,
                                     const struct PsiGateOp* operations, size_t op_count);

const struct PsiVector* psi_compute_circuit(struct PsiQuantumCircuit* circuit);
const struct PsiVector* psi_compute_circuit_with(struct PsiQuantumCircuit* circuit,
                                                 enum PsiRuntime runtime);
const struct PsiVector* psi_compute_circuit_with_config(struct PsiQuantumCircuit* circuit,
                                                        struct PsiRuntimeConfig config);
double psi_circuit_probability(struct PsiQuantumCircuit* circuit, size_t state_index);
void psi_circuit_probabilities(struct PsiQuantumCircuit* circuit, double* out);