fix AMDGPU device selection test

eth-cscs · Jan 22, 2024 · 65e3e49 · 65e3e49
1 parent 040facc
commit 65e3e49
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/test_select_device.jl b/test/test_select_device.jl
@@ -33,13 +33,13 @@ nprocs = MPI.Comm_size(MPI.COMM_WORLD); # NOTE: these tests can run with any num
  @testset "\"AMDGPU\"" begin
  me, = init_global_grid(3, 4, 5; quiet=true, init_MPI=false, device_type="AMDGPU");
  gpu_id = select_device();
- @test gpu_id < length(AMDGPU.devices())
+ @test gpu_id <= length(AMDGPU.devices())
  finalize_global_grid(finalize_MPI=false);
  end;
  @testset "\"auto\"" begin
  me, = init_global_grid(3, 4, 5; quiet=true, init_MPI=false, device_type="auto");
  gpu_id = select_device();
- @test gpu_id < length(AMDGPU.devices())
+ @test gpu_id <= length(AMDGPU.devices())
  finalize_global_grid(finalize_MPI=false);
  end;
  end