diff --git a/.dep-versions b/.dep-versions index d90cdf9883..9e23d13975 100644 --- a/.dep-versions +++ b/.dep-versions @@ -10,7 +10,7 @@ enzyme=v0.0.203 # For a custom PL version, update the package version here and at # 'doc/requirements.txt' -pennylane=0.45.0-dev1 +pennylane=0.45.0-dev10 # For a custom LQ/LK version, update the package version here and at # 'doc/requirements.txt' diff --git a/doc/requirements.txt b/doc/requirements.txt index a9804c9cff..58e8cef206 100644 --- a/doc/requirements.txt +++ b/doc/requirements.txt @@ -34,4 +34,4 @@ lxml_html_clean --extra-index-url https://test.pypi.org/simple/ pennylane-lightning-kokkos==0.45.0-dev8 pennylane-lightning==0.45.0-dev8 -pennylane==0.45.0-dev1 +pennylane==0.45.0-dev10 diff --git a/frontend/catalyst/jax_extras/tracing.py b/frontend/catalyst/jax_extras/tracing.py index 969b7edb4c..d658229952 100644 --- a/frontend/catalyst/jax_extras/tracing.py +++ b/frontend/catalyst/jax_extras/tracing.py @@ -1004,7 +1004,7 @@ def uses_transform(qnode, transform_name): bool: True if `transform_name` is detected (and is only one if only_one=True), False otherwise """ - transform_program = getattr(qnode, "transform_program", []) - transform_funcs = [transform_container.transform for transform_container in transform_program] + compile_pipeline = getattr(qnode, "compile_pipeline", []) + transform_funcs = [bound_transform.transform for bound_transform in compile_pipeline] return any(transform_name in func.__name__ for func in transform_funcs) diff --git a/frontend/catalyst/jax_tracer.py b/frontend/catalyst/jax_tracer.py index 8c1f235233..45cb5490a0 100644 --- a/frontend/catalyst/jax_tracer.py +++ b/frontend/catalyst/jax_tracer.py @@ -1527,7 +1527,7 @@ def is_leaf(obj): else: device_program = qml.CompilePipeline() - qnode_program = qnode.transform_program if qnode else qml.CompilePipeline() + qnode_program = qnode.compile_pipeline if qnode else qml.CompilePipeline() tapes, post_processing, tracing_mode = apply_transforms( qnode_program, diff --git a/frontend/catalyst/qfunc.py b/frontend/catalyst/qfunc.py index 11fa16e3ac..6ad5459233 100644 --- a/frontend/catalyst/qfunc.py +++ b/frontend/catalyst/qfunc.py @@ -251,14 +251,14 @@ def __call__(self, *args, **kwargs): assert isinstance(self, qml.QNode) - new_transform_program, new_pipeline = _extract_passes(self.transform_program) + new_compile_pipeline, new_pipeline = _extract_passes(self.compile_pipeline) # Update the qnode with peephole pipeline old_pipeline = kwargs.pop("pass_pipeline", None) processed_old_pipeline = tuple(dictionary_to_list_of_passes(old_pipeline)) pass_pipeline = processed_old_pipeline + new_pipeline new_qnode = copy(self) # pylint: disable=attribute-defined-outside-init, protected-access - new_qnode._transform_program = new_transform_program + new_qnode._compile_pipeline = new_compile_pipeline # Mid-circuit measurement configuration: one_shot_results = configure_mcm_and_try_one_shot(new_qnode, args, kwargs, pass_pipeline)