Support inverse gates

pennylane_lightning/lightning_qubit.py

@@ -136,7 +136,8 @@ def apply_lightning(self, state, operations):
         op_names = [o.name for o in operations]
         op_wires = [self.wires.indices(o.wires) for o in operations]
         op_param = [o.parameters for o in operations]
+        op_inverse = [o.inverse for o in operations]
 
         state_vector = np.ravel(state)
-        apply(state_vector, op_names, op_wires, op_param, self.num_wires)
+        apply(state_vector, op_names, op_wires, op_param, op_inverse, self.num_wires)
         return np.reshape(state_vector, state.shape)

pennylane_lightning/src/Apply.cpp

@@ -53,6 +53,7 @@ void Pennylane::constructAndApplyOperation(
     const string& opLabel,
     const vector<unsigned int>& opWires,
     const vector<double>& opParams,
+    bool inverse,
     const unsigned int qubits
 ) {
     unique_ptr<AbstractGate> gate = constructGate(opLabel, opParams);
@@ -64,14 +65,15 @@ void Pennylane::constructAndApplyOperation(
     vector<unsigned int> externalWires = getIndicesAfterExclusion(opWires, qubits);
     vector<size_t> externalIndices = generateBitPatterns(externalWires, qubits);
 
-    gate->applyKernel(state, internalIndices, externalIndices);
+    gate->applyKernel(state, internalIndices, externalIndices, inverse);
 }
 
 void Pennylane::apply(
     StateVector& state,
     const vector<string>& ops,
     const vector<vector<unsigned int>>& wires,
     const vector<vector<double>>& params,
+    const std::vector<bool>& inverse,
     const unsigned int qubits
 ) { 
     if (qubits <= 0)
@@ -86,7 +88,7 @@ void Pennylane::apply(
         throw std::invalid_argument("Invalid arguments: number of operations, wires, and parameters must all be equal");
 
     for (int i = 0; i < numOperations; i++) {
-        constructAndApplyOperation(state, ops[i], wires[i], params[i], qubits);
+        constructAndApplyOperation(state, ops[i], wires[i], params[i], inverse[i], qubits);
     }
 
 }

pennylane_lightning/src/Apply.hpp

@@ -62,13 +62,15 @@ namespace Pennylane {
      * @param opLabel unique string corresponding to a gate type
      * @param opWires index of qubits on which the gate acts
      * @param opParams defines the gate parameterisation (may be zero-length for some gates)
+     * @param inverse boolean indicating whether to apply the gate or its inverse
      * @param qubits number of qubits
      */
     void constructAndApplyOperation(
         StateVector& state,
         const std::string& opLabel,
         const std::vector<unsigned int>& opWires,
         const std::vector<double>& opParams,
+        bool inverse,
         const unsigned int qubits
     );
 
@@ -79,12 +81,15 @@ namespace Pennylane {
      * @param ops list of unique string names corresponding to gate types, in the order they should be applied
      * @param wires list of wires on which each gate acts
      * @param params list of parameters that defines the gate parameterisation
+     * @param inverse list of bools indicating whether a given gate or its inverse should be applied
+     * @param qubits total number of qubits
      */
     void apply(
         StateVector& state,
         const std::vector<std::string>& ops,
         const std::vector<std::vector<unsigned int>>& wires,
         const std::vector<std::vector<double>>& params,
+        const std::vector<bool>& inverse,
         const unsigned int qubits
     );
 

pennylane_lightning/src/Bindings.cpp

@@ -39,10 +39,11 @@ void apply(
     vector<string> ops,
     vector<vector<unsigned int>> wires,
     vector<vector<double>> params,
+    vector<bool> inverse,
     const unsigned int qubits
 ) {
     StateVector state = create(&stateNumpyArray);
-    Pennylane::apply(state, ops, wires, params, qubits);
+    Pennylane::apply(state, ops, wires, params, inverse, qubits);
 }
 
 PYBIND11_MODULE(lightning_qubit_ops, m)