Minor fix to AMDs device id tests.

FluxML · Aug 20, 2023 · f1ab569 · f1ab569
1 parent b47a6f4
commit f1ab569
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test/ext_amdgpu/get_devices.jl b/test/ext_amdgpu/get_devices.jl
@@ -29,13 +29,13 @@ if AMDGPU.functional() && AMDGPU.functional(:MIOpen)
   for ordinal in 0:(length(AMDGPU.devices()) - 1)
     amd_device = Flux.get_device("AMD", ordinal)
     @test typeof(amd_device.deviceID) <: AMDGPU.HIPDevice
-    @test AMDGPU.device_id(amd_device.deviceID) == ordinal
+    @test AMDGPU.device_id(amd_device.deviceID) == ordinal + 1
 
     dense_model = dense_model |> amd_device
     @test dense_model.weight isa AMDGPU.ROCArray
     @test dense_model.bias isa AMDGPU.ROCArray
-    @test ADMGPU.device_id(AMDGPU.device(dense_model.weight)) == ordinal
-    @test ADMGPU.device_id(AMDGPU.device(dense_model.bias)) == ordinal
+    @test ADMGPU.device_id(AMDGPU.device(dense_model.weight)) == ordinal + 1
+    @test ADMGPU.device_id(AMDGPU.device(dense_model.bias)) == ordinal + 1
     @test isequal(Flux.cpu(dense_model.weight), weight)
     @test isequal(Flux.cpu(dense_model.bias), bias)
   end