Merge pull request #253 from JuliaGPU/jps/device-id-methods

device: Add device_id method

Author: jpsamaroo

src/highlevel.jl

@@ -37,6 +37,15 @@ Sets the default device to `AMDGPU.devices(kind)[idx]`. See
 default_device_id!(idx::Integer, kind::Symbol=:gpu) =
     default_device!(devices(kind)[idx])
 
+"""
+    device_id(device::ROCDevice, kind::Symbol=:gpu) -> Int
+
+Returns the numerical device ID for `device`. See [`default_device_id`](@ref)
+for details on the numbering semantics.
+"""
+device_id(device::ROCDevice, kind::Symbol=:gpu) =
+    something(findfirst(dev->dev === device, devices(kind)))
+
 """
     device_type(device::ROCDevice) -> Symbol
 

test/hsa/device.jl

@@ -1,4 +1,13 @@
-@testset "Agent" begin
+@testset "Devices" begin
+    @testset "Device IDs" begin
+        for kind in (:cpu, :gpu)
+            devices = AMDGPU.devices()
+            for (idx,device) in enumerate(devices)
+                @test AMDGPU.device_id(device) == idx
+            end
+        end
+    end
+
     @testset "Default selection" begin
         agent = AMDGPU.default_device()
         @test agent !== nothing