return 0;
 }
 
-static int cxl_map_regs(struct cxl_dev_state *cxlds, struct cxl_register_map *map)
-{
-       struct device *dev = cxlds->dev;
-       struct pci_dev *pdev = to_pci_dev(dev);
-
-       switch (map->reg_type) {
-       case CXL_REGLOC_RBI_COMPONENT:
-               cxl_map_component_regs(pdev, &cxlds->regs.component, map);
-               dev_dbg(dev, "Mapping component registers...\n");
-               break;
-       case CXL_REGLOC_RBI_MEMDEV:
-               cxl_map_device_regs(pdev, &cxlds->regs.device_regs, map);
-               dev_dbg(dev, "Probing device registers...\n");
-               break;
-       default:
-               break;
-       }
-
-       return 0;
-}
-
 static int cxl_setup_regs(struct pci_dev *pdev, enum cxl_regloc_type type,
                          struct cxl_register_map *map)
 {
        if (rc)
                return rc;
 
-       rc = cxl_map_regs(cxlds, &map);
+       rc = cxl_map_device_regs(pdev, &cxlds->regs.device_regs, &map);
        if (rc)
                return rc;