-
Notifications
You must be signed in to change notification settings - Fork 19
/
Copy pathpy_amdgpu_mpi_wrapper.cpp
86 lines (81 loc) · 3.36 KB
/
py_amdgpu_mpi_wrapper.cpp
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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
// ---------------------------------------------------------------------------
// DM-Sim: Density-Matrix Quantum Circuit Simulation Environment
// ---------------------------------------------------------------------------
// Ang Li, Senior Computer Scientist
// Pacific Northwest National Laboratory(PNNL), U.S.
// Homepage: http://www.angliphd.com
// GitHub repo: http://www.github.com/pnnl/DM-Sim
// PNNL-IPID: 31919-E, ECCN: EAR99, IR: PNNL-SA-143160
// BSD Lincese.
// ---------------------------------------------------------------------------
// File: py_amdgpu_mpi_wrapper.cpp
// Python wrapper for the MPI implementation with AMD GPU backend.
// ---------------------------------------------------------------------------
#include <stdio.h>
#include <pybind11/pybind11.h>
#include <pybind11/numpy.h>
#include "config.hpp"
#include "util_amdgpu.hpp"
#include "dmsim_amdgpu_mpi.hpp"
namespace py = pybind11;
using namespace DMSim;
//Pybind11
PYBIND11_MODULE(libdmsim_py_amdgpu_mpi, m)
{
py::class_<Gate>(m, "Gate")
.def(py::init<enum OP,IdxType,IdxType,IdxType,
IdxType,IdxType,ValType,ValType,ValType>())
;
py::class_<Simulation>(m, "Simulation")
.def(py::init<IdxType>())
.def("append", &Simulation::append)
.def("upload", &Simulation::upload)
.def("clear_circuit", &Simulation::clear_circuit)
.def("run", &Simulation::sim)
.def("reset", &Simulation::reset)
.def("measure",[](Simulation &s, unsigned repetition) -> py::list{
IdxType* m_rtn = s.measure(repetition);
py::list rtn;
for (unsigned i=0; i<repetition; i++) rtn.append(m_rtn[i]);
delete m_rtn;
return rtn;})
.def_static("U3", &Simulation::U3)
.def_static("U2", &Simulation::U2)
.def_static("U1", &Simulation::U1)
.def_static("CX", &Simulation::CX)
.def_static("ID", &Simulation::ID)
.def_static("X", &Simulation::X)
.def_static("Y", &Simulation::Y)
.def_static("Z", &Simulation::Z)
.def_static("H", &Simulation::H)
.def_static("S", &Simulation::S)
.def_static("SDG", &Simulation::SDG)
.def_static("T", &Simulation::T)
.def_static("TDG", &Simulation::TDG)
.def_static("RX", &Simulation::RX)
.def_static("RY", &Simulation::RY)
.def_static("RZ", &Simulation::RZ)
.def_static("CZ", &Simulation::CZ)
.def_static("CY", &Simulation::CY)
.def_static("SWAP", &Simulation::SWAP)
.def_static("CH", &Simulation::CH)
.def_static("CCX", &Simulation::CCX)
.def_static("CSWAP", &Simulation::CSWAP)
.def_static("CRX", &Simulation::CRX)
.def_static("CRY", &Simulation::CRY)
.def_static("CRZ", &Simulation::CRZ)
.def_static("CU1", &Simulation::CU1)
.def_static("CU3", &Simulation::CU3)
.def_static("RXX", &Simulation::RXX)
.def_static("RZZ", &Simulation::RZZ)
.def_static("RCCX", &Simulation::RCCX)
.def_static("RC3X", &Simulation::RC3X)
.def_static("C3X", &Simulation::C3X)
.def_static("C3SQRTX", &Simulation::C3SQRTX)
.def_static("C4X", &Simulation::C4X)
.def_static("R", &Simulation::R)
.def_static("SRN", &Simulation::SRN)
.def_static("W", &Simulation::W)
.def_static("RYY", &Simulation::RYY)
;
}