SciML / OperatorLearning.jl

No need to train, he's a smooth operator
https://operatorlearning.sciml.ai/dev
MIT License
43 stars 8 forks source link

Pre-allocate for efficiency ? #40

Closed dynamic-queries closed 2 years ago

dynamic-queries commented 2 years ago

In The Forward pass of the Fourier Layer, temporary artifacts - "linear", "𝔉", "i𝔉" are allocated for every call.

There should be a speedup, if this is preallocated in the form of a "Cache" struct. Is this right ?

pzimbrod commented 2 years ago

Yup, that's right. That discussion already came up earlier in #32 and #14. Therefore I'd close this for now as we don't need to track the same thing twice.

Unfortunately due to lack of time and resources, not much has happened on that front. If you're willing, you're welcome to work on that of course!