Skip to content

Improve pjit#299

Merged
mschwoer merged 12 commits into
mainfrom
improve_pjit
Jan 28, 2026
Merged

Improve pjit#299
mschwoer merged 12 commits into
mainfrom
improve_pjit

Conversation

@mschwoer

@mschwoer mschwoer commented Jun 27, 2025

Copy link
Copy Markdown
Contributor
  • add the option to deactivate pjit (only functional change: 730572a)
  • refactor for readability

@mschwoer mschwoer requested review from GeorgWa, Copilot and swillems June 27, 2025 14:03

Copilot AI left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

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

Pull Request Overview

This PR improves the pjit functionality by adding an option to deactivate numba njit, refactoring for readability, and increasing the progress bar sleep time.

  • Added a conditional decorator to optionally apply numba.njit via a new parameter.
  • Refactored the pjit wrapper and progress callback mechanism.
  • Adjusted sleep timing in the progress callback to slow down updates.

Comment thread alphatims/utils.py Outdated
Comment thread alphatims/utils.py Outdated
@mschwoer mschwoer requested a review from jalew188 June 27, 2025 14:09
@mschwoer mschwoer merged commit b18e155 into main Jan 28, 2026
6 checks passed
@mschwoer mschwoer deleted the improve_pjit branch February 16, 2026 10:54
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