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

[Frontend] Adds fwd and bwd to pure_callbacks #743

Conversation

erick-xanadu
Copy link
Contributor

@erick-xanadu erick-xanadu commented May 15, 2024

Context: Changes in the frontend to support fwd and bwd in pure_callbacks

Description of the Change: Class that will contain fwd and bwd along with the minimal sufficient support with MLIR.

Related GitHub Issues: Depends on #742, #735 (The first commit are these two prs squashed, which is what we will see when they are merged. When these are merged, I will just drop and rebase).

(This is marked as work in progress to prioritize review on #735)

[sc-60499]

@erick-xanadu erick-xanadu changed the title Adds fwd and bwd to pure_callbacks [Frontend] Adds fwd and bwd to pure_callbacks May 15, 2024
@erick-xanadu erick-xanadu marked this pull request as draft May 24, 2024 19:25
@erick-xanadu erick-xanadu force-pushed the eochoa/2024-05-14/callbacks-frontend-ii branch from 1e9bcaa to c1c150a Compare May 24, 2024 19:46
@erick-xanadu erick-xanadu changed the base branch from main to eochoa/2024-05-09/specialize-callbacks-part-two May 24, 2024 19:46
Copy link

codecov bot commented May 24, 2024

Codecov Report

Attention: Patch coverage is 71.01449% with 20 lines in your changes are missing coverage. Please review.

Project coverage is 97.87%. Comparing base (e1a1cc4) to head (5c21ebc).

Files Patch % Lines
frontend/catalyst/jax_primitives.py 35.71% 16 Missing and 2 partials ⚠️
frontend/catalyst/api_extensions/callbacks.py 95.12% 1 Missing and 1 partial ⚠️
Additional details and impacted files
@@                                 Coverage Diff                                 @@
##           eochoa/2024-05-09/specialize-callbacks-part-two     #743      +/-   ##
===================================================================================
- Coverage                                            98.07%   97.87%   -0.20%     
===================================================================================
  Files                                                   70       70              
  Lines                                                 9603     9659      +56     
  Branches                                               757      764       +7     
===================================================================================
+ Hits                                                  9418     9454      +36     
- Misses                                                 151      168      +17     
- Partials                                                34       37       +3     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

@erick-xanadu
Copy link
Contributor Author

Closed in favour of #788

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.

None yet

1 participant