Commit efeb7ed8 authored by Nguyen, Thien Minh's avatar Nguyen, Thien Minh
Browse files

Add ability to configure quench model params from command line 



and need to support passing a list of int from Python to HetMap (initial spins)

Signed-off-by: default avatarThien Nguyen <nguyentm@ornl.gov>
parent 74c0ad03

examples/qsim/demo/HeisenbergModel/QuantumQuenchSimulation.cpp

+46 −8
Original line number Diff line number Diff line
@@ -7,21 +7,59 @@ 
/// $ qcor -qpu qsim QuantumQuenchSimulation.cpp

/// $ ./a.out

int main(int argc, char **argv) {

  // Handle input parameters:

  // --n-spins : number of spins/qubits (default = 7)

  // --g: g parameter - ZZ coupling strength (default = 0.0)

  // --dt: Trotter dt (default = 0.05)

  // --steps: number of steps (defalt = 100)



  // Process the input arguments

  std::vector<std::string> arguments(argv + 1, argv + argc);

  int n_spins = 7;

  double g = 0.0;

  double dt = 0.05;

  int n_steps = 100;



  for (int i = 0; i < arguments.size(); i++) {

    if (arguments[i] == "--n-spins") {

      n_spins = std::stoi(arguments[i + 1]);

    }

    if (arguments[i] == "--g") {

      g = std::stod(arguments[i + 1]);

    }

    if (arguments[i] == "--dt") {

      dt = std::stod(arguments[i + 1]);

    }

    if (arguments[i] == "--steps") {

      n_steps = std::stoi(arguments[i + 1]);

    }

  }



  std::cout << "Running QSim for a quenching model with " << n_spins

            << " spins and g = " << g << "\n";

  std::cout << "dt " << dt << "; number of steps = " << n_steps << "\n";

  using ModelType = qcor::qsim::ModelBuilder::ModelType;

  // Initial spin state: Neel state (7 qubits/spins)

  const std::vector<int> initial_spins{0, 1, 0, 1, 0, 1, 0};

  // Initial spin state: Neel state

  std::vector<int> initial_spins;

  for (int i = 0; i < n_spins; ++i) {

    // Alternating up/down spins

    initial_spins.emplace_back(i % 2);

  }



  auto problemModel = qsim::ModelBuilder::createModel(

      ModelType::Heisenberg, {{"Jx", 1.0},

                              {"Jy", 1.0},

                              {"Jz", 0.2},

                              // Jz == g parameter

                              {"Jz", g},

                              // No external field

                              {"h_ext", 0.0},

                              {"ext_dir", "X"},

                              {"num_spins", 7},

                              {"num_spins", n_spins},

                              {"initial_spins", initial_spins},

                              {"observable", "staggered_magnetization"}});

  // Workflow parameters:

  auto workflow = qsim::getWorkflow(

      "td-evolution", {{"dt", 0.05}, {"steps", 100}});

  auto workflow =

      qsim::getWorkflow("td-evolution", {{"dt", dt}, {"steps", n_steps}});



  // Result should contain the observable expectation value along Trotter steps.

  auto result = workflow->execute(problemModel);

python/py-qcor.cpp

+1 −1
Original line number Diff line number Diff line
@@ -80,7 +80,7 @@ class KernelArgDictToHeterogeneousMap { 


// Add type name to this list to support receiving from Python.

using PyHeterogeneousMapTypes =

    xacc::Variant<bool, int, double, std::string,

    xacc::Variant<bool, int, double, std::string, std::vector<int>,

                  std::shared_ptr<qcor::Optimizer>, std::vector<double>, std::vector<std::vector<double>>>;

using PyHeterogeneousMap = std::map<std::string, PyHeterogeneousMapTypes>;