JIT rewrite endpoint for testgen more suitable for JIT compiled functions #1055

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.

Sign up for GitHub

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

Jump to bottom

Merged

aseembits93 merged 30 commits into main from instrument-jit

Jan 16, 2026

Contributor

aseembits93 commented Jan 14, 2026

No description provided.

Codeflash Bot and others added 11 commits

December 30, 2025 09:44


          need to test now

2e0c578

wip

fed0595

wip

e38df62

wip

17042a2


          bugfix

f945fef


          mlx is problematic

d555e8b


          failsafe

7bf6681


          comparator fix

f1e4735


          new gpu instrumentation

81d0599


          bug fix

6e19826


          improve comparator

d7a69ab

CLAassistant commented Jan 14, 2026

Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you sign our Contributor License Agreement before we can accept your contribution.

Codeflash Bot seems not to be a GitHub user. You need a GitHub account to be able to sign the CLA. If you have already a GitHub account, please add the email address used for this commit to your account.
_{You have signed the CLA already but the status is still pending? Let us recheck it.}

aseembits93 added 2 commits

January 14, 2026 12:18


          Merge branch 'main' into instrument-jit

c0f2ada


          reverting tests

722f252

aseembits93 changed the title ~~Instrument jit~~ JIT rewrite endpoint for testgen more suitable for JIT compiled functions

aseembits93 commented

View reviewed changes

codeflash/api/aiservice.py Outdated Show resolved Hide resolved

aseembits93 commented

View reviewed changes

codeflash/api/aiservice.py Outdated Show resolved Hide resolved

aseembits93 commented

View reviewed changes

codeflash/api/aiservice.py Outdated Show resolved Hide resolved

aseembits93 commented

View reviewed changes

codeflash/api/aiservice.py Outdated Show resolved Hide resolved

aseembits93 commented

View reviewed changes

codeflash/api/aiservice.py Outdated Show resolved Hide resolved

aseembits93 commented

View reviewed changes

codeflash/api/aiservice.py Outdated Show resolved Hide resolved

aseembits93 added 4 commits

January 14, 2026 13:19


          Cleaning up

c597702


          jit rewrite type


          Merge branch 'main' into instrument-jit

f993993


          cleaning up

d0c6643

codeflash-ai bot mentioned this pull request

⚡️ Speed up function is_numerical_code by 70% in PR #1055 (instrument-jit) #1057

Closed

Contributor

codeflash-ai bot commented Jan 15, 2026

⚡️ Codeflash found optimizations for this PR

📄 70% (0.70x) speedup for `is_numerical_code` in `codeflash/code_utils/code_extractor.py`

⏱️ Runtime : 41.3 milliseconds → 24.3 milliseconds (best of 63 runs)

A dependent PR with the suggested changes has been created. Please review:

⚡️ Speed up function is_numerical_code by 70% in PR #1055 (instrument-jit) #1057

If you approve, it will be merged into this PR (branch instrument-jit).

aseembits93 added 3 commits

January 14, 2026 19:00


          expected payload is correct

256332f


          almost ready

6c83c34


          Merge branch 'main' into instrument-jit

b33a53f

aseembits93 added 6 commits

January 15, 2026 14:43


          add prompt conditionally

ecce649


          some jit heavy functions

184b853


          local aiservice temporary

3a0e418


          almost ready

4d28c17


          tensorflow, jax, pytorch now working on mac metal

ee6872c


          ready to review

9d4fa82

aseembits93 commented

View reviewed changes

codeflash/api/aiservice.py Outdated Show resolved Hide resolved

aseembits93 commented

View reviewed changes

codeflash/api/aiservice.py Outdated Show resolved Hide resolved

aseembits93 commented

View reviewed changes

codeflash/api/aiservice.py Outdated Show resolved Hide resolved

aseembits93 commented

View reviewed changes

codeflash/optimization/function_optimizer.py Outdated Show resolved Hide resolved


          Apply suggestions from code review

01a8ebf

aseembits93 commented

View reviewed changes

codeflash/optimization/function_optimizer.py Outdated Show resolved Hide resolved

aseembits93 added 2 commits

January 16, 2026 12:15


          sentry capture

b41d387


          final formatting

074ad24

HeshamHM28 approved these changes

View reviewed changes

Contributor

HeshamHM28 left a comment

LGTM

aseembits93 enabled auto-merge

January 16, 2026 20:25

aseembits93 merged commit c935efe into main

23 of 25 checks passed

This was referenced Jan 16, 2026

⚡️ Speed up function _lis_outer_body_tf by 1,249% #1082

Closed

⚡️ Speed up function _lis_inner_body_tf by 913% #1081

Closed

aseembits93 deleted the instrument-jit branch

January 16, 2026 20:39

This was referenced Jan 16, 2026

⚡️ Speed up function _leapfrog_step_body_tf by 589% #1080

Closed

⚡️ Speed up function tridiagonal_solve_tf by 282% #1079

Closed

⚡️ Speed up function _tridiagonal_forward_body_tf by 72% #1078

Closed

⚡️ Speed up function _lis_outer_body_jax by 6,122% #1077

Closed

⚡️ Speed up function _lis_inner_body_jax by 2,485% #1076

Closed

⚡️ Speed up function leapfrog_integration_jax by 19,337% #1075

Closed

⚡️ Speed up function tridiagonal_solve_jax by 1,995% #1074

Closed

⚡️ Speed up function _tridiagonal_forward_step_jax by 8% #1073

Closed

⚡️ Speed up function longest_increasing_subsequence_length by 40,778% #1072

Closed

⚡️ Speed up function leapfrog_integration by 32,859% #1071

Closed

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet