bool rx_win_ord_mode;
 };
 
+/*
+ * Helper to map a chip id to VAS id.
+ * For POWER9, this is a 1:1 mapping. In the future this maybe a 1:N
+ * mapping in which case, we will need to update this helper.
+ *
+ * Return the VAS id or -1 if no matching vasid is found.
+ */
+int chip_to_vas_id(int chipid);
+
 /*
  * Helper to initialize receive window attributes to defaults for an
  * NX window.
 
        return NULL;
 }
 
+int chip_to_vas_id(int chipid)
+{
+       int cpu;
+
+       for_each_possible_cpu(cpu) {
+               if (cpu_to_chip_id(cpu) == chipid)
+                       return per_cpu(cpu_vas_id, cpu);
+       }
+       return -1;
+}
+
 static int vas_probe(struct platform_device *pdev)
 {
        return init_vas_instance(pdev);
 
                return -EINVAL;
        }
 
-       for_each_compatible_node(dn, NULL, "ibm,power9-vas-x") {
-               if (of_get_ibm_chip_id(dn) == chip_id)
-                       break;
-       }
-
-       if (!dn) {
-               pr_err("Missing VAS device node\n");
+       vasid = chip_to_vas_id(chip_id);
+       if (vasid < 0) {
+               pr_err("Unable to map chip_id %d to vasid\n", chip_id);
                return -EINVAL;
        }
 
-       if (of_property_read_u32(dn, "ibm,vas-id", &vasid)) {
-               pr_err("Missing ibm,vas-id device property\n");
-               of_node_put(dn);
-               return -EINVAL;
-       }
-
-       of_node_put(dn);
-
        for_each_child_of_node(pn, dn) {
                if (of_device_is_compatible(dn, "ibm,p9-nx-842")) {
                        ret = vas_cfg_coproc_info(dn, chip_id, vasid);