#include "txrx.h"
 #include "util.h"
 
+static u32 rtw89_phy0_phy1_offset(struct rtw89_dev *rtwdev, u32 addr)
+{
+       const struct rtw89_phy_gen_def *phy = rtwdev->chip->phy_def;
+
+       return phy->phy0_phy1_offset(rtwdev, addr);
+}
+
 static u16 get_max_amsdu_len(struct rtw89_dev *rtwdev,
                             const struct rtw89_ra_report *report)
 {
                rtw89_rfk_parser(rtwdev, chip->nctl_post_table);
 }
 
-static u32 rtw89_phy0_phy1_offset(struct rtw89_dev *rtwdev, u32 addr)
+static u32 rtw89_phy0_phy1_offset_ax(struct rtw89_dev *rtwdev, u32 addr)
 {
        u32 phy_page = addr >> 8;
        u32 ofst = 0;
 
-       if (rtwdev->chip->chip_gen == RTW89_CHIP_BE)
-               return addr < 0x10000 ? 0x20000 : 0;
-
        switch (phy_page) {
        case 0x6:
        case 0x7:
        .ccx = &rtw89_ccx_regs_ax,
        .physts = &rtw89_physts_regs_ax,
        .cfo = &rtw89_cfo_regs_ax,
+       .phy0_phy1_offset = rtw89_phy0_phy1_offset_ax,
        .config_bb_gain = rtw89_phy_config_bb_gain_ax,
        .preinit_rf_nctl = rtw89_phy_preinit_rf_nctl_ax,
        .bb_wrap_init = NULL,
 
        const struct rtw89_ccx_regs *ccx;
        const struct rtw89_physts_regs *physts;
        const struct rtw89_cfo_regs *cfo;
+       u32 (*phy0_phy1_offset)(struct rtw89_dev *rtwdev, u32 addr);
        void (*config_bb_gain)(struct rtw89_dev *rtwdev,
                               const struct rtw89_reg2_def *reg,
                               enum rtw89_rf_path rf_path,
 
        .valid_0_mask = B_DCFO_OPT_EN_V1,
 };
 
+static u32 rtw89_phy0_phy1_offset_be(struct rtw89_dev *rtwdev, u32 addr)
+{
+       u32 phy_page = addr >> 8;
+       u32 ofst = 0;
+
+       if ((phy_page >= 0x4 && phy_page <= 0xF) ||
+           (phy_page >= 0x20 && phy_page <= 0x2B) ||
+           (phy_page >= 0x40 && phy_page <= 0x4f) ||
+           (phy_page >= 0x60 && phy_page <= 0x6f) ||
+           (phy_page >= 0xE4 && phy_page <= 0xE5) ||
+           (phy_page >= 0xE8 && phy_page <= 0xED))
+               ofst = 0x1000;
+       else
+               ofst = 0x0;
+
+       return ofst;
+}
+
 union rtw89_phy_bb_gain_arg_be {
        u32 addr;
        struct {
        .ccx = &rtw89_ccx_regs_be,
        .physts = &rtw89_physts_regs_be,
        .cfo = &rtw89_cfo_regs_be,
+       .phy0_phy1_offset = rtw89_phy0_phy1_offset_be,
        .config_bb_gain = rtw89_phy_config_bb_gain_be,
        .preinit_rf_nctl = rtw89_phy_preinit_rf_nctl_be,
        .bb_wrap_init = rtw89_phy_bb_wrap_init_be,