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

Solver Compilation #275

Open
dario-coscia opened this issue Apr 3, 2024 · 1 comment
Open

Solver Compilation #275

dario-coscia opened this issue Apr 3, 2024 · 1 comment
Labels
enhancement New feature or request v0.2 implementation in v0.2

Comments

@dario-coscia
Copy link
Collaborator

dario-coscia commented Apr 3, 2024

Is your feature request related to a problem? Please describe.
From PyTorch 2.0 there is the possibility to compile the code, resulting in faster training and inference time, see here for reference. Also PyTorch Lightning supports the torch compilation as shown here.

It would be nice to implement a compilation flag to all PINA solvers to enable compilation for the model and equations (if present in the problem).

Describe the solution you'd like
This could be implemented by using the two abstract class SolverInterface, Equation. The amount of code change should be minimal, since PyTorch Lightning already deals itself to compile all internal functions (training_step, forward, ...).

@dario-coscia dario-coscia added enhancement New feature or request v0.2 implementation in v0.2 labels Apr 3, 2024
@dario-coscia
Copy link
Collaborator Author

Let's implement this feature in the new release

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request v0.2 implementation in v0.2
Projects
None yet
Development

No branches or pull requests

1 participant