Merge pull request #159 from ali-ramadhan/ali/has-rocm-gpu

Add `has_rocm_gpu` util function

Author: jpsamaroo

Project.toml

@@ -1,7 +1,7 @@
 name = "AMDGPU"
 uuid = "21141c5a-9bdb-4563-92ae-f87d6854732e"
 authors = ["Julian P Samaroo <jpsamaroo@jpsamaroo.me>"]
-version = "0.2.12"
+version = "0.2.13"
 
 [deps]
 AbstractFFTs = "621f4979-c628-5d54-868e-fcf4e3e8185c"

src/AMDGPU.jl

@@ -19,6 +19,7 @@ import Core: LLVMPtr
 export HSAAgent, HSAQueue, HSAExecutable, HSAKernelInstance, HSASignal
 export get_agents, profile, get_first_isa_string, launch!
 export get_default_agent, get_default_queue
+export has_rocm_gpu
 
 export ROCArray, ROCVector, ROCMatrix, ROCVecOrMat
 export roc

src/utils.jl

@@ -27,3 +27,11 @@ function versioninfo(io::IO=stdout)
         end
     end
 end
+
+function has_rocm_gpu()
+    if !AMDGPU.hsa_configured 
+        return false
+    else
+        return length(AMDGPU.get_agents(:gpu)) > 0
+    end
+end

test/hsa/utils.jl

@@ -0,0 +1,10 @@
+@testset "Utils" begin
+    @testset "versioninfo" begin
+        AMDGPU.versioninfo()
+        @test true  # Just test that versioninfo() prints something.
+    end
+
+    @testset "has_rocm_gpu" begin
+        @test AMDGPU.has_rocm_gpu()
+    end
+end

test/runtests.jl

@@ -37,6 +37,7 @@ if AMDGPU.configured
             include("hsa/error.jl")
             include("hsa/agent.jl")
             include("hsa/memory.jl")
+            include("hsa/utils.jl")
         end
         @testset "Codegen" begin
             include("codegen/synchronization.jl")