Add missing imports

PTsolvers · Nov 9, 2023 · bdd03d8 · bdd03d8
1 parent f463269
commit bdd03d8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/ext/AMDGPUExt/AMDGPUExt.jl b/ext/AMDGPUExt/AMDGPUExt.jl
@@ -2,7 +2,7 @@ module AMDGPUExt
 
 using AMDGPU, AMDGPU.ROCKernels
 
-import FastIce.Architecture: heuristic_groupsize, set_device!
+import FastIce.Architecture: heuristic_groupsize, set_device!, get_device
 
 set_device!(dev::HIPDevice) = AMDGPU.device!(dev)
 

diff --git a/ext/CUDAExt/CUDAExt.jl b/ext/CUDAExt/CUDAExt.jl
@@ -2,7 +2,7 @@ module CUDAExt
 
 using CUDA, CUDA.CUDAKernels
 
-import FastIce.Architecture: heuristic_groupsize, set_device!
+import FastIce.Architecture: heuristic_groupsize, set_device!, get_device
 
 set_device!(dev::CuDevice) = CUDA.device!(dev)