for (i = 0; i < fb_info->num_fb; ++i)
                hw_params.fb[i] = &fb_info->fb[i];
 
-       switch (adev->asic_type) {
-       case CHIP_YELLOW_CARP:
-               if (dc->ctx->asic_id.hw_internal_rev != YELLOW_CARP_A0) {
-                       hw_params.dpia_supported = true;
+       switch (adev->ip_versions[DCE_HWIP][0]) {
+       case IP_VERSION(3, 1, 3): /* Only for this asic hw internal rev B0 */
+               hw_params.dpia_supported = true;
 #if defined(CONFIG_DRM_AMD_DC_DCN)
-                       hw_params.disable_dpia = dc->debug.dpia_debug.bits.disable_dpia;
+               hw_params.disable_dpia = dc->debug.dpia_debug.bits.disable_dpia;
 #endif
-               }
                break;
        default:
                break;