search

firedrakeproject / firedrake

Firedrake is an automated system for the portable solution of partial differential equations using the finite element method (FEM)
https://firedrakeproject.org
Other
488 stars 157 forks source link

How to enable GPU for Firedrake #2212

Open EstebanFO opened 2 years ago

EstebanFO commented 2 years ago

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 commented 2 years ago

Hi Esteban, Firedrake is not GPU capable yet. We had started working towards it in https://github.com/firedrakeproject/firedrake/pull/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 commented 2 years ago

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.