Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Update plxpr tracing pipeline to happen inside an EvaluationContext #1539

Open
wants to merge 5 commits into
base: main
Choose a base branch
from

Conversation

mudit2812
Copy link
Contributor

@mudit2812 mudit2812 commented Feb 27, 2025

Context:
When tracing the program with experimental_capture=True, PennyLane is not able to detect if qjit is being used or not, but it should be.

Description of the Change:

  • Update QJIT.capture to use the EvaluationContext context manager with CLASSICAL_COMPILATION evaluation mode when tracing using plxpr.

Benefits:

Possible Drawbacks:

Related GitHub Issues:

[sc-85399]

Copy link
Contributor

Hello. You may have forgotten to update the changelog!
Please edit doc/releases/changelog-dev.md on your branch with:

  • A one-to-two sentence description of the change. You may include a small working example for new features.
  • A link back to this PR.
  • Your name (or GitHub username) in the contributors section.

@mudit2812 mudit2812 marked this pull request as ready for review March 4, 2025 00:18
@mudit2812 mudit2812 requested a review from dime10 March 4, 2025 00:18
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants