search

slimgroup / Devito4PyTorch

Integrating Devito operators into PyTorch
MIT License
12 stars 1 forks source link

Expose GPU kernels for Pytorch #2

Open loliverhennigh opened 8 months ago

loliverhennigh commented 8 months ago

This is an interesting application of Devito. Was wondering if you know of a way to expose GPU computing in devito. I see all the examples go pytorch GPU -> numpy cpu -> Devito -> numpy cpu -> pytorch GPU.

mloubout commented 8 months ago

DeVito supports GPU through openacc offloading (or openmp offloading) but will need to go through numpy in-between still as there is currently no api to directly use GPU tensors