Merge pull request #278 from FourierFlows/ncc/gpu-functionality-info

navidcy · web-flow · commit 20136a748836 · 2021-05-09T08:24:57.000+10:00
Display info about loading GPU-related methods
diff --git a/src/FourierFlows.jl b/src/FourierFlows.jl
@@ -113,6 +113,7 @@ include("diffusion.jl")
 
 # CUDA functionality
 if CUDA.has_cuda()
+  @info "GPU functionality for FourierFlows is loaded"
   include("CuFourierFlows.jl")
 end
 
