Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Set ExecutionSpace #197

Closed
wants to merge 15 commits into from
Closed

Set ExecutionSpace #197

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
15 commits
Select commit Hold shift + click to select a range
922df0f
templatized statevector
gandalfr-KY Jan 14, 2025
bdd0381
Merge branch 'main' into set-space
gandalfr-KY Jan 17, 2025
120de54
templatize update
gandalfr-KY Jan 17, 2025
0989281
templatize gate
gandalfr-KY Jan 17, 2025
1391e2b
cpu-statevector on gpu
gandalfr-KY Jan 17, 2025
501a367
fix xgate
gandalfr-KY Jan 21, 2025
3f2221c
update function ok
gandalfr-KY Jan 21, 2025
2b60e14
complete templatize
gandalfr-KY Jan 21, 2025
3642eb2
fixing test
gandalfr-KY Jan 21, 2025
ac2197c
fix statevector test
gandalfr-KY Jan 24, 2025
9420d56
Merge branch 'main' into set-space
gandalfr-KY Jan 24, 2025
137fb37
add test
gandalfr-KY Jan 28, 2025
e2b8bbf
operator test 一旦 ok
gandalfr-KY Jan 28, 2025
2498663
gate test ok
gandalfr-KY Jan 28, 2025
626bc06
param_gate test ok
gandalfr-KY Jan 31, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -119,8 +119,9 @@ if ((${CMAKE_CXX_COMPILER_ID} STREQUAL "GNU") OR (${CMAKE_CXX_COMPILER_ID} STREQ
if(SCALUQ_USE_CUDA)
# to remove tremendous warnings of Eigen
target_compile_options(scaluq PUBLIC
-Wno-unknown-pragmas
--expt-relaxed-constexpr
-Wno-unknown-pragmas
-Wno-deprecated-enum-enum-conversion
Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Eigenのwarningを抑制します．
まだ警告が残っているのでTODO

-DEIGEN_NO_CUDA
)
endif()
Expand Down
362 changes: 224 additions & 138 deletions exe/main.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,159 +7,245 @@ using namespace nlohmann;

int main() {
scaluq::initialize(); // must be called before using any scaluq methods

{
std::uint64_t n_qubits = 3;
scaluq::StateVector<double> state(n_qubits);
state.load({0, 1, 2, 3, 4, 5, 6, 7});
Json j = state;
std::cout << j << std::endl;
state = j;
std::cout << state << std::endl;

scaluq::Circuit<double> circuit(n_qubits);
circuit.add_gate(scaluq::gate::X<double>(0));
circuit.add_gate(scaluq::gate::CNot<double>(0, 1));
circuit.add_gate(scaluq::gate::Y<double>(1));
circuit.add_gate(scaluq::gate::RX<double>(1, std::numbers::pi / 2));
circuit.update_quantum_state(state);

scaluq::Operator<double> observable(n_qubits);
observable.add_random_operator(1, 0);
auto value = observable.get_expectation_value(state);
std::cout << value << std::endl;
}
{
std::uint64_t n_qubits = 2, batch_size = 2;
scaluq::StateVectorBatched<double> states(batch_size, n_qubits);
states.set_Haar_random_state(batch_size, n_qubits, false);

Circuit<double> cirq(n_qubits);
cirq.add_param_gate(gate::ParamRX<double>(1, {2}), "RX");
std::map<std::string, std::vector<double>> params = {{"RX", {0.5, 1.5}}};
cirq.update_quantum_state(states, params);

Json j = states;
std::cout << j << std::endl;
states = j;
std::cout << states << std::endl;
}
{
double coef = 2.0;
std::string pauli_string = "X 0 Z 1 Y 2";
PauliOperator<double> pauli(pauli_string, coef);
Json j = pauli;
std::cout << j << std::endl;
pauli = j;
}
{
std::uint64_t n_qubits = 3;
Operator<double> op(n_qubits);
op.add_operator({0b001, 0b010, Complex<double>(2)});
op.add_operator({"X 2 Y 1", 1});
Json j = op;
std::cout << j << std::endl;
op = j;
}
{
std::cout << Json(gate::I<double>()) << std::endl;
std::cout << Json(gate::X<double>(2, {0, 3})) << std::endl;
std::cout << Json(gate::Y<double>(2, {0, 3})) << std::endl;
std::cout << Json(gate::Z<double>(2, {0, 3})) << std::endl;
std::cout << Json(gate::H<double>(2, {0, 3})) << std::endl;
std::cout << Json(gate::S<double>(2, {0, 3})) << std::endl;
std::cout << Json(gate::Sdag<double>(2, {0, 3})) << std::endl;
std::cout << Json(gate::T<double>(2, {0, 3})) << std::endl;
std::cout << Json(gate::Tdag<double>(2, {0, 3})) << std::endl;
std::cout << Json(gate::SqrtX<double>(2, {0, 3})) << std::endl;
std::cout << Json(gate::SqrtXdag<double>(2, {0, 3})) << std::endl;
std::cout << Json(gate::SqrtY<double>(2, {0, 3})) << std::endl;
std::cout << Json(gate::SqrtYdag<double>(2, {0, 3})) << std::endl;
std::cout << Json(gate::RX<double>(2, 0.5, {0, 3})) << std::endl;
std::cout << Json(gate::RY<double>(2, 0.5, {0, 3})) << std::endl;
std::cout << Json(gate::RZ<double>(2, 0.5, {0, 3})) << std::endl;
std::cout << Json(gate::U1<double>(2, 0.5, {0, 3})) << std::endl;
std::cout << Json(gate::U2<double>(2, 0.5, 0.3, {0, 3})) << std::endl;
std::cout << Json(gate::U3<double>(2, 0.5, 0.3, 0.1, {0, 3})) << std::endl;
std::cout << Json(gate::Swap<double>(1, 2, {0, 3})) << std::endl;

PauliOperator<double> pauli("X 2 Y 1");
std::cout << Json(gate::Pauli<double>(pauli)) << std::endl;
std::cout << Json(gate::PauliRotation<double>(pauli, 0.5)) << std::endl;

auto probgate =
gate::Probablistic<double>({.1, .9}, {gate::X<double>(0), gate::I<double>()});
std::cout << Json(probgate) << std::endl;

std::cout << Json(gate::ParamRX<double>(2, 1.5, {0, 3})) << std::endl;
std::cout << Json(gate::ParamRY<double>(2, 1.5, {0, 3})) << std::endl;
std::cout << Json(gate::ParamRZ<double>(2, 1.5, {0, 3})) << std::endl;
std::cout << Json(gate::ParamPauliRotation<double>(pauli, 0.5)) << std::endl;

auto paramprobgate = gate::ParamProbablistic<double>(
{.1, .9}, {gate::ParamRX<double>(0), gate::I<double>()});
std::cout << Json(paramprobgate) << std::endl;
}
{
auto x = gate::X<double>(1, {2});
Json j = x;
std::cout << j << std::endl;
Gate<double> gate = j;
std::cout << gate << std::endl;
Kokkos::Timer tm;
std::uint64_t n_qubits = 28;
scaluq::StateVector<double, DefaultSpace> gpu_state(n_qubits);

for (int i = 0; i < 10; ++i) {
auto xgate = scaluq::gate::X<double, DefaultSpace>(2);
xgate->update_quantum_state(gpu_state);
}
std::cout << tm.seconds() << std::endl;
}
{
auto x = gate::RX<double>(1, 0.5, {2});
Json j = x;
std::cout << j << std::endl;
Gate<double> gate = j;
std::cout << gate << std::endl;
Kokkos::Timer tm;
std::uint64_t n_qubits = 28;
scaluq::StateVector<double, CPUSpace> cpu_state(n_qubits);

for (int i = 0; i < 10; ++i) {
auto xgate = scaluq::gate::X<double, CPUSpace>(2);
xgate->update_quantum_state(cpu_state);
}
std::cout << tm.seconds() << std::endl;
}
{
auto x = gate::Swap<double>(1, 3, {2});
Json j = x;
std::cout << j << std::endl;
Gate<double> gate = j;
std::cout << gate << std::endl;
Kokkos::Timer tm;
std::uint64_t n_qubits = 28;
scaluq::StateVector<float, DefaultSpace> gpu_state(n_qubits);

for (int i = 0; i < 10; ++i) {
auto xgate = scaluq::gate::X<float, DefaultSpace>(2);
xgate->update_quantum_state(gpu_state);
}
std::cout << tm.seconds() << std::endl;
}
{
PauliOperator<double> pauli("X 2 Y 1");
auto x = gate::PauliRotation<double>(pauli, 1.5, {0, 3});
Json j = x;
std::cout << j << std::endl;
Gate<double> gate = j;
std::cout << gate << std::endl;
Kokkos::Timer tm;
std::uint64_t n_qubits = 28;
scaluq::StateVector<float, CPUSpace> cpu_state(n_qubits);

for (int i = 0; i < 10; ++i) {
auto xgate = scaluq::gate::X<float, CPUSpace>(2);
xgate->update_quantum_state(cpu_state);
}
std::cout << tm.seconds() << std::endl;
}

{
auto probgate =
gate::Probablistic<double>({.1, .9}, {gate::X<double>(0, {2, 3}), gate::I<double>()});
Json j = probgate;
std::cout << j << std::endl;
Gate<double> gate = j;
std::cout << gate << std::endl;
Kokkos::Timer tm;
std::uint64_t n_qubits = 5;
scaluq::StateVector<double, DefaultSpace> gpu_state(n_qubits);

for (int i = 0; i < 1000; ++i) {
auto xgate = scaluq::gate::X<double, DefaultSpace>(2);
xgate->update_quantum_state(gpu_state);
}
std::cout << tm.seconds() << std::endl;
}
{
auto x = gate::ParamRX<double>(1, {2});
Json j = x;
std::cout << j << std::endl;
ParamGate<double> gate = j;
std::cout << gate << std::endl;
Kokkos::Timer tm;
std::uint64_t n_qubits = 5;
scaluq::StateVector<double, CPUSpace> cpu_state(n_qubits);

for (int i = 0; i < 1000; ++i) {
auto xgate = scaluq::gate::X<double, CPUSpace>(2);
xgate->update_quantum_state(cpu_state);
}
std::cout << tm.seconds() << std::endl;
}
{
auto paramprobgate = gate::ParamProbablistic<double>(
{.1, .9}, {gate::ParamRX<double>(0), gate::I<double>()});
Json j = paramprobgate;
std::cout << j << std::endl;
ParamGate<double> gate = j;
std::cout << gate << std::endl;
Kokkos::Timer tm;
std::uint64_t n_qubits = 5;
scaluq::StateVector<float, DefaultSpace> gpu_state(n_qubits);

for (int i = 0; i < 1000; ++i) {
auto xgate = scaluq::gate::X<float, DefaultSpace>(2);
xgate->update_quantum_state(gpu_state);
}
std::cout << tm.seconds() << std::endl;
}
{
Circuit<double> circuit(10);
circuit.add_gate(gate::X<double>(0, {3}));
circuit.add_param_gate(gate::ParamRX<double>(0, 0.5, {3}), "RX");
Json j = circuit;
std::cout << j << std::endl;
Circuit<double> c = j;
std::cout << Json(c) << std::endl;
std::uint64_t n_qubits = 5;
scaluq::StateVector<float, CPUSpace> cpu_state(n_qubits);

Kokkos::Timer tm;
for (int i = 0; i < 1000; ++i) {
auto xgate = scaluq::gate::X<float, CPUSpace>(2);
xgate->update_quantum_state(cpu_state);
}
std::cout << tm.seconds() << std::endl;
}

scaluq::finalize();
// {
// std::uint64_t n_qubits = 3;
// scaluq::StateVector<double> state(n_qubits);
// state.load({0, 1, 2, 3, 4, 5, 6, 7});
// Json j = state;
// std::cout << j << std::endl;
// state = j;
// std::cout << state << std::endl;

// scaluq::Circuit<double> circuit(n_qubits);
// circuit.add_gate(scaluq::gate::X<double>(0));
// circuit.add_gate(scaluq::gate::CNot<double>(0, 1));
// circuit.add_gate(scaluq::gate::Y<double>(1));
// circuit.add_gate(scaluq::gate::RX<double>(1, std::numbers::pi / 2));
// circuit.update_quantum_state(state);

// scaluq::Operator<double> observable(n_qubits);
// observable.add_random_operator(1, 0);
// auto value = observable.get_expectation_value(state);
// std::cout << value << std::endl;
// }
// {
// std::uint64_t n_qubits = 2, batch_size = 2;
// scaluq::StateVectorBatched<double> states(batch_size, n_qubits);
// states.set_Haar_random_state(batch_size, n_qubits, false);
// Json j = states;
// std::cout << j << std::endl;
// states = j;
// std::cout << states << std::endl;
// }
// {
// double coef = 2.0;
// std::string pauli_string = "X 0 Z 1 Y 2";
// PauliOperator<double> pauli(pauli_string, coef);
// Json j = pauli;
// std::cout << j << std::endl;
// pauli = j;
// }
// {
// std::uint64_t n_qubits = 3;
// Operator<double> op(n_qubits);
// op.add_operator({0b001, 0b010, Complex<double>(2)});
// op.add_operator({"X 2 Y 1", 1});
// Json j = op;
// std::cout << j << std::endl;
// op = j;
// }
// {
// std::cout << Json(gate::I<double>()) << std::endl;
// std::cout << Json(gate::X<double>(2, {0, 3})) << std::endl;
// std::cout << Json(gate::Y<double>(2, {0, 3})) << std::endl;
// std::cout << Json(gate::Z<double>(2, {0, 3})) << std::endl;
// std::cout << Json(gate::H<double>(2, {0, 3})) << std::endl;
// std::cout << Json(gate::S<double>(2, {0, 3})) << std::endl;
// std::cout << Json(gate::Sdag<double>(2, {0, 3})) << std::endl;
// std::cout << Json(gate::T<double>(2, {0, 3})) << std::endl;
// std::cout << Json(gate::Tdag<double>(2, {0, 3})) << std::endl;
// std::cout << Json(gate::SqrtX<double>(2, {0, 3})) << std::endl;
// std::cout << Json(gate::SqrtXdag<double>(2, {0, 3})) << std::endl;
// std::cout << Json(gate::SqrtY<double>(2, {0, 3})) << std::endl;
// std::cout << Json(gate::SqrtYdag<double>(2, {0, 3})) << std::endl;
// std::cout << Json(gate::RX<double>(2, 0.5, {0, 3})) << std::endl;
// std::cout << Json(gate::RY<double>(2, 0.5, {0, 3})) << std::endl;
// std::cout << Json(gate::RZ<double>(2, 0.5, {0, 3})) << std::endl;
// std::cout << Json(gate::U1<double>(2, 0.5, {0, 3})) << std::endl;
// std::cout << Json(gate::U2<double>(2, 0.5, 0.3, {0, 3})) << std::endl;
// std::cout << Json(gate::U3<double>(2, 0.5, 0.3, 0.1, {0, 3})) << std::endl;
// std::cout << Json(gate::Swap<double>(1, 2, {0, 3})) << std::endl;

// PauliOperator<double> pauli("X 2 Y 1");
// std::cout << Json(gate::Pauli<double>(pauli)) << std::endl;
// std::cout << Json(gate::PauliRotation<double>(pauli, 0.5)) << std::endl;

// auto probgate =
// gate::Probablistic<double>({.1, .9}, {gate::X<double>(0), gate::I<double>()});
// std::cout << Json(probgate) << std::endl;

// std::cout << Json(gate::ParamRX<double>(2, 1.5, {0, 3})) << std::endl;
// std::cout << Json(gate::ParamRY<double>(2, 1.5, {0, 3})) << std::endl;
// std::cout << Json(gate::ParamRZ<double>(2, 1.5, {0, 3})) << std::endl;
// std::cout << Json(gate::ParamPauliRotation<double>(pauli, 0.5)) << std::endl;

// auto paramprobgate = gate::ParamProbablistic<double>(
// {.1, .9}, {gate::ParamRX<double>(0), gate::I<double>()});
// std::cout << Json(paramprobgate) << std::endl;
// }
// {
// auto x = gate::X<double>(1, {2});
// Json j = x;
// std::cout << j << std::endl;
// Gate<double> gate = j;
// std::cout << gate << std::endl;
// }
// {
// auto x = gate::RX<double>(1, 0.5, {2});
// Json j = x;
// std::cout << j << std::endl;
// Gate<double> gate = j;
// std::cout << gate << std::endl;
// }
// {
// auto x = gate::Swap<double>(1, 3, {2});
// Json j = x;
// std::cout << j << std::endl;
// Gate<double> gate = j;
// std::cout << gate << std::endl;
// }
// {
// PauliOperator<double> pauli("X 2 Y 1");
// auto x = gate::PauliRotation<double>(pauli, 1.5, {0, 3});
// Json j = x;
// std::cout << j << std::endl;
// Gate<double> gate = j;
// std::cout << gate << std::endl;
// }
// {
// auto probgate =
// gate::Probablistic<double>({.1, .9}, {gate::X<double>(0, {2, 3}),
// gate::I<double>()});
// Json j = probgate;
// std::cout << j << std::endl;
// Gate<double> gate = j;
// std::cout << gate << std::endl;
// }
// {
// auto x = gate::ParamRX<double>(1, {2});
// Json j = x;
// std::cout << j << std::endl;
// ParamGate<double> gate = j;
// std::cout << gate << std::endl;
// }
// {
// auto paramprobgate = gate::ParamProbablistic<double>(
// {.1, .9}, {gate::ParamRX<double>(0), gate::I<double>()});
// Json j = paramprobgate;
// std::cout << j << std::endl;
// ParamGate<double> gate = j;
// std::cout << gate << std::endl;
// }
// {
// Circuit<double> circuit(10);
// circuit.add_gate(gate::X<double>(0, {3}));
// circuit.add_param_gate(gate::ParamRX<double>(0, 0.5, {3}), "RX");
// Json j = circuit;
// std::cout << j << std::endl;
// Circuit<double> c = j;
// std::cout << Json(c) << std::endl;
// }

Kokkos::finalize();
}
Loading
Loading