search

ITMO-NSS-team / torch_DE_solver

Combines power of torch, numerical methods to conquer and solve ALL {O,P}DEs
https://torch-de-solver.readthedocs.io
BSD 3-Clause "New" or "Revised" License
59 stars 9 forks source link

Lazy computation mode #25

Open SuperSashka opened 1 year ago

SuperSashka commented 1 year ago

We need to make sure that we can use LazyTensor mode https://arxiv.org/pdf/2102.13267.pdf and implement it for faster computation.

nikiniki1 commented 1 year ago

Some computation speedup probably can be achieved with pytorch 2.0 using the compiler.

SuperSashka commented 1 year ago

At the time being it works only in Linux or WSL. We'll add functionality as soon as possible. The test version is in https://github.com/ITMO-NSS-team/torch_DE_solver/tree/compiler branch.