u32 nr_sensors;
        struct sensor_device sensors[MAX_SENSORS];
        u64 workarounds;
+       void (*early_config)(void);
 };
 
 extern struct efi_memory_map_loongson *loongson_memmap;
 extern struct loongson_system_configuration loongson_sysconf;
 
+extern u32 node_id_offset;
+extern void ls7a_early_config(void);
+extern void rs780e_early_config(void);
+
 #endif
 
 #include <linux/dma-direct.h>
 #include <linux/init.h>
 #include <linux/swiotlb.h>
+#include <boot_param.h>
 
 dma_addr_t __phys_to_dma(struct device *dev, phys_addr_t paddr)
 {
        /* We extract 2bit node id (bit 44~47, only bit 44~45 used now) from
         * Loongson-3's 48bit address space and embed it into 40bit */
        long nid = (paddr >> 44) & 0x3;
-       return ((nid << 44) ^ paddr) | (nid << 37);
+
+       return ((nid << 44) ^ paddr) | (nid << node_id_offset);
 }
 
 phys_addr_t __dma_to_phys(struct device *dev, dma_addr_t daddr)
 {
        /* We extract 2bit node id (bit 44~47, only bit 44~45 used now) from
         * Loongson-3's 48bit address space and embed it into 40bit */
-       long nid = (daddr >> 37) & 0x3;
-       return ((nid << 37) ^ daddr) | (nid << 44);
+       long nid = (daddr >> node_id_offset) & 0x3;
+
+       return ((nid << node_id_offset) ^ daddr) | (nid << 44);
 }
 
 void __init plat_swiotlb_setup(void)
 
        if (vendor == PCI_VENDOR_ID_LOONGSON && device == 0x7a00) {
                pr_info("The bridge chip is LS7A\n");
                loongson_sysconf.bridgetype = LS7A;
+               loongson_sysconf.early_config = ls7a_early_config;
        } else {
                pr_info("The bridge chip is RS780E or SR5690\n");
                loongson_sysconf.bridgetype = RS780E;
+               loongson_sysconf.early_config = rs780e_early_config;
        }
 }
 
 #include <asm/fw/fw.h>
 
 #include <loongson.h>
+#include <boot_param.h>
+
+#define NODE_ID_OFFSET_ADDR    ((void __iomem *)TO_UNCAC(0x1001041c))
+
+u32 node_id_offset;
 
 static void __init mips_nmi_setup(void)
 {
        flush_icache_range((unsigned long)base, (unsigned long)base + 0x80);
 }
 
+void ls7a_early_config(void)
+{
+       node_id_offset = ((readl(NODE_ID_OFFSET_ADDR) >> 8) & 0x1f) + 36;
+}
+
+void rs780e_early_config(void)
+{
+       node_id_offset = 37;
+}
+
 void __init prom_init(void)
 {
        fw_init_cmdline();
        set_io_port_base((unsigned long)
                ioremap(LOONGSON_PCIIO_BASE, LOONGSON_PCIIO_SIZE));
 
+       loongson_sysconf.early_config();
+
        prom_init_numa_memory();
 
        /* Hardcode to CPU UART 0 */