GPU operators #163
GPU operators #163
Conversation
src/discretization/operator.jl
Outdated
| element_cache = setup_element_cache(protocol, element_qr, ip, sdh) | ||
| push!(eles_caches, element_cache) | ||
| end | ||
| return dh.subdofhandlers |> cu, eles_caches |> cu |
There was a problem hiding this comment.
Note that this will be super slow, as we need to do this possibly at each time step.
I already thought about whether it might make sense (and how) to pass the device type around to give more precise control over this funny stuff here.
There was a problem hiding this comment.
I was thinking about the same issue because yes, you are absolutely right this is gonna be super slow. The problem though is that I am trying to change right now the gpudofhandler and gpusubdofhandler so I needed to commit everything in case things blow up, didn't actually intend to push tho 😂 but working after Holidays is like forgetting everthing 😢 .
There was a problem hiding this comment.
No worries. You can also just start with copy pasting the existing linear operator and changing the internals to your liking to figure out a better API. :)
Just initial rough ideas for the design of GPU linear operators 🧑🎄