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 GPUs, use the CUDA.jl package:

using CUDA, JustRelax

and if running on AMD GPUs, use the AMDGPU.jl package:

using AMDGPU, JustRelax

Two and three dimensional solvers are implemented in different submodules, which also need to be loaded. To access the two-dimensional module:

using JustRelax.JustRelax2D

and for the three-dimensional module:

using JustRelax.JustRelax3D