Selecting the backend

JustRelax.jl supports three backends: the default CPU backend, and two GPU backends for Nvidia and AMD GPUs. The default CPU backend is selected upon loading JustRelax:

using JustRelax

The GPU backends are implemented as extensions, and can be selected upon loading the appropriate GPU package before loading JustRelax. If running on Nvidia or AMD GPUs, use the CUDA.jl or the AMDGPU.jl package, respectively:

Nvidia GPUs

using CUDA, JustRelax

AMD GPUs

using AMDGPU, JustRelax

Two and three dimensional solvers are implemented in different submodules, which also need to be loaded:

2D module

using JustRelax.JustRelax2D

3D module

using JustRelax.JustRelax3D