#include <linux/netdevice.h>
 #include <linux/interrupt.h>
+#include <linux/pci.h>
 #include "thunder_bgx.h"
 
 /* PCI device IDs */
        struct bgx_link_status  link_status;
 };
 
+#define NIC_NODE_ID_MASK       0x03
+#define NIC_NODE_ID_SHIFT      44
+
+static inline int nic_get_node_id(struct pci_dev *pdev)
+{
+       u64 addr = pci_resource_start(pdev, PCI_CFG_REG_BAR_NUM);
+       return ((addr >> NIC_NODE_ID_SHIFT) & NIC_NODE_ID_MASK);
+}
+
 int nicvf_set_real_num_queues(struct net_device *netdev,
                              int tx_queues, int rx_queues);
 int nicvf_open(struct net_device *netdev);
 
 struct nicpf {
        struct pci_dev          *pdev;
        u8                      rev_id;
-#define NIC_NODE_ID_MASK       0x300000000000
-#define NIC_NODE_ID(x)         ((x & NODE_ID_MASK) >> 44)
        u8                      node;
        unsigned int            flags;
        u8                      num_vf_en;      /* No of VF enabled */
 
        pci_read_config_byte(pdev, PCI_REVISION_ID, &nic->rev_id);
 
-       nic->node = NIC_NODE_ID(pci_resource_start(pdev, PCI_CFG_REG_BAR_NUM));
+       nic->node = nic_get_node_id(pdev);
 
        nic_set_lmac_vf_mapping(nic);
 
 
                goto err_release_regions;
        }
        bgx->bgx_id = (pci_resource_start(pdev, PCI_CFG_REG_BAR_NUM) >> 24) & 1;
-       bgx->bgx_id += NODE_ID(pci_resource_start(pdev, PCI_CFG_REG_BAR_NUM))
-                                                       * MAX_BGX_PER_CN88XX;
+       bgx->bgx_id += nic_get_node_id(pdev) * MAX_BGX_PER_CN88XX;
+
        bgx_vnic[bgx->bgx_id] = bgx;
        bgx_get_qlm_mode(bgx);
 
 
 
 #define    MAX_LMAC    (MAX_BGX_PER_CN88XX * MAX_LMAC_PER_BGX)
 
-#define    NODE_ID_MASK                                0x300000000000
-#define    NODE_ID(x)                          ((x & NODE_ID_MASK) >> 44)
-
 /* Registers */
 #define BGX_CMRX_CFG                   0x00
 #define  CMR_PKT_TX_EN                         BIT_ULL(13)