qml.compiler.active_compiler

active_compiler()

Check which compiler is activated inside a qjit() evaluation context.

This helper function may be used during implementation to allow differing logic for transformations or operations that are just-in-time compiled, versus those that are not.

Returns

Name of the active compiler inside a qjit() evaluation context. If there is no active compiler, None will be returned.

Return type

Optional[str]

Example

This method can be used to execute logical branches that are conditioned on whether hybrid compilation with a specific compiler is occurring.

dev = qml.device("lightning.qubit", wires=2)

@qml.qnode(dev)
def circuit(phi, theta):
    if qml.compiler.active_compiler() == "catalyst":
        qml.RX(phi, wires=0)
    qml.CNOT(wires=[0, 1])
    qml.PhaseShift(theta, wires=0)
    return qml.expval(qml.Z(0))

>>> circuit(np.pi, np.pi / 2)
1.0
>>> qml.qjit(circuit)(np.pi, np.pi / 2)
-1.0

code/api/pennylane.compiler.active_compiler

 

Download Python script

 

Download Notebook

 

View on GitHub