Skip to content

Binomial checkpointing#2485

Open
Pangoraw wants to merge 31 commits into
mainfrom
pb/binomial
Open

Binomial checkpointing#2485
Pangoraw wants to merge 31 commits into
mainfrom
pb/binomial

Conversation

@Pangoraw
Copy link
Copy Markdown
Collaborator

@Pangoraw Pangoraw commented May 15, 2026

aka AD on a tight budget.

  • test use of iv in reverse (for i in 1:10 x .= x .+ i)
  • non-constant iter counts (require caching and popping)
  • non-deterministic tests (fail only in ci)

Comment thread src/enzyme_ad/jax/Dialect/EnzymeXLAOps.td Outdated
Comment thread src/enzyme_ad/jax/Implementations/StableHLOAutoDiffOpInterfaceImpl.cpp Outdated
@codecov
Copy link
Copy Markdown

codecov Bot commented May 15, 2026

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 25.67%. Comparing base (5db2b68) to head (0e2e111).
⚠️ Report is 4 commits behind head on main.

Additional details and impacted files
@@            Coverage Diff             @@
##             main    #2485      +/-   ##
==========================================
+ Coverage   25.59%   25.67%   +0.08%     
==========================================
  Files         221      221              
  Lines       44707    44796      +89     
==========================================
+ Hits        11443    11502      +59     
- Misses      33264    33294      +30     

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

🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

Copy link
Copy Markdown
Member

@wsmoses wsmoses left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

didn't investigate super deeply, but at a high level all looks good to me

@wsmoses
Copy link
Copy Markdown
Member

wsmoses commented May 15, 2026

please fix the tests before merge though

@wsmoses
Copy link
Copy Markdown
Member

wsmoses commented May 15, 2026

and yeah we shuold support the non constant iter counts, iv tests, etc

@Pangoraw
Copy link
Copy Markdown
Collaborator Author

Pangoraw commented Jun 4, 2026

cc @avik-pal if you can review the WhileLoopInfo changes

Copy link
Copy Markdown
Collaborator

@avik-pal avik-pal left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Checked on the WhileLoopInfo changes

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.

3 participants