Add Preferences

Project.toml

@@ -3,6 +3,9 @@ uuid = "0979c8fe-16a4-4796-9b82-89a9f10403ea"
 authors = ["pedrovalerolara <valerolarap@ornl.gov>", "williamfgc <williamfgc@yahoo.com>"]
 version = "0.0.1"
 
+[deps]
+Preferences = "21216c6a-2e73-6563-6e65-726566657250"
+
 [weakdeps]
 AMDGPU = "21141c5a-9bdb-4563-92ae-f87d6854732e"
 CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
@@ -14,6 +17,7 @@ JACCCUDA = ["CUDA"]
 JACCONEAPI = ["oneAPI"]
 
 [compat]
+Preferences = "1.4.0"
 julia = "1.9.0"
 
 [extras]

ext/JACCAMDGPU/JACCAMDGPU.jl

@@ -300,7 +300,10 @@ function reduce_kernel_amdgpu_MN((M, N), red, ret)
 end
 
 function __init__()
-  const JACC.default_backend = ROCBackend()
+  if JACC.JACCPreferences.backend == "amdgpu"
+    const JACC.default_backend = ROCBackend()
+    @info "Set default backend to $(JACC.default_backend)"
+  end
 end
 
 end # module JACCAMDGPU

ext/JACCCUDA/JACCCUDA.jl

@@ -294,7 +294,10 @@ function reduce_kernel_cuda_MN((M, N), red, ret)
 end
 
 function __init__()
-  const JACC.default_backend = CUDABackend()
+  if JACC.JACCPreferences.backend == "cuda"
+    const JACC.default_backend = CUDABackend()
+    @info "Set default backend to $(JACC.default_backend)"
+  end
 end
 
 end # module JACCCUDA

ext/JACCONEAPI/JACCONEAPI.jl

@@ -294,7 +294,10 @@ function reduce_kernel_oneapi_MN((M, N), red, ret)
 end
 
 function __init__()
-  const JACC.default_backend = oneAPIBackend()
+  if JACC.JACCPreferences.backend == "oneapi"
+    const JACC.default_backend = oneAPIBackend()
+    @info "Set default backend to $(JACC.default_backend)"
+  end
 end
 
 end # module JACCONEAPI

src/JACC.jl

@@ -1,6 +1,7 @@
 module JACC
 
 # module to set back end preferences
+include("JACCPreferences.jl")
 include("helper.jl")
 
 export parallel_for, parallel_reduce, ThreadsBackend, print_default_backend
@@ -69,7 +70,10 @@ function parallel_reduce(::ThreadsBackend, (M, N)::Tuple{I,I}, f::F, x...) where
 end
 
 function __init__()
-    const JACC.default_backend = ThreadsBackend()
+  if JACCPreferences.backend == "threads"
+      const JACC.default_backend = ThreadsBackend()
+      @info "Set default backend to $(JACC.default_backend)"
+  end
 end
 
 function print_default_backend()

src/JACCPreferences.jl

@@ -0,0 +1,22 @@
+
+module JACCPreferences
+
+using Preferences
+
+# taken from https://github.com/JuliaPackaging/Preferences.jl
+function set_backend(new_backend::String)
+
+    new_backend_lc = lowercase(new_backend)
+    if !(new_backend_lc in ("threads", "cuda", "amdgpu", "oneapi"))
+        throw(ArgumentError("Invalid backend: \"$(new_backend)\""))
+    end
+
+    # Set it in our runtime values, as well as saving it to disk
+    @set_preferences!("backend" => new_backend_lc)
+    @info("New backend set; restart your Julia session for this change to take effect!")
+end
+
+const backend = @load_preference("backend", "threads")
+
+
+end # module JACCPreferences

test/tests_threads_perf.jl

@@ -35,4 +35,4 @@ using Test
     for i in 1:ntimes
         @time JACC.parallel_for(ThreadsBackend(), N, axpy, alpha, x_JACC, y_JACC)
     end
-end
+end