void pci_disable_ptm(struct pci_dev *dev)
 {
-       int ptm;
+       u16 ptm = dev->ptm_cap;
        u16 ctrl;
 
-       if (!pci_is_pcie(dev))
-               return;
-
-       ptm = pci_find_ext_capability(dev, PCI_EXT_CAP_ID_PTM);
        if (!ptm)
                return;
 
 
 void pci_save_ptm_state(struct pci_dev *dev)
 {
-       int ptm;
+       u16 ptm = dev->ptm_cap;
        struct pci_cap_saved_state *save_state;
        u16 *cap;
 
-       if (!pci_is_pcie(dev))
-               return;
-
-       ptm = pci_find_ext_capability(dev, PCI_EXT_CAP_ID_PTM);
        if (!ptm)
                return;
 
 
 void pci_restore_ptm_state(struct pci_dev *dev)
 {
+       u16 ptm = dev->ptm_cap;
        struct pci_cap_saved_state *save_state;
-       int ptm;
        u16 *cap;
 
-       if (!pci_is_pcie(dev))
+       if (!ptm)
                return;
 
        save_state = pci_find_saved_ext_cap(dev, PCI_EXT_CAP_ID_PTM);
-       ptm = pci_find_ext_capability(dev, PCI_EXT_CAP_ID_PTM);
-       if (!save_state || !ptm)
+       if (!save_state)
                return;
 
        cap = (u16 *)&save_state->cap.data[0];
 
 void pci_ptm_init(struct pci_dev *dev)
 {
-       int pos;
+       u16 ptm;
        u32 cap, ctrl;
        u8 local_clock;
        struct pci_dev *ups;
                return;
        }
 
-       pos = pci_find_ext_capability(dev, PCI_EXT_CAP_ID_PTM);
-       if (!pos)
+       ptm = pci_find_ext_capability(dev, PCI_EXT_CAP_ID_PTM);
+       if (!ptm)
                return;
 
+       dev->ptm_cap = ptm;
        pci_add_ext_cap_save_buffer(dev, PCI_EXT_CAP_ID_PTM, sizeof(u16));
 
-       pci_read_config_dword(dev, pos + PCI_PTM_CAP, &cap);
+       pci_read_config_dword(dev, ptm + PCI_PTM_CAP, &cap);
        local_clock = (cap & PCI_PTM_GRANULARITY_MASK) >> 8;
 
        /*
        }
 
        ctrl |= dev->ptm_granularity << 8;
-       pci_write_config_dword(dev, pos + PCI_PTM_CTRL, ctrl);
+       pci_write_config_dword(dev, ptm + PCI_PTM_CTRL, ctrl);
        dev->ptm_enabled = 1;
 
        pci_ptm_info(dev);
 
 int pci_enable_ptm(struct pci_dev *dev, u8 *granularity)
 {
-       int pos;
+       u16 ptm;
        u32 cap, ctrl;
        struct pci_dev *ups;
 
        if (!pci_is_pcie(dev))
                return -EINVAL;
 
-       pos = pci_find_ext_capability(dev, PCI_EXT_CAP_ID_PTM);
-       if (!pos)
+       ptm = pci_find_ext_capability(dev, PCI_EXT_CAP_ID_PTM);
+       if (!ptm)
                return -EINVAL;
 
-       pci_read_config_dword(dev, pos + PCI_PTM_CAP, &cap);
+       dev->ptm_cap = ptm;
+       pci_read_config_dword(dev, ptm + PCI_PTM_CAP, &cap);
        if (!(cap & PCI_PTM_CAP_REQ))
                return -EINVAL;
 
 
        ctrl = PCI_PTM_CTRL_ENABLE;
        ctrl |= dev->ptm_granularity << 8;
-       pci_write_config_dword(dev, pos + PCI_PTM_CTRL, ctrl);
+       pci_write_config_dword(dev, ptm + PCI_PTM_CTRL, ctrl);
        dev->ptm_enabled = 1;
 
        pci_ptm_info(dev);