drm/amdgpu: use the num_rings variable for checking vce rings

Difference families may have different numbers of rings. Use
the variable rather than a hardcoded number.

Reviewed-by: Christian König <christian.koenig@amd.com>
Signed-off-by: Alex Deucher <alexander.deucher@amd.com>

1 files changed, 2 insertions, 2 deletions

diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu_cs.c b/drivers/gpu/drm/amd/amdgpu/amdgpu_cs.c
index 8d019ca7d9c7..cf2e8c4e9b8b 100644
--- a/drivers/gpu/drm/amd/amdgpu/amdgpu_cs.c
+++ b/drivers/gpu/drm/amd/amdgpu/amdgpu_cs.c
@@ -75,10 +75,10 @@ int amdgpu_cs_get_ring(struct amdgpu_device *adev, u32 ip_type,
                 *out_ring = &adev->uvd.ring;
                 break;
         case AMDGPU_HW_IP_VCE:
-                if (ring < 2){
+                if (ring < adev->vce.num_rings){
                         *out_ring = &adev->vce.ring[ring];
                 } else {
-                        DRM_ERROR("only two VCE rings are supported\n");
+                        DRM_ERROR("only %d VCE rings are supported\n", adev->vce.num_rings);
                         return -EINVAL;
                 }
                 break;