search

ar4 / deepwave

Wave propagation modules for PyTorch.
MIT License
216 stars 47 forks source link

Add GPU propagators #2

Closed ar4 closed 6 years ago

ar4 commented 6 years ago

Finite difference wave propagation involves a type of computation that GPUs (graphics processing units) are suitable for. Creating GPU versions of the propagators could therefore enable speed-ups for users who have appropriate GPUs. One of the benefits of basing Deepwave on PyTorch is the ease of using GPUs. There is a PyTorch tutorial that explains how to create a CUDA extension.

ar4 commented 6 years ago

Closed by e5d7e2cbaeadaa917a445c11ed6390bd490c5413