fix other GPU bug

README.md

@@ -96,14 +96,14 @@ This implementation doesn't support data with an oblique angle acquisition yet.
 ```julia
 using QuantitativeSusceptibilityMappingTGV
 
-sz = (20, 20, 20)
-res = [1, 1, 1]
-TE = 0.01
+sz = (20, 20, 20);
+res = [1, 1, 1];
+TE = 0.01;
 
-phase = randn(sz)
-mask = trues(sz)
+phase = randn(sz);
+mask = trues(sz);
 
-chi = qsm_tgv(phase, mask, res; TE)
+chi = qsm_tgv(phase, mask, res; TE);
 ```
 
 ## Settings to reproduce the original version

src/tgv.jl

@@ -165,16 +165,16 @@ end
 function select_device(library::Module)
     if Symbol(library) == :CUDA
         println("Using the GPU via CUDA")
-        return CUDA.CUDAKernels.CUDABackend(), CUDA.cu
+        return library.CUDAKernels.CUDABackend(), library.cu
     elseif Symbol(library) == :AMDGPU
         println("Using the GPU via AMDGPU")
-        return AMDGPU.ROCKernels.ROCBackend(), AMDGPU.ROCArray
+        return library.ROCKernels.ROCBackend(), library.ROCArray
     elseif Symbol(library) == :oneAPI
         println("Using the GPU via oneAPI")
-        return oneAPI.oneAPIKernels.oneAPIBackend(), oneAPI.oneArray
+        return library.oneAPIKernels.oneAPIBackend(), library.oneArray
     elseif Symbol(library) == :Metal
         println("Using the GPU via Metal")
-        return Metal.MetalKernels.MetalBackend(), Metal.MtlArray
+        return library.MetalKernels.MetalBackend(), library.MtlArray
     else
         println("Using $(Threads.nthreads()) CPU cores")
         return CPU(), identity