In order to support wireless offloading on MT7981 we need to load the
appropriate firmware. Recognize MT7981 and load mt7981_wo.bin.
Signed-off-by: Daniel Golle <daniel@makrotopia.org>
Signed-off-by: David S. Miller <davem@davemloft.net>
                wo->hw->index + 1);
 
        /* load firmware */
-       fw_name = wo->hw->index ? MT7986_FIRMWARE_WO1 : MT7986_FIRMWARE_WO0;
+       if (of_device_is_compatible(wo->hw->node, "mediatek,mt7981-wed"))
+               fw_name = MT7981_FIRMWARE_WO;
+       else
+               fw_name = wo->hw->index ? MT7986_FIRMWARE_WO1 : MT7986_FIRMWARE_WO0;
+
        ret = request_firmware(&fw, fw_name, wo->hw->dev);
        if (ret)
                return ret;
                                  100, MTK_FW_DL_TIMEOUT);
 }
 
+MODULE_FIRMWARE(MT7981_FIRMWARE_WO);
 MODULE_FIRMWARE(MT7986_FIRMWARE_WO0);
 MODULE_FIRMWARE(MT7986_FIRMWARE_WO1);
 
        MTK_WED_DUMMY_CR_WO_STATUS,
 };
 
+#define MT7981_FIRMWARE_WO     "mediatek/mt7981_wo.bin"
 #define MT7986_FIRMWARE_WO0    "mediatek/mt7986_wo_0.bin"
 #define MT7986_FIRMWARE_WO1    "mediatek/mt7986_wo_1.bin"