diff --git a/frontend/catalyst/python_interface/doc/unified_compiler_cookbook.rst b/frontend/catalyst/python_interface/doc/unified_compiler_cookbook.rst index 25e2d23c57..837ccc136f 100644 --- a/frontend/catalyst/python_interface/doc/unified_compiler_cookbook.rst +++ b/frontend/catalyst/python_interface/doc/unified_compiler_cookbook.rst @@ -664,10 +664,12 @@ currently rely on JAX’s API to lower to MLIR. This has the special effect of lowering to a specific dialect called StableHLO, which is used to represent all arithmetic operations present in the program. -Once lowered to MLIR, if the original ``qjit`` decorator specified the -xDSL pass plugin, we pass control over to the xDSL layer, which applies -all transforms that were requested by the user. We can request the use -of the xDSL plugin like so: +Once lowered to MLIR, if any xDSL registered passes are detected, we pass the control over to +the xDSL layer, which automatically detects and applies all xDSL transforms that were requested +by the user. + +However, if you want to manually trigger the xDSL layer without using any xDSL registered passes, +you can do so by specifying the ``pass_plugins`` parameter: .. code-block:: python @@ -1003,9 +1005,7 @@ currently accessible as qml.capture.enable() dev = qml.device("lightning.qubit", wires=1) - @qml.qjit( - pass_plugins=[catalyst.passes.xdsl_plugin.getXDSLPluginAbsolutePath()] - ) + @qml.qjit @my_pass @qml.qnode(dev) def circuit(x): @@ -1295,8 +1295,6 @@ will explain what is going on. .. code-block:: python - from catalyst.passes.xdsl_plugin import getXDSLPluginAbsolutePath - def test_h_to_x_pass_integration(run_filecheck_qjit): """Test that Hadamard gets converted into PauliX.""" # The original program simply applies a Hadamard to a circuit @@ -1305,7 +1303,7 @@ will explain what is going on. # `compiler_transform`. To make sure that the xDSL API works # correctly, program capture must be enabled. # qml.capture.enable() - @qml.qjit(pass_plugins=[getXDSLPluginAbsolutePath]) + @qml.qjit @h_to_x_pass def circuit(): # CHECK: [[q0:%.+]] = "test.op"() : () -> !quantum.bit diff --git a/frontend/test/pytest/python_interface/conftest.py b/frontend/test/pytest/python_interface/conftest.py index 7548811594..909315ffe5 100644 --- a/frontend/test/pytest/python_interface/conftest.py +++ b/frontend/test/pytest/python_interface/conftest.py @@ -184,7 +184,7 @@ def test_qjit(self, run_filecheck_qjit): # Test that the merge_rotations_pass works as expected when used with `qjit` dev = qml.device("lightning.qubit", wires=2) - @qml.qjit(target="mlir", pass_plugins=[getXDSLPluginAbsolutePath()]) + @qml.qjit(target="mlir") @merge_rotations_pass @qml.qnode(dev) def circuit(x: float, y: float): diff --git a/frontend/test/pytest/python_interface/inspection/test_draw_unified_compiler.py b/frontend/test/pytest/python_interface/inspection/test_draw_unified_compiler.py index 3f22feeb3e..e7280dd6c0 100644 --- a/frontend/test/pytest/python_interface/inspection/test_draw_unified_compiler.py +++ b/frontend/test/pytest/python_interface/inspection/test_draw_unified_compiler.py @@ -23,7 +23,6 @@ import jax import pennylane as qml -from catalyst.passes.xdsl_plugin import getXDSLPluginAbsolutePath from catalyst.python_interface.inspection import draw from catalyst.python_interface.transforms import ( iterative_cancel_inverses_pass, @@ -91,9 +90,7 @@ def test_multiple_levels_xdsl(self, transforms_circuit, level, qjit, expected): ) if qjit: - transforms_circuit = qml.qjit(pass_plugins=[getXDSLPluginAbsolutePath()])( - transforms_circuit - ) + transforms_circuit = qml.qjit(transforms_circuit) assert draw(transforms_circuit, level=level)() == expected @@ -127,9 +124,7 @@ def test_multiple_levels_catalyst(self, transforms_circuit, level, qjit, expecte ) if qjit: - transforms_circuit = qml.qjit(pass_plugins=[getXDSLPluginAbsolutePath()])( - transforms_circuit - ) + transforms_circuit = qml.qjit(transforms_circuit) assert draw(transforms_circuit, level=level)() == expected @@ -162,9 +157,7 @@ def test_multiple_levels_xdsl_catalyst(self, transforms_circuit, level, qjit, ex qml.transforms.merge_rotations(transforms_circuit) ) if qjit: - transforms_circuit = qml.qjit(pass_plugins=[getXDSLPluginAbsolutePath()])( - transforms_circuit - ) + transforms_circuit = qml.qjit(transforms_circuit) assert draw(transforms_circuit, level=level)() == expected @@ -208,9 +201,7 @@ def test_no_passes(self, transforms_circuit, level, qjit, expected): """Test that if no passes are applied, the circuit is still visualized.""" if qjit: - transforms_circuit = qml.qjit(pass_plugins=[getXDSLPluginAbsolutePath()])( - transforms_circuit - ) + transforms_circuit = qml.qjit(transforms_circuit) assert draw(transforms_circuit, level=level)() == expected @@ -487,7 +478,7 @@ def circ(arg): def adjoint_op_not_implemented(self): """Test that NotImplementedError is raised when AdjointOp is used.""" - @qml.qjit(pass_plugins=[getXDSLPluginAbsolutePath()]) + @qml.qjit @qml.qnode(qml.device("lightning.qubit", wires=1)) def circuit(): qml.adjoint(qml.QubitUnitary)(jax.numpy.array([[0, 1], [1, 0]]), wires=[0]) @@ -499,7 +490,7 @@ def circuit(): def test_cond_not_implemented(self): """Test that NotImplementedError is raised when cond is used.""" - @qml.qjit(pass_plugins=[getXDSLPluginAbsolutePath()]) + @qml.qjit @qml.qnode(qml.device("lightning.qubit", wires=2)) def circuit(): m0 = qml.measure(0, reset=False, postselect=0) @@ -512,7 +503,7 @@ def circuit(): def test_for_loop_not_implemented(self): """Test that NotImplementedError is raised when for loop is used.""" - @qml.qjit(pass_plugins=[getXDSLPluginAbsolutePath()], autograph=True) + @qml.qjit(autograph=True) @qml.qnode(qml.device("lightning.qubit", wires=1)) def circuit(): for _ in range(3): @@ -525,7 +516,7 @@ def circuit(): def test_while_loop_not_implemented(self): """Test that NotImplementedError is raised when while loop is used.""" - @qml.qjit(pass_plugins=[getXDSLPluginAbsolutePath()], autograph=True) + @qml.qjit(autograph=True) @qml.qnode(qml.device("lightning.qubit", wires=1)) def circuit(): i = 0 diff --git a/frontend/test/pytest/python_interface/inspection/test_mlir_graph.py b/frontend/test/pytest/python_interface/inspection/test_mlir_graph.py index 380fe6a9e5..eb4c669320 100644 --- a/frontend/test/pytest/python_interface/inspection/test_mlir_graph.py +++ b/frontend/test/pytest/python_interface/inspection/test_mlir_graph.py @@ -24,7 +24,6 @@ import pennylane as qml -from catalyst.passes.xdsl_plugin import getXDSLPluginAbsolutePath from catalyst.python_interface.inspection import generate_mlir_graph from catalyst.python_interface.transforms import ( iterative_cancel_inverses_pass, @@ -68,7 +67,7 @@ def _(): return qml.state() if qjit: - _ = qml.qjit(pass_plugins=[getXDSLPluginAbsolutePath()])(_) + _ = qml.qjit(_) generate_mlir_graph(_)() assert collect_files(tmp_path) == {"QNode_level_0_no_transforms.svg"} @@ -88,7 +87,7 @@ def _(): return qml.state() if qjit: - _ = qml.qjit(pass_plugins=[getXDSLPluginAbsolutePath()])(_) + _ = qml.qjit(_) generate_mlir_graph(_)() assert_files( @@ -113,7 +112,7 @@ def _(x, y, w1, w2): return qml.state() if qjit: - _ = qml.qjit(pass_plugins=[getXDSLPluginAbsolutePath()])(_) + _ = qml.qjit(_) generate_mlir_graph(_)(0.1, 0.2, 0, 1) assert_files( @@ -138,7 +137,7 @@ def _(x, y, w1, w2): return qml.state() if qjit: - _ = qml.qjit(pass_plugins=[getXDSLPluginAbsolutePath()])(_) + _ = qml.qjit(_) generate_mlir_graph(_)(0.1, 0.2, 0, 1) assert_files( @@ -164,7 +163,7 @@ def _(x, y, w1, w2): return qml.state() if qjit: - _ = qml.qjit(pass_plugins=[getXDSLPluginAbsolutePath()])(_) + _ = qml.qjit(_) generate_mlir_graph(_)(0.1, 0.2, 0, 1) assert_files( diff --git a/frontend/test/pytest/python_interface/test_unified_compiler.py b/frontend/test/pytest/python_interface/test_unified_compiler.py index fe5c3f9f60..12d19a71be 100644 --- a/frontend/test/pytest/python_interface/test_unified_compiler.py +++ b/frontend/test/pytest/python_interface/test_unified_compiler.py @@ -301,7 +301,7 @@ def test_integration_catalyst_xdsl_pass_with_capture(self, capsys): assert capture_enabled() - @qjit(pass_plugins=[getXDSLPluginAbsolutePath()]) + @qjit @hello_world_pass @qml.qnode(qml.device("lightning.qubit", wires=2)) def f(x): @@ -319,7 +319,7 @@ def test_integration_catalyst_xdsl_pass_no_capture(self, capsys): assert not capture_enabled() - @qjit(pass_plugins=[getXDSLPluginAbsolutePath()]) + @qjit @apply_pass("hello-world") @qml.qnode(qml.device("lightning.qubit", wires=2)) def f(x): @@ -338,7 +338,7 @@ def test_integration_catalyst_mixed_passes_with_capture(self, capsys): assert capture_enabled() - @qjit(pass_plugins=[getXDSLPluginAbsolutePath()]) + @qjit @hello_world_pass @qml.transforms.cancel_inverses @qml.qnode(qml.device("lightning.qubit", wires=2)) @@ -359,7 +359,7 @@ def test_integration_catalyst_mixed_passes_no_capture(self, capsys): assert not capture_enabled() - @qjit(pass_plugins=[getXDSLPluginAbsolutePath()]) + @qjit @apply_pass("hello-world") @catalyst_cancel_inverses @qml.qnode(qml.device("lightning.qubit", wires=2)) @@ -495,7 +495,7 @@ def print_between_passes(_, module, __, pass_level=0): print("=== Between Pass ===") print(module) - @qml.qjit(pass_plugins=[getXDSLPluginAbsolutePath()]) + @qml.qjit @iterative_cancel_inverses_pass @merge_rotations_pass @qml.qnode(qml.device("null.qubit", wires=2)) diff --git a/frontend/test/pytest/python_interface/transforms/mbqc/test_xdsl_outline_state_evolution.py b/frontend/test/pytest/python_interface/transforms/mbqc/test_xdsl_outline_state_evolution.py index 72fd95d9df..8300f663e2 100644 --- a/frontend/test/pytest/python_interface/transforms/mbqc/test_xdsl_outline_state_evolution.py +++ b/frontend/test/pytest/python_interface/transforms/mbqc/test_xdsl_outline_state_evolution.py @@ -22,7 +22,6 @@ from pennylane.ftqc import RotXZX from catalyst.ftqc import mbqc_pipeline -from catalyst.passes.xdsl_plugin import getXDSLPluginAbsolutePath from catalyst.python_interface.transforms import ( OutlineStateEvolutionPass, convert_to_mbqc_formalism_pass, @@ -147,10 +146,7 @@ def test_outline_state_evolution_no_error(self): """Test outline_state_evolution_pass does not raise error for circuit with classical operations only.""" - @qml.qjit( - target="mlir", - pass_plugins=[getXDSLPluginAbsolutePath()], - ) + @qml.qjit(target="mlir") @outline_state_evolution_pass def circuit(x, y): return x * y + 5 @@ -165,10 +161,7 @@ def test_outline_state_evolution_no_terminal_op_error(self): # the program is captured. dev = qml.device("null.qubit", wires=10) - @qml.qjit( - target="mlir", - pass_plugins=[getXDSLPluginAbsolutePath()], - ) + @qml.qjit(target="mlir") @outline_state_evolution_pass @qml.qnode(dev) def circuit(): @@ -184,10 +177,7 @@ def test_outline_state_evolution_pass_only(self, run_filecheck_qjit): """Test the outline_state_evolution_pass only.""" dev = qml.device("lightning.qubit", wires=1000) - @qml.qjit( - target="mlir", - pass_plugins=[getXDSLPluginAbsolutePath()], - ) + @qml.qjit(target="mlir") @outline_state_evolution_pass @qml.set_shots(1000) @qml.qnode(dev) @@ -223,11 +213,7 @@ def test_outline_state_evolution_pass_with_convert_to_mbqc_formalism(self, run_f on lightning.qubit.""" dev = qml.device("lightning.qubit", wires=1000) - @qml.qjit( - target="mlir", - pass_plugins=[getXDSLPluginAbsolutePath()], - pipelines=mbqc_pipeline(), - ) + @qml.qjit(target="mlir", pipelines=mbqc_pipeline()) @decompose_graph_state_pass @convert_to_mbqc_formalism_pass @outline_state_evolution_pass @@ -273,11 +259,7 @@ def test_outline_state_evolution_pass_with_mbqc_pipeline(self, run_filecheck_qji null.qubit.""" dev = qml.device("null.qubit", wires=1000) - @qml.qjit( - target="mlir", - pass_plugins=[getXDSLPluginAbsolutePath()], - pipelines=mbqc_pipeline(), - ) + @qml.qjit(target="mlir", pipelines=mbqc_pipeline()) @decompose_graph_state_pass @convert_to_mbqc_formalism_pass @measurements_from_samples_pass @@ -323,11 +305,7 @@ def test_outline_state_evolution_pass_with_mbqc_pipeline_run_on_nullqubit(self): transform pipeline can be executed on null.qubit.""" dev = qml.device("null.qubit", wires=1000) - @qml.qjit( - target="mlir", - pass_plugins=[getXDSLPluginAbsolutePath()], - pipelines=mbqc_pipeline(), - ) + @qml.qjit(target="mlir", pipelines=mbqc_pipeline()) @decompose_graph_state_pass @convert_to_mbqc_formalism_pass @measurements_from_samples_pass @@ -367,10 +345,7 @@ def while_fn(i): i = i + 1 return i - @qml.qjit( - target="mlir", - pass_plugins=[getXDSLPluginAbsolutePath()], - ) + @qml.qjit(target="mlir") @outline_state_evolution_pass @qml.qnode(dev) def circuit(): diff --git a/frontend/test/pytest/python_interface/transforms/quantum/test_xdsl_cancel_inverses.py b/frontend/test/pytest/python_interface/transforms/quantum/test_xdsl_cancel_inverses.py index d358e1ce20..8281a775e6 100644 --- a/frontend/test/pytest/python_interface/transforms/quantum/test_xdsl_cancel_inverses.py +++ b/frontend/test/pytest/python_interface/transforms/quantum/test_xdsl_cancel_inverses.py @@ -21,7 +21,6 @@ import pennylane as qml -from catalyst.passes.xdsl_plugin import getXDSLPluginAbsolutePath from catalyst.python_interface.transforms import ( IterativeCancelInversesPass, iterative_cancel_inverses_pass, @@ -194,7 +193,7 @@ def test_qjit(self, run_filecheck_qjit): """Test that the IterativeCancelInversesPass works correctly with qjit.""" dev = qml.device("lightning.qubit", wires=2) - @qml.qjit(target="mlir", pass_plugins=[getXDSLPluginAbsolutePath()]) + @qml.qjit(target="mlir") @iterative_cancel_inverses_pass @qml.qnode(dev) def circuit(): @@ -212,7 +211,7 @@ def test_qjit_no_cancellation(self, run_filecheck_qjit): there are no operations that can be cancelled.""" dev = qml.device("lightning.qubit", wires=2) - @qml.qjit(target="mlir", pass_plugins=[getXDSLPluginAbsolutePath()]) + @qml.qjit(target="mlir") @iterative_cancel_inverses_pass @qml.qnode(dev) def circuit(): diff --git a/frontend/test/pytest/python_interface/transforms/quantum/test_xdsl_combine_global_phases.py b/frontend/test/pytest/python_interface/transforms/quantum/test_xdsl_combine_global_phases.py index 8e45bbf04a..b393cbf279 100644 --- a/frontend/test/pytest/python_interface/transforms/quantum/test_xdsl_combine_global_phases.py +++ b/frontend/test/pytest/python_interface/transforms/quantum/test_xdsl_combine_global_phases.py @@ -20,7 +20,6 @@ import pennylane as qml -from catalyst.passes.xdsl_plugin import getXDSLPluginAbsolutePath from catalyst.python_interface.transforms import ( CombineGlobalPhasesPass, combine_global_phases_pass, @@ -224,7 +223,7 @@ def test_qjit(self, run_filecheck_qjit): """Test that the CombineGlobalPhasesPass works correctly with qjit.""" dev = qml.device("lightning.qubit", wires=2) - @qml.qjit(target="mlir", pass_plugins=[getXDSLPluginAbsolutePath()]) + @qml.qjit(target="mlir") @combine_global_phases_pass @qml.qnode(dev) def circuit(x: float, y: float): diff --git a/frontend/test/pytest/python_interface/transforms/quantum/test_xdsl_diagonalize_measurements.py b/frontend/test/pytest/python_interface/transforms/quantum/test_xdsl_diagonalize_measurements.py index 79a60e7857..3a325c838b 100644 --- a/frontend/test/pytest/python_interface/transforms/quantum/test_xdsl_diagonalize_measurements.py +++ b/frontend/test/pytest/python_interface/transforms/quantum/test_xdsl_diagonalize_measurements.py @@ -24,7 +24,7 @@ import pennylane as qml -from catalyst.passes import apply_pass, xdsl_plugin +from catalyst.passes import apply_pass from catalyst.python_interface.transforms import ( DiagonalizeFinalMeasurementsPass, diagonalize_final_measurements_pass, @@ -343,7 +343,6 @@ def circuit_ref(phi): ), "Sanity check failed, is expected_res correct?" circuit_compiled = qml.qjit( diagonalize_final_measurements_pass(circuit_ref), - pass_plugins=[xdsl_plugin.getXDSLPluginAbsolutePath()], ) assert np.allclose(expected_res(angle), circuit_compiled(angle)) @@ -376,7 +375,6 @@ def expected_res(x, y): ), "Sanity check failed, is expected_res correct?" circuit_compiled = qml.qjit( diagonalize_final_measurements_pass(circuit_ref), - pass_plugins=[xdsl_plugin.getXDSLPluginAbsolutePath()], ) assert np.allclose(expected_res(phi, theta), circuit_compiled(phi, theta)) @@ -406,7 +404,6 @@ def expected_res(x, y): circuit_compiled = qml.qjit( diagonalize_final_measurements_pass(circuit_ref), - pass_plugins=[xdsl_plugin.getXDSLPluginAbsolutePath()], ) assert np.allclose(expected_res(phi, theta), circuit_compiled(phi, theta)) @@ -418,7 +415,7 @@ def test_overlapping_observables_raises_error(self): dev = qml.device("lightning.qubit", wires=2) - @qml.qjit(pass_plugins=[xdsl_plugin.getXDSLPluginAbsolutePath()]) + @qml.qjit @diagonalize_final_measurements_pass @qml.qnode(dev) def circuit(x): @@ -437,7 +434,7 @@ def test_non_commuting_observables_raise_error(self): non-commuting observables.""" dev = qml.device("lightning.qubit", wires=1) - @qml.qjit(pass_plugins=[xdsl_plugin.getXDSLPluginAbsolutePath()]) + @qml.qjit @diagonalize_final_measurements_pass @qml.qnode(dev) def circuit(x): diff --git a/frontend/test/pytest/python_interface/transforms/quantum/test_xdsl_measurements_from_samples.py b/frontend/test/pytest/python_interface/transforms/quantum/test_xdsl_measurements_from_samples.py index a0f6caa518..bce8b4ad02 100644 --- a/frontend/test/pytest/python_interface/transforms/quantum/test_xdsl_measurements_from_samples.py +++ b/frontend/test/pytest/python_interface/transforms/quantum/test_xdsl_measurements_from_samples.py @@ -26,7 +26,6 @@ import pennylane as qml -from catalyst.passes import xdsl_plugin from catalyst.python_interface.transforms import ( MeasurementsFromSamplesPass, measurements_from_samples_pass, @@ -552,7 +551,6 @@ def circuit_ref(): assert expected_res == circuit_ref(), "Sanity check failed, is expected_res correct?" circuit_compiled = qml.qjit( measurements_from_samples_pass(circuit_ref), - pass_plugins=[xdsl_plugin.getXDSLPluginAbsolutePath()], ) assert expected_res == circuit_compiled() @@ -584,7 +582,6 @@ def circuit_ref(): ), "Sanity check failed, is expected_res correct?" circuit_compiled = qml.qjit( measurements_from_samples_pass(circuit_ref), - pass_plugins=[xdsl_plugin.getXDSLPluginAbsolutePath()], ) assert np.array_equal(expected_res, circuit_compiled()) @@ -622,7 +619,6 @@ def circuit_ref(): circuit_compiled = qml.qjit( measurements_from_samples_pass(circuit_ref), - pass_plugins=[xdsl_plugin.getXDSLPluginAbsolutePath()], ) assert np.array_equal(expected_res, _counts_catalyst_to_pl(*circuit_compiled())) @@ -652,7 +648,6 @@ def circuit_ref(): circuit_compiled = qml.qjit( measurements_from_samples_pass(circuit_ref), - pass_plugins=[xdsl_plugin.getXDSLPluginAbsolutePath()], ) expected_res = expected_res_base * np.ones(shape=(shots, 1), dtype=int) @@ -694,7 +689,6 @@ def circuit_ref(): assert expected_res == circuit_ref(), "Sanity check failed, is expected_res correct?" circuit_compiled = qml.qjit( measurements_from_samples_pass(circuit_ref), - pass_plugins=[xdsl_plugin.getXDSLPluginAbsolutePath()], ) assert expected_res == circuit_compiled() @@ -735,10 +729,7 @@ def circuit_ref(): assert expected_res == circuit_ref(), "Sanity check failed, is expected_res correct?" - circuit_compiled = qml.qjit( - measurements_from_samples_pass(circuit_ref), - pass_plugins=[xdsl_plugin.getXDSLPluginAbsolutePath()], - ) + circuit_compiled = qml.qjit(measurements_from_samples_pass(circuit_ref)) assert expected_res == circuit_compiled() @@ -769,10 +760,7 @@ def circuit_ref(): assert np.array_equal( expected_res, circuit_ref() ), "Sanity check failed, is expected_res correct?" - circuit_compiled = qml.qjit( - measurements_from_samples_pass(circuit_ref), - pass_plugins=[xdsl_plugin.getXDSLPluginAbsolutePath()], - ) + circuit_compiled = qml.qjit(measurements_from_samples_pass(circuit_ref)) assert np.array_equal(expected_res, circuit_compiled()) @@ -803,10 +791,7 @@ def circuit_ref(): assert np.array_equal( expected_res, circuit_ref() ), "Sanity check failed, is expected_res correct?" - circuit_compiled = qml.qjit( - measurements_from_samples_pass(circuit_ref), - pass_plugins=[xdsl_plugin.getXDSLPluginAbsolutePath()], - ) + circuit_compiled = qml.qjit(measurements_from_samples_pass(circuit_ref)) assert np.array_equal(expected_res, circuit_compiled()) @@ -819,7 +804,7 @@ def test_exec_expval_dynamic_shots(self): This use case is not currently supported. """ - @qml.qjit(pass_plugins=[xdsl_plugin.getXDSLPluginAbsolutePath()]) + @qml.qjit def workload(shots): dev = qml.device("lightning.qubit", wires=1) @@ -837,7 +822,7 @@ def test_qjit_filecheck(self, run_filecheck_qjit): """Test that the measurements_from_samples_pass works correctly with qjit.""" dev = qml.device("lightning.qubit", wires=2) - @qml.qjit(target="mlir", pass_plugins=[xdsl_plugin.getXDSLPluginAbsolutePath()]) + @qml.qjit(target="mlir") @measurements_from_samples_pass @qml.qnode(dev, shots=25) def circuit(): @@ -858,7 +843,7 @@ def test_integrate_with_decompose(self): with the decompose pass.""" dev = qml.device("null.qubit", wires=4) - @qml.qjit(target="mlir", pass_plugins=[xdsl_plugin.getXDSLPluginAbsolutePath()]) + @qml.qjit(target="mlir") @measurements_from_samples_pass @partial( qml.transforms.decompose, diff --git a/frontend/test/pytest/python_interface/transforms/quantum/test_xdsl_merge_rotations.py b/frontend/test/pytest/python_interface/transforms/quantum/test_xdsl_merge_rotations.py index e24b3ca487..7445ad9c0f 100644 --- a/frontend/test/pytest/python_interface/transforms/quantum/test_xdsl_merge_rotations.py +++ b/frontend/test/pytest/python_interface/transforms/quantum/test_xdsl_merge_rotations.py @@ -20,7 +20,6 @@ import pennylane as qml -from catalyst.passes.xdsl_plugin import getXDSLPluginAbsolutePath from catalyst.python_interface.transforms import MergeRotationsPass, merge_rotations_pass @@ -227,7 +226,7 @@ def test_qjit(self, run_filecheck_qjit): """Test that the MergeRotationsPass works correctly with qjit.""" dev = qml.device("lightning.qubit", wires=1) - @qml.qjit(target="mlir", pass_plugins=[getXDSLPluginAbsolutePath()]) + @qml.qjit(target="mlir") @merge_rotations_pass @qml.qnode(dev) def circuit(x: float, y: float): diff --git a/frontend/test/pytest/python_interface/transforms/quantum/test_xdsl_split_non_commuting.py b/frontend/test/pytest/python_interface/transforms/quantum/test_xdsl_split_non_commuting.py index 4a24b76e39..59b365f136 100644 --- a/frontend/test/pytest/python_interface/transforms/quantum/test_xdsl_split_non_commuting.py +++ b/frontend/test/pytest/python_interface/transforms/quantum/test_xdsl_split_non_commuting.py @@ -20,7 +20,6 @@ import pennylane as qml -from catalyst.passes.xdsl_plugin import getXDSLPluginAbsolutePath from catalyst.python_interface.transforms import ( SplitNonCommutingPass, split_non_commuting_pass, @@ -229,10 +228,7 @@ def _while_for(i): i = i + 1 return i - @qml.qjit( - target="mlir", - pass_plugins=[getXDSLPluginAbsolutePath()], - ) + @qml.qjit(target="mlir") @split_non_commuting_pass @qml.set_shots(10) @qml.qnode(dev) @@ -276,10 +272,7 @@ def while_fn(i): i = i + 1 return i - @qml.qjit( - target="mlir", - pass_plugins=[getXDSLPluginAbsolutePath()], - ) + @qml.qjit(target="mlir") @split_non_commuting_pass @qml.qnode(dev) def circuit():