try printing versioninfo

MolloiLab · Sep 26, 2024 · 3381c43 · 3381c43
1 parent 9ef144a
commit 3381c43
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/Project.toml b/Project.toml
@@ -5,6 +5,7 @@ version = "0.2.2"
 
 [deps]
 GPUArraysCore = "46192b85-c4d5-4398-a991-12ede77f4527"
+InteractiveUtils = "b77e0a4c-d291-57a0-90e8-8db25a27a240"
 KernelAbstractions = "63c18a36-062a-441e-b654-da1e3ab1ce7c"
 
 [compat]

diff --git a/benchmarks/setup.jl b/benchmarks/setup.jl
@@ -3,18 +3,25 @@ using DistanceTransforms
 using ImageMorphology: distance_transform, feature_transform
 using KernelAbstractions
 using Random
+import InteractiveUtils
 
 const BENCHMARK_GROUP = get(ENV, "BENCHMARK_GROUP", "CPU")
 
+InteractiveUtils.versioninfo()
+
 # Only load the selected backend to avoid unnecessary initializations
 if BENCHMARK_GROUP == "Metal"
     using Metal
+    Metal.versioninfo()
 elseif BENCHMARK_GROUP == "CUDA"
     using CUDA
+    CUDA.versioninfo()
 elseif BENCHMARK_GROUP == "AMDGPU"
     using AMDGPU
+    AMDGPU.versioninfo()
 elseif BENCHMARK_GROUP == "oneAPI"
     using oneAPI
+    oneAPI.versioninfo()
 end
 
 function setup_benchmarks(suite::BenchmarkGroup, backend::String, num_cpu_threads::Int64)