Vincent Bourdon
43 lines
1.1 KiB
Rust
43 lines
1.1 KiB
Rust
use clap::Parser;
|
|
use spinoza::core::reservoir_sampling;
|
|
use spinoza::{
|
|
config::{Config, QSArgs},
|
|
core::{iqft, State, CONFIG},
|
|
gates::{apply, Gate},
|
|
math::{pow2f, PI},
|
|
utils::{pretty_print_int, to_table},
|
|
};
|
|
|
|
fn run(n: usize, show_results: bool) {
|
|
let now = std::time::Instant::now();
|
|
let mut state = State::new(n);
|
|
|
|
let v = 2.4;
|
|
for i in 0..n {
|
|
apply(Gate::H, &mut state, i);
|
|
}
|
|
for i in 0..n {
|
|
apply(Gate::P(2.0 * PI / (pow2f(i + 1)) * v), &mut state, i);
|
|
}
|
|
let targets: Vec<usize> = (0..n).rev().collect();
|
|
|
|
iqft(&mut state, &targets);
|
|
|
|
let elapsed = now.elapsed().as_micros();
|
|
println!("{}", pretty_print_int(elapsed));
|
|
if show_results {
|
|
println!("{}", to_table(&state));
|
|
}
|
|
|
|
let reservoir = reservoir_sampling(&state, state.len(), state.len() * 100_000);
|
|
let histogram = reservoir.get_outcome_count();
|
|
println!("{:?}", histogram);
|
|
}
|
|
|
|
fn main() {
|
|
let args = QSArgs::parse();
|
|
let config = Config::from_cli(args);
|
|
CONFIG.set(config).unwrap();
|
|
run(config.qubits.into(), config.print);
|
|
}
|