move module imports triggering extensions before ImplicitGlobalGrid

eth-cscs · Jan 18, 2024 · ead45ca · ead45ca
1 parent 441aacd
commit ead45ca
Show file tree

Hide file tree

Showing 6 changed files with 8 additions and 11 deletions.
diff --git a/test/test_finalize_global_grid.jl b/test/test_finalize_global_grid.jl
@@ -1,7 +1,7 @@
 push!(LOAD_PATH, "../src")
 using Test
+import MPI, CUDA, AMDGPU
 using ImplicitGlobalGrid; GG = ImplicitGlobalGrid
-import MPI
 import ImplicitGlobalGrid: @require
 
 

diff --git a/test/test_gather.jl b/test/test_gather.jl
@@ -1,7 +1,7 @@
 push!(LOAD_PATH, "../src")
 using Test
+import MPI, CUDA, AMDGPU
 using ImplicitGlobalGrid; GG = ImplicitGlobalGrid
-import MPI
 import ImplicitGlobalGrid: @require
 
 

diff --git a/test/test_init_global_grid.jl b/test/test_init_global_grid.jl
@@ -1,7 +1,7 @@
 push!(LOAD_PATH, "../src")
 using Test
+import MPI, CUDA, AMDGPU
 using ImplicitGlobalGrid; GG = ImplicitGlobalGrid
-import MPI
 import ImplicitGlobalGrid: @require
 
 

diff --git a/test/test_select_device.jl b/test/test_select_device.jl
@@ -1,10 +1,9 @@
 # NOTE: All tests of this file can be run with any number of processes.
 push!(LOAD_PATH, "../src")
 using Test
-using ImplicitGlobalGrid; GG = ImplicitGlobalGrid
 import MPI
-using CUDA
-using AMDGPU
+using CUDA, AMDGPU
+using ImplicitGlobalGrid; GG = ImplicitGlobalGrid
 import ImplicitGlobalGrid: @require
 
 test_cuda = CUDA.functional()

diff --git a/test/test_tools.jl b/test/test_tools.jl
@@ -1,7 +1,7 @@
 push!(LOAD_PATH, "../src")
 using Test
+import MPI, CUDA, AMDGPU
 using ImplicitGlobalGrid; GG = ImplicitGlobalGrid
-import MPI
 import ImplicitGlobalGrid: @require
 macro coords(i) :(GG.global_grid().coords[$i]) end
 

diff --git a/test/test_update_halo.jl b/test/test_update_halo.jl
@@ -4,11 +4,9 @@
 
 push!(LOAD_PATH, "../src")
 using Test
-import LoopVectorization
+import MPI, LoopVectorization
+using CUDA, AMDGPU
 using ImplicitGlobalGrid; GG = ImplicitGlobalGrid
-import MPI
-using CUDA
-using AMDGPU
 import ImplicitGlobalGrid: @require, longnameof
 
 test_cuda = CUDA.functional()