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

Simplify and speed up InterruptionEventLoop #361

Merged
merged 33 commits into from
Nov 14, 2023

Conversation

eb8680
Copy link
Contributor

@eb8680 eb8680 commented Oct 25, 2023

Blocked by #360

Part of addressing performance and design issues from #354 #360

@eb8680 eb8680 added enhancement New feature or request refactor status:WIP Work-in-progress not yet ready for review blocked module:dynamical labels Oct 25, 2023
@eb8680 eb8680 added this to the ASKEM November 2023 hackathon milestone Oct 25, 2023
Base automatically changed from eb-shallow-handler to master October 25, 2023 13:39
@SamWitty SamWitty removed the blocked label Oct 25, 2023
@eb8680 eb8680 removed the status:WIP Work-in-progress not yet ready for review label Nov 13, 2023
@eb8680 eb8680 added the status:awaiting review Awaiting response from reviewer label Nov 13, 2023
@SamWitty
Copy link
Collaborator

@eb8680 , this was tagged "status: awaiting review", but it looks there have been updates since that tag was assigned. Are those updates now complete?

@SamWitty SamWitty self-requested a review November 14, 2023 19:24
Copy link
Collaborator

@SamWitty SamWitty left a comment

Choose a reason for hiding this comment

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

Discussed this in a pair-session. Looks good!

@SamWitty SamWitty merged commit 0df0f1d into master Nov 14, 2023
@SamWitty SamWitty deleted the eb-dynamical-shallow-interruptions branch November 14, 2023 19:25
rfl-urbaniak pushed a commit that referenced this pull request Nov 29, 2023
* Add shallow handler

* tests

* almost works

* nit

* type nits

* lint

* nit

* fix test and move base interruption to internals

* fix remaining failing test

* simplify torchdiffeq event fn

* heapq

* heapq works

* move prioritized to utils

* move prioritized to utils

* inf

* refactoring simulate_to_interruption

* consolidate simulate_to_interruption

* lint

* move method

* make all internal ops into pyro ops

* simplify internal

* removing get_solver

* error message

* mark simulate final

* remove vestigial get_solver

* pop solver

* format

* force kwargs update

* add fast path for static interruption

* add assertion to rule out edge case
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request module:dynamical refactor status:awaiting review Awaiting response from reviewer
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants