Fix

Author: michel2323

ext/OceananigansCUDAExt.jl

@@ -1,6 +1,7 @@
 module OceananigansCUDAExt
 
 using Oceananigans
+using InteractiveUtils
 using CUDA, CUDA.CUSPARSE, CUDA.CUFFT
 using KernelAbstractions
 import Oceananigans.Architectures as AC
@@ -42,11 +43,17 @@ function AC.GPU()
     end
 end
 
+function UT.versioninfo_with_gpu(::CUDAGPU)
+    s = sprint(versioninfo)
+    gpu_name = CUDA.CuDevice(0) |> CUDA.name
+    return "CUDA GPU: $gpu_name"
+end
+
 
 Base.summary(::CUDAGPU) = "CUDAGPU"
 
 AC.architecture(::CuArray) = CUDAGPU()
-AC.architecture(::CuSparseMatrixCSC) = AC.GPU()
+AC.architecture(::CuSparseMatrixCSC) = CUDAGPU()
 AC.array_type(::AC.GPU{CUDABackend}) = CuArray
 
 AC.on_architecture(::AC.CPU, a::CuArray) = Array(a)

src/Utils/versioninfo.jl

@@ -3,12 +3,7 @@ using InteractiveUtils
 using Oceananigans.Architectures
 
 function versioninfo_with_gpu()
-    s = sprint(versioninfo)
-    if isdefined(Main, :CUDABackend)
-        gpu_name = CUDA.CuDevice(0) |> CUDA.name
-        s = s * "  GPU: $gpu_name\n"
-    end
-    return s
+    return versioninfo_with_gpu(GPU())
 end
 
 function oceananigans_versioninfo()