Merge branch 'main' into adm/variational

PTsolvers · Oct 25, 2024 · cdf7783 · cdf7783
2 parents f4a78f1 + a39bb58
commit cdf7783
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 6 deletions.
diff --git a/src/JustRelax_CPU.jl b/src/JustRelax_CPU.jl
@@ -21,7 +21,7 @@ import JustRelax:
 
 import JustPIC._2D: numphases, nphases
 
-@init_parallel_stencil(Threads, Float64, 2)
+__init__() = @init_parallel_stencil(Threads, Float64, 2)
 
 include("common.jl")
 include("stokes/Stokes2D.jl")
@@ -52,7 +52,7 @@ import JustRelax:
 
 import JustPIC._3D: numphases, nphases
 
-@init_parallel_stencil(Threads, Float64, 3)
+__init__() = @init_parallel_stencil(Threads, Float64, 3)
 
 include("common.jl")
 include("stokes/Stokes3D.jl")

diff --git a/src/ext/AMDGPU/2D.jl b/src/ext/AMDGPU/2D.jl
@@ -31,7 +31,7 @@ import JustRelax:
 
 import JustPIC._2D: nphases, numphases
 
-@init_parallel_stencil(AMDGPU, Float64, 2)
+__init__() = @init_parallel_stencil(AMDGPU, Float64, 2)
 
 include("../../common.jl")
 include("../../stokes/Stokes2D.jl")

diff --git a/src/ext/AMDGPU/3D.jl b/src/ext/AMDGPU/3D.jl
@@ -31,7 +31,7 @@ import JustRelax:
 
 import JustPIC._3D: numphases, nphases
 
-@init_parallel_stencil(AMDGPU, Float64, 3)
+__init__() = @init_parallel_stencil(AMDGPU, Float64, 3)
 
 include("../../common.jl")
 include("../../stokes/Stokes3D.jl")

diff --git a/src/ext/CUDA/2D.jl b/src/ext/CUDA/2D.jl
@@ -24,7 +24,7 @@ import JustRelax:
 
 import JustPIC._2D: numphases, nphases
 
-@init_parallel_stencil(CUDA, Float64, 2)
+__init__() = @init_parallel_stencil(CUDA, Float64, 2)
 
 include("../../common.jl")
 include("../../stokes/Stokes2D.jl")

diff --git a/src/ext/CUDA/3D.jl b/src/ext/CUDA/3D.jl
@@ -23,7 +23,7 @@ import JustRelax:
 
 import JustPIC._3D: numphases, nphases
 
-@init_parallel_stencil(CUDA, Float64, 3)
+__init__() = @init_parallel_stencil(CUDA, Float64, 3)
 
 include("../../common.jl")
 include("../../stokes/Stokes3D.jl")