}
 }
 
+static void rvu_setup_rvum_blk_revid(struct rvu *rvu)
+{
+       rvu_write64(rvu, BLKADDR_RVUM,
+                   RVU_PRIV_BLOCK_TYPEX_REV(BLKTYPE_RVUM),
+                   RVU_BLK_RVUM_REVID);
+}
+
+static void rvu_clear_rvum_blk_revid(struct rvu *rvu)
+{
+       rvu_write64(rvu, BLKADDR_RVUM,
+                   RVU_PRIV_BLOCK_TYPEX_REV(BLKTYPE_RVUM), 0x00);
+}
+
 int rvu_lf_reset(struct rvu *rvu, struct rvu_block *block, int lf)
 {
        int err;
        if (err)
                goto err_flr;
 
+       rvu_setup_rvum_blk_revid(rvu);
+
        /* Enable AF's VFs (if any) */
        err = rvu_enable_sriov(rvu);
        if (err)
        rvu_fwdata_exit(rvu);
        rvu_reset_all_blocks(rvu);
        rvu_free_hw_resources(rvu);
+       rvu_clear_rvum_blk_revid(rvu);
 err_release_regions:
        pci_release_regions(pdev);
 err_disable_device:
        rvu_disable_sriov(rvu);
        rvu_reset_all_blocks(rvu);
        rvu_free_hw_resources(rvu);
-
+       rvu_clear_rvum_blk_revid(rvu);
        pci_release_regions(pdev);
        pci_disable_device(pdev);
        pci_set_drvdata(pdev, NULL);