How to enable GPU for Firedrake

[EstebanFO]

Dear Firedrakers,

I'm currently running a thermal analysis using Firedrake and everything has been smoothly with MPI. Now I want to use my GPU to speed up the solution process. I was able to configure PETSc with CUDA enabled and PETSc is actually using my GPU when I run the test examples for PETSc.

However, I don't see Firedrake using my GPU when compiling my traditional *.py for the thermal analysis.

How can I call/configure Firedrake or my Python script in order to allow Firedrake to call PETSc with GPU capabilities enabled? Is there any documentation I can check regarding this topic?

Thank you for your help!

[kaushikcfd]

Hi Esteban,
Firedrake is not GPU capable yet. We had started working towards it in #1605. I was slightly occupied with some other things in the past couple of months. However, I will have some time from next week and will pick it up. Hopefully that should be ready in a couple of weeks.

[EstebanFO]

Dear @kaushikcfd,

I was wondering if there are any news regarding GPU support on Firedrake. I've been following #1605 and I see there are some things still left to be solved.

But do you know if I can now install Firedrake with GPU capabilities and run my thermal analysis regardless of some limitations.