teleport_scaffold.cpp 3.27 KB
Newer Older
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
/***********************************************************************************
 * Copyright (c) 2016, UT-Battelle
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *   * Redistributions of source code must retain the above copyright
 *     notice, this list of conditions and the following disclaimer.
 *   * Redistributions in binary form must reproduce the above copyright
 *     notice, this list of conditions and the following disclaimer in the
 *     documentation and/or other materials provided with the distribution.
 *   * Neither the name of the xacc nor the
 *     names of its contributors may be used to endorse or promote products
 *     derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL <COPYRIGHT HOLDER> BE LIABLE FOR ANY
 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 * Contributors:
 *   Initial API and implementation - Alex McCaskey
 *
 **********************************************************************************/
31
#include "XACC.hpp"
32
#include "FireTensorAccelerator.hpp"
33
34
35

// Quantum Kernel executing teleportation of
// qubit state to another.
36
const std::string src("__qpu__ teleport (qbit qreg) {\n"
Mccaskey, Alex's avatar
Mccaskey, Alex committed
37
38
39
40
41
42
43
44
45
46
47
48
49
	"   cbit creg[2];\n"
	"   // Init qubit 0 to 1\n"
	"   X(qreg[0]);\n"
	"   // Now teleport...\n"
	"   H(qreg[1]);\n"
	"   CNOT(qreg[1],qreg[2]);\n"
	"   CNOT(qreg[0],qreg[1]);\n"
	"   H(qreg[0]);\n"
	"   creg[0] = MeasZ(qreg[0]);\n"
	"   creg[1] = MeasZ(qreg[1]);\n"
	"   if (creg[0] == 1) Z(qreg[2]);\n"
	"   if (creg[1] == 1) X(qreg[2]);\n"
	"}\n");
50
51
52

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

53
54
	xacc::Initialize();

Mccaskey, Alex's avatar
Mccaskey, Alex committed
55
56
	// Create a convenient alias for our simulator...
	using CircuitSimulator = xacc::quantum::FireTensorAccelerator<6>;
57

Mccaskey, Alex's avatar
Mccaskey, Alex committed
58
	// Create a reference to the 6 qubit simulation Accelerator
Mccaskey, Alex's avatar
Mccaskey, Alex committed
59
	auto qpu = std::make_shared<CircuitSimulator>();
60

Mccaskey, Alex's avatar
Mccaskey, Alex committed
61
	// Allocate 3 qubits, give them a unique identifier...
62
63
	auto qubitReg = qpu->createBuffer("qreg", 3);
	using QubitRegisterType = decltype(qubitReg);
64

Mccaskey, Alex's avatar
Mccaskey, Alex committed
65
	// Construct a new XACC Program
66
	xacc::Program quantumProgram(qpu, src);
67

Mccaskey, Alex's avatar
Mccaskey, Alex committed
68
69
	// Build the program using Scaffold comipler
	// and output the Graph Intermediate Representation
70
71
	quantumProgram.build("--compiler scaffold "
			"--writeIR teleport.xir");
72

Mccaskey, Alex's avatar
Mccaskey, Alex committed
73
74
	// Retrieve the created kernel. It takes a 
	// qubit register as input
75
	auto teleport = quantumProgram.getKernel<QubitRegisterType>("teleport");
76

Mccaskey, Alex's avatar
Mccaskey, Alex committed
77
	// Execute the kernel with the qubit register!
78
	teleport(qubitReg);
79

Mccaskey, Alex's avatar
Mccaskey, Alex committed
80
	// Pretty print the resultant state
81
	qubitReg->printBufferState(std::cout);
82

83
84
	xacc::Finalize();

85
86
87
88
89
	return 0;
}