drm/amdgpu: fix incorrect amd-staging-drm-next merge

Looks like some code got lost in the 6.1 merge, re-add it.

Not for upstreaming!

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