Merge tag 'i2c-for-6.1-rc1' of git://git.kernel.org/pub/scm/linux/kernel/git/wsa...
[platform/kernel/linux-starfive.git] / drivers / i2c / busses / i2c-mlxbf.c
index ad5efd7..e68e775 100644 (file)
@@ -32,8 +32,6 @@
        (MLXBF_I2C_FUNC_SMBUS_DEFAULT | MLXBF_I2C_FUNC_SMBUS_BLOCK | \
         I2C_FUNC_SMBUS_QUICK | I2C_FUNC_SLAVE)
 
-#define MLXBF_I2C_SMBUS_MAX        3
-
 /* Shared resources info in BlueField platforms. */
 
 #define MLXBF_I2C_COALESCE_TYU_ADDR    0x02801300
@@ -48,6 +46,9 @@
 #define MLXBF_I2C_COREPLL_YU_ADDR      0x02800c30
 #define MLXBF_I2C_COREPLL_YU_SIZE      0x00c
 
+#define MLXBF_I2C_COREPLL_RSH_YU_ADDR  0x13409824
+#define MLXBF_I2C_COREPLL_RSH_YU_SIZE  0x00c
+
 #define MLXBF_I2C_SHARED_RES_MAX       3
 
 /*
 /* Slave busy bit reset. */
 #define MLXBF_I2C_CAUSE_S_GW_BUSY_FALL        BIT(18)
 
-#define MLXBF_I2C_CAUSE_SLAVE_ARBITER_BITS_MASK     GENMASK(20, 0)
-
 /* Cause coalesce registers. */
 #define MLXBF_I2C_CAUSE_COALESCE_0        0x00
-#define MLXBF_I2C_CAUSE_COALESCE_1        0x04
-#define MLXBF_I2C_CAUSE_COALESCE_2        0x08
 
-#define MLXBF_I2C_CAUSE_TYU_SLAVE_BIT   MLXBF_I2C_SMBUS_MAX
+#define MLXBF_I2C_CAUSE_TYU_SLAVE_BIT   3
 #define MLXBF_I2C_CAUSE_YU_SLAVE_BIT    1
 
 /* Functional enable register. */
 #define MLXBF_I2C_GPIO_SMBUS_GW_ASSERT_PINS(num, val) \
        ((val) | (0x3 << MLXBF_I2C_GPIO_SMBUS_GW_PINS(num)))
 
-/* SMBus timing parameters. */
-#define MLXBF_I2C_SMBUS_TIMER_SCL_LOW_SCL_HIGH    0x00
-#define MLXBF_I2C_SMBUS_TIMER_FALL_RISE_SPIKE     0x04
-#define MLXBF_I2C_SMBUS_TIMER_THOLD               0x08
-#define MLXBF_I2C_SMBUS_TIMER_TSETUP_START_STOP   0x0c
-#define MLXBF_I2C_SMBUS_TIMER_TSETUP_DATA         0x10
-#define MLXBF_I2C_SMBUS_THIGH_MAX_TBUF            0x14
-#define MLXBF_I2C_SMBUS_SCL_LOW_TIMEOUT           0x18
-
 /*
  * Defines SMBus operating frequency and core clock frequency.
  * According to ADB files, default values are compliant to 100KHz SMBus
 #define MLXBF_I2C_COREPLL_CORE_OD_YU_MASK   GENMASK(3, 0)
 #define MLXBF_I2C_COREPLL_CORE_R_YU_MASK    GENMASK(31, 26)
 
+/* SMBus timing parameters. */
+#define MLXBF_I2C_SMBUS_TIMER_SCL_LOW_SCL_HIGH    0x00
+#define MLXBF_I2C_SMBUS_TIMER_FALL_RISE_SPIKE     0x04
+#define MLXBF_I2C_SMBUS_TIMER_THOLD               0x08
+#define MLXBF_I2C_SMBUS_TIMER_TSETUP_START_STOP   0x0c
+#define MLXBF_I2C_SMBUS_TIMER_TSETUP_DATA         0x10
+#define MLXBF_I2C_SMBUS_THIGH_MAX_TBUF            0x14
+#define MLXBF_I2C_SMBUS_SCL_LOW_TIMEOUT           0x18
+
+#define MLXBF_I2C_SHIFT_0   0
+#define MLXBF_I2C_SHIFT_8   8
+#define MLXBF_I2C_SHIFT_16  16
+#define MLXBF_I2C_SHIFT_24  24
+
+#define MLXBF_I2C_MASK_8    GENMASK(7, 0)
+#define MLXBF_I2C_MASK_16   GENMASK(15, 0)
 
-/* Core PLL frequency. */
-static u64 mlxbf_i2c_corepll_frequency;
+#define MLXBF_I2C_MST_ADDR_OFFSET         0x200
 
 /* SMBus Master GW. */
-#define MLXBF_I2C_SMBUS_MASTER_GW     0x200
+#define MLXBF_I2C_SMBUS_MASTER_GW         0x0
 /* Number of bytes received and sent. */
-#define MLXBF_I2C_SMBUS_RS_BYTES      0x300
+#define MLXBF_I2C_YU_SMBUS_RS_BYTES       0x100
+#define MLXBF_I2C_RSH_YU_SMBUS_RS_BYTES   0x10c
 /* Packet error check (PEC) value. */
-#define MLXBF_I2C_SMBUS_MASTER_PEC    0x304
+#define MLXBF_I2C_SMBUS_MASTER_PEC        0x104
 /* Status bits (ACK/NACK/FW Timeout). */
-#define MLXBF_I2C_SMBUS_MASTER_STATUS 0x308
+#define MLXBF_I2C_SMBUS_MASTER_STATUS     0x108
 /* SMbus Master Finite State Machine. */
-#define MLXBF_I2C_SMBUS_MASTER_FSM    0x310
-
-/*
- * When enabled, the master will issue a stop condition in case of
- * timeout while waiting for FW response.
- */
-#define MLXBF_I2C_SMBUS_EN_FW_TIMEOUT 0x31c
+#define MLXBF_I2C_YU_SMBUS_MASTER_FSM     0x110
+#define MLXBF_I2C_RSH_YU_SMBUS_MASTER_FSM 0x100
 
 /* SMBus master GW control bits offset in MLXBF_I2C_SMBUS_MASTER_GW[31:3]. */
 #define MLXBF_I2C_MASTER_LOCK_BIT         BIT(31) /* Lock bit. */
@@ -231,14 +230,14 @@ static u64 mlxbf_i2c_corepll_frequency;
 #define MLXBF_I2C_MASTER_ENABLE_READ \
        (MLXBF_I2C_MASTER_ENABLE | MLXBF_I2C_MASTER_CTL_READ_BIT)
 
-#define MLXBF_I2C_MASTER_SLV_ADDR_SHIFT   12 /* Slave address shift. */
-#define MLXBF_I2C_MASTER_WRITE_SHIFT      21 /* Control write bytes shift. */
-#define MLXBF_I2C_MASTER_SEND_PEC_SHIFT   20 /* Send PEC byte shift. */
-#define MLXBF_I2C_MASTER_PARSE_EXP_SHIFT  11 /* Parse expected bytes shift. */
-#define MLXBF_I2C_MASTER_READ_SHIFT       4  /* Control read bytes shift. */
+#define MLXBF_I2C_MASTER_WRITE_SHIFT      21 /* Control write bytes */
+#define MLXBF_I2C_MASTER_SEND_PEC_SHIFT   20 /* Send PEC byte when set to 1 */
+#define MLXBF_I2C_MASTER_PARSE_EXP_SHIFT  11 /* Control parse expected bytes */
+#define MLXBF_I2C_MASTER_SLV_ADDR_SHIFT   12 /* Slave address */
+#define MLXBF_I2C_MASTER_READ_SHIFT       4  /* Control read bytes */
 
 /* SMBus master GW Data descriptor. */
-#define MLXBF_I2C_MASTER_DATA_DESC_ADDR   0x280
+#define MLXBF_I2C_MASTER_DATA_DESC_ADDR   0x80
 #define MLXBF_I2C_MASTER_DATA_DESC_SIZE   0x80 /* Size in bytes. */
 
 /* Maximum bytes to read/write per SMBus transaction. */
@@ -264,19 +263,21 @@ static u64 mlxbf_i2c_corepll_frequency;
 #define MLXBF_I2C_SMBUS_MASTER_FSM_STOP_MASK      BIT(31)
 #define MLXBF_I2C_SMBUS_MASTER_FSM_PS_STATE_MASK  BIT(15)
 
+#define MLXBF_I2C_SLV_ADDR_OFFSET             0x400
+
 /* SMBus slave GW. */
-#define MLXBF_I2C_SMBUS_SLAVE_GW              0x400
+#define MLXBF_I2C_SMBUS_SLAVE_GW              0x0
 /* Number of bytes received and sent from/to master. */
-#define MLXBF_I2C_SMBUS_SLAVE_RS_MASTER_BYTES 0x500
+#define MLXBF_I2C_SMBUS_SLAVE_RS_MASTER_BYTES 0x100
 /* Packet error check (PEC) value. */
-#define MLXBF_I2C_SMBUS_SLAVE_PEC             0x504
+#define MLXBF_I2C_SMBUS_SLAVE_PEC             0x104
 /* SMBus slave Finite State Machine (FSM). */
-#define MLXBF_I2C_SMBUS_SLAVE_FSM             0x510
+#define MLXBF_I2C_SMBUS_SLAVE_FSM             0x110
 /*
  * Should be set when all raised causes handled, and cleared by HW on
  * every new cause.
  */
-#define MLXBF_I2C_SMBUS_SLAVE_READY           0x52c
+#define MLXBF_I2C_SMBUS_SLAVE_READY           0x12c
 
 /* SMBus slave GW control bits offset in MLXBF_I2C_SMBUS_SLAVE_GW[31:19]. */
 #define MLXBF_I2C_SLAVE_BUSY_BIT         BIT(30) /* Busy bit. */
@@ -289,23 +290,74 @@ static u64 mlxbf_i2c_corepll_frequency;
 #define MLXBF_I2C_SLAVE_SEND_PEC_SHIFT    21 /* Send PEC byte shift. */
 
 /* SMBus slave GW Data descriptor. */
-#define MLXBF_I2C_SLAVE_DATA_DESC_ADDR   0x480
+#define MLXBF_I2C_SLAVE_DATA_DESC_ADDR   0x80
 #define MLXBF_I2C_SLAVE_DATA_DESC_SIZE   0x80 /* Size in bytes. */
 
 /* SMbus slave configuration registers. */
-#define MLXBF_I2C_SMBUS_SLAVE_ADDR_CFG        0x514
+#define MLXBF_I2C_SMBUS_SLAVE_ADDR_CFG        0x114
 #define MLXBF_I2C_SMBUS_SLAVE_ADDR_CNT        16
-#define MLXBF_I2C_SMBUS_SLAVE_ADDR_EN_BIT     7
+#define MLXBF_I2C_SMBUS_SLAVE_ADDR_EN_BIT     BIT(7)
 #define MLXBF_I2C_SMBUS_SLAVE_ADDR_MASK       GENMASK(6, 0)
 
-#define MLXBF_I2C_SLAVE_ADDR_ENABLED(addr) \
-       ((addr) & (1 << MLXBF_I2C_SMBUS_SLAVE_ADDR_EN_BIT))
-
 /*
  * Timeout is given in microsends. Note also that timeout handling is not
  * exact.
  */
 #define MLXBF_I2C_SMBUS_TIMEOUT   (300 * 1000) /* 300ms */
+#define MLXBF_I2C_SMBUS_LOCK_POLL_TIMEOUT (300 * 1000) /* 300ms */
+
+/* Polling frequency in microseconds. */
+#define MLXBF_I2C_POLL_FREQ_IN_USEC        200
+
+#define MLXBF_I2C_SMBUS_OP_CNT_1   1
+#define MLXBF_I2C_SMBUS_OP_CNT_2   2
+#define MLXBF_I2C_SMBUS_OP_CNT_3   3
+#define MLXBF_I2C_SMBUS_MAX_OP_CNT MLXBF_I2C_SMBUS_OP_CNT_3
+
+/* Helper macro to define an I2C resource parameters. */
+#define MLXBF_I2C_RES_PARAMS(addr, size, str) \
+       { \
+               .start = (addr), \
+               .end = (addr) + (size) - 1, \
+               .name = (str) \
+       }
+
+enum {
+       MLXBF_I2C_TIMING_100KHZ = 100000,
+       MLXBF_I2C_TIMING_400KHZ = 400000,
+       MLXBF_I2C_TIMING_1000KHZ = 1000000,
+};
+
+enum {
+       MLXBF_I2C_F_READ = BIT(0),
+       MLXBF_I2C_F_WRITE = BIT(1),
+       MLXBF_I2C_F_NORESTART = BIT(3),
+       MLXBF_I2C_F_SMBUS_OPERATION = BIT(4),
+       MLXBF_I2C_F_SMBUS_BLOCK = BIT(5),
+       MLXBF_I2C_F_SMBUS_PEC = BIT(6),
+       MLXBF_I2C_F_SMBUS_PROCESS_CALL = BIT(7),
+};
+
+/* Mellanox BlueField chip type. */
+enum mlxbf_i2c_chip_type {
+       MLXBF_I2C_CHIP_TYPE_1, /* Mellanox BlueField-1 chip. */
+       MLXBF_I2C_CHIP_TYPE_2, /* Mellanox BlueField-2 chip. */
+       MLXBF_I2C_CHIP_TYPE_3 /* Mellanox BlueField-3 chip. */
+};
+
+/* List of chip resources that are being accessed by the driver. */
+enum {
+       MLXBF_I2C_SMBUS_RES,
+       MLXBF_I2C_MST_CAUSE_RES,
+       MLXBF_I2C_SLV_CAUSE_RES,
+       MLXBF_I2C_COALESCE_RES,
+       MLXBF_I2C_SMBUS_TIMER_RES,
+       MLXBF_I2C_SMBUS_MST_RES,
+       MLXBF_I2C_SMBUS_SLV_RES,
+       MLXBF_I2C_COREPLL_RES,
+       MLXBF_I2C_GPIO_RES,
+       MLXBF_I2C_END_RES
+};
 
 /* Encapsulates timing parameters. */
 struct mlxbf_i2c_timings {
@@ -326,27 +378,12 @@ struct mlxbf_i2c_timings {
        u32 timeout;            /* Detect clock low timeout. */
 };
 
-enum {
-       MLXBF_I2C_F_READ = BIT(0),
-       MLXBF_I2C_F_WRITE = BIT(1),
-       MLXBF_I2C_F_NORESTART = BIT(3),
-       MLXBF_I2C_F_SMBUS_OPERATION = BIT(4),
-       MLXBF_I2C_F_SMBUS_BLOCK = BIT(5),
-       MLXBF_I2C_F_SMBUS_PEC = BIT(6),
-       MLXBF_I2C_F_SMBUS_PROCESS_CALL = BIT(7),
-};
-
 struct mlxbf_i2c_smbus_operation {
        u32 flags;
        u32 length; /* Buffer length in bytes. */
        u8 *buffer;
 };
 
-#define MLXBF_I2C_SMBUS_OP_CNT_1       1
-#define MLXBF_I2C_SMBUS_OP_CNT_2       2
-#define MLXBF_I2C_SMBUS_OP_CNT_3       3
-#define MLXBF_I2C_SMBUS_MAX_OP_CNT     MLXBF_I2C_SMBUS_OP_CNT_3
-
 struct mlxbf_i2c_smbus_request {
        u8 slave;
        u8 operation_cnt;
@@ -360,24 +397,38 @@ struct mlxbf_i2c_resource {
        u8 type;
 };
 
-/* List of chip resources that are being accessed by the driver. */
-enum {
-       MLXBF_I2C_SMBUS_RES,
-       MLXBF_I2C_MST_CAUSE_RES,
-       MLXBF_I2C_SLV_CAUSE_RES,
-       MLXBF_I2C_COALESCE_RES,
-       MLXBF_I2C_COREPLL_RES,
-       MLXBF_I2C_GPIO_RES,
-       MLXBF_I2C_END_RES,
+struct mlxbf_i2c_chip_info {
+       enum mlxbf_i2c_chip_type type;
+       /* Chip shared resources that are being used by the I2C controller. */
+       struct mlxbf_i2c_resource *shared_res[MLXBF_I2C_SHARED_RES_MAX];
+
+       /* Callback to calculate the core PLL frequency. */
+       u64 (*calculate_freq)(struct mlxbf_i2c_resource *corepll_res);
+
+       /* Registers' address offset */
+       u32 smbus_master_rs_bytes_off;
+       u32 smbus_master_fsm_off;
 };
 
-/* Helper macro to define an I2C resource parameters. */
-#define MLXBF_I2C_RES_PARAMS(addr, size, str) \
-       { \
-               .start = (addr), \
-               .end = (addr) + (size) - 1, \
-               .name = (str) \
-       }
+struct mlxbf_i2c_priv {
+       const struct mlxbf_i2c_chip_info *chip;
+       struct i2c_adapter adap;
+       struct mlxbf_i2c_resource *smbus;
+       struct mlxbf_i2c_resource *timer;
+       struct mlxbf_i2c_resource *mst;
+       struct mlxbf_i2c_resource *slv;
+       struct mlxbf_i2c_resource *mst_cause;
+       struct mlxbf_i2c_resource *slv_cause;
+       struct mlxbf_i2c_resource *coalesce;
+       u64 frequency; /* Core frequency in Hz. */
+       int bus; /* Physical bus identifier. */
+       int irq;
+       struct i2c_client *slave[MLXBF_I2C_SMBUS_SLAVE_ADDR_CNT];
+       u32 resource_version;
+};
+
+/* Core PLL frequency. */
+static u64 mlxbf_i2c_corepll_frequency;
 
 static struct resource mlxbf_i2c_coalesce_tyu_params =
                MLXBF_I2C_RES_PARAMS(MLXBF_I2C_COALESCE_TYU_ADDR,
@@ -391,6 +442,10 @@ static struct resource mlxbf_i2c_corepll_yu_params =
                MLXBF_I2C_RES_PARAMS(MLXBF_I2C_COREPLL_YU_ADDR,
                                     MLXBF_I2C_COREPLL_YU_SIZE,
                                     "COREPLL_MEM");
+static struct resource mlxbf_i2c_corepll_rsh_yu_params =
+               MLXBF_I2C_RES_PARAMS(MLXBF_I2C_COREPLL_RSH_YU_ADDR,
+                                    MLXBF_I2C_COREPLL_RSH_YU_SIZE,
+                                    "COREPLL_MEM");
 static struct resource mlxbf_i2c_gpio_tyu_params =
                MLXBF_I2C_RES_PARAMS(MLXBF_I2C_GPIO_TYU_ADDR,
                                     MLXBF_I2C_GPIO_TYU_SIZE,
@@ -400,34 +455,6 @@ static struct mutex mlxbf_i2c_coalesce_lock;
 static struct mutex mlxbf_i2c_corepll_lock;
 static struct mutex mlxbf_i2c_gpio_lock;
 
-/* Mellanox BlueField chip type. */
-enum mlxbf_i2c_chip_type {
-       MLXBF_I2C_CHIP_TYPE_1, /* Mellanox BlueField-1 chip. */
-       MLXBF_I2C_CHIP_TYPE_2, /* Mallanox BlueField-2 chip. */
-};
-
-struct mlxbf_i2c_chip_info {
-       enum mlxbf_i2c_chip_type type;
-       /* Chip shared resources that are being used by the I2C controller. */
-       struct mlxbf_i2c_resource *shared_res[MLXBF_I2C_SHARED_RES_MAX];
-
-       /* Callback to calculate the core PLL frequency. */
-       u64 (*calculate_freq)(struct mlxbf_i2c_resource *corepll_res);
-};
-
-struct mlxbf_i2c_priv {
-       const struct mlxbf_i2c_chip_info *chip;
-       struct i2c_adapter adap;
-       struct mlxbf_i2c_resource *smbus;
-       struct mlxbf_i2c_resource *mst_cause;
-       struct mlxbf_i2c_resource *slv_cause;
-       struct mlxbf_i2c_resource *coalesce;
-       u64 frequency; /* Core frequency in Hz. */
-       int bus; /* Physical bus identifier. */
-       int irq;
-       struct i2c_client *slave;
-};
-
 static struct mlxbf_i2c_resource mlxbf_i2c_coalesce_res[] = {
        [MLXBF_I2C_CHIP_TYPE_1] = {
                .params = &mlxbf_i2c_coalesce_tyu_params,
@@ -447,6 +474,11 @@ static struct mlxbf_i2c_resource mlxbf_i2c_corepll_res[] = {
                .params = &mlxbf_i2c_corepll_yu_params,
                .lock = &mlxbf_i2c_corepll_lock,
                .type = MLXBF_I2C_COREPLL_RES,
+       },
+       [MLXBF_I2C_CHIP_TYPE_3] = {
+               .params = &mlxbf_i2c_corepll_rsh_yu_params,
+               .lock = &mlxbf_i2c_corepll_lock,
+               .type = MLXBF_I2C_COREPLL_RES,
        }
 };
 
@@ -463,24 +495,13 @@ static u8 mlxbf_i2c_bus_count;
 
 static struct mutex mlxbf_i2c_bus_lock;
 
-/* Polling frequency in microseconds. */
-#define MLXBF_I2C_POLL_FREQ_IN_USEC        200
-
-#define MLXBF_I2C_SHIFT_0   0
-#define MLXBF_I2C_SHIFT_8   8
-#define MLXBF_I2C_SHIFT_16  16
-#define MLXBF_I2C_SHIFT_24  24
-
-#define MLXBF_I2C_MASK_8    GENMASK(7, 0)
-#define MLXBF_I2C_MASK_16   GENMASK(15, 0)
-
 /*
  * Function to poll a set of bits at a specific address; it checks whether
  * the bits are equal to zero when eq_zero is set to 'true', and not equal
  * to zero when eq_zero is set to 'false'.
  * Note that the timeout is given in microseconds.
  */
-static u32 mlxbf_smbus_poll(void __iomem *io, u32 addr, u32 mask,
+static u32 mlxbf_i2c_poll(void __iomem *io, u32 addr, u32 mask,
                            bool eq_zero, u32  timeout)
 {
        u32 bits;
@@ -502,18 +523,37 @@ static u32 mlxbf_smbus_poll(void __iomem *io, u32 addr, u32 mask,
  * a transaction. Accordingly, this function polls the Master FSM stop
  * bit; it returns false when the bit is asserted, true if not.
  */
-static bool mlxbf_smbus_master_wait_for_idle(struct mlxbf_i2c_priv *priv)
+static bool mlxbf_i2c_smbus_master_wait_for_idle(struct mlxbf_i2c_priv *priv)
 {
        u32 mask = MLXBF_I2C_SMBUS_MASTER_FSM_STOP_MASK;
-       u32 addr = MLXBF_I2C_SMBUS_MASTER_FSM;
+       u32 addr = priv->chip->smbus_master_fsm_off;
        u32 timeout = MLXBF_I2C_SMBUS_TIMEOUT;
 
-       if (mlxbf_smbus_poll(priv->smbus->io, addr, mask, true, timeout))
+       if (mlxbf_i2c_poll(priv->mst->io, addr, mask, true, timeout))
+               return true;
+
+       return false;
+}
+
+/*
+ * wait for the lock to be released before acquiring it.
+ */
+static bool mlxbf_i2c_smbus_master_lock(struct mlxbf_i2c_priv *priv)
+{
+       if (mlxbf_i2c_poll(priv->mst->io, MLXBF_I2C_SMBUS_MASTER_GW,
+                          MLXBF_I2C_MASTER_LOCK_BIT, true,
+                          MLXBF_I2C_SMBUS_LOCK_POLL_TIMEOUT))
                return true;
 
        return false;
 }
 
+static void mlxbf_i2c_smbus_master_unlock(struct mlxbf_i2c_priv *priv)
+{
+       /* Clear the gw to clear the lock */
+       writel(0, priv->mst->io + MLXBF_I2C_SMBUS_MASTER_GW);
+}
+
 static bool mlxbf_i2c_smbus_transaction_success(u32 master_status,
                                                u32 cause_status)
 {
@@ -551,7 +591,7 @@ static int mlxbf_i2c_smbus_check_status(struct mlxbf_i2c_priv *priv)
         * then read the cause and master status bits to determine if
         * errors occurred during the transaction.
         */
-       mlxbf_smbus_poll(priv->smbus->io, MLXBF_I2C_SMBUS_MASTER_GW,
+       mlxbf_i2c_poll(priv->mst->io, MLXBF_I2C_SMBUS_MASTER_GW,
                         MLXBF_I2C_MASTER_BUSY_BIT, true,
                         MLXBF_I2C_SMBUS_TIMEOUT);
 
@@ -564,7 +604,7 @@ static int mlxbf_i2c_smbus_check_status(struct mlxbf_i2c_priv *priv)
         * Parse both Cause and Master GW bits, then return transaction status.
         */
 
-       master_status_bits = readl(priv->smbus->io +
+       master_status_bits = readl(priv->mst->io +
                                        MLXBF_I2C_SMBUS_MASTER_STATUS);
        master_status_bits &= MLXBF_I2C_SMBUS_MASTER_STATUS_MASK;
 
@@ -589,7 +629,8 @@ static int mlxbf_i2c_smbus_check_status(struct mlxbf_i2c_priv *priv)
 }
 
 static void mlxbf_i2c_smbus_write_data(struct mlxbf_i2c_priv *priv,
-                                      const u8 *data, u8 length, u32 addr)
+                                      const u8 *data, u8 length, u32 addr,
+                                      bool is_master)
 {
        u8 offset, aligned_length;
        u32 data32;
@@ -606,12 +647,16 @@ static void mlxbf_i2c_smbus_write_data(struct mlxbf_i2c_priv *priv,
         */
        for (offset = 0; offset < aligned_length; offset += sizeof(u32)) {
                data32 = *((u32 *)(data + offset));
-               iowrite32be(data32, priv->smbus->io + addr + offset);
+               if (is_master)
+                       iowrite32be(data32, priv->mst->io + addr + offset);
+               else
+                       iowrite32be(data32, priv->slv->io + addr + offset);
        }
 }
 
 static void mlxbf_i2c_smbus_read_data(struct mlxbf_i2c_priv *priv,
-                                     u8 *data, u8 length, u32 addr)
+                                     u8 *data, u8 length, u32 addr,
+                                     bool is_master)
 {
        u32 data32, mask;
        u8 byte, offset;
@@ -627,14 +672,20 @@ static void mlxbf_i2c_smbus_read_data(struct mlxbf_i2c_priv *priv,
         */
 
        for (offset = 0; offset < (length & ~mask); offset += sizeof(u32)) {
-               data32 = ioread32be(priv->smbus->io + addr + offset);
+               if (is_master)
+                       data32 = ioread32be(priv->mst->io + addr + offset);
+               else
+                       data32 = ioread32be(priv->slv->io + addr + offset);
                *((u32 *)(data + offset)) = data32;
        }
 
        if (!(length & mask))
                return;
 
-       data32 = ioread32be(priv->smbus->io + addr + offset);
+       if (is_master)
+               data32 = ioread32be(priv->mst->io + addr + offset);
+       else
+               data32 = ioread32be(priv->slv->io + addr + offset);
 
        for (byte = 0; byte < (length & mask); byte++) {
                data[offset + byte] = data32 & GENMASK(7, 0);
@@ -660,16 +711,16 @@ static int mlxbf_i2c_smbus_enable(struct mlxbf_i2c_priv *priv, u8 slave,
        command |= rol32(pec_en, MLXBF_I2C_MASTER_SEND_PEC_SHIFT);
 
        /* Clear status bits. */
-       writel(0x0, priv->smbus->io + MLXBF_I2C_SMBUS_MASTER_STATUS);
+       writel(0x0, priv->mst->io + MLXBF_I2C_SMBUS_MASTER_STATUS);
        /* Set the cause data. */
        writel(~0x0, priv->mst_cause->io + MLXBF_I2C_CAUSE_OR_CLEAR);
        /* Zero PEC byte. */
-       writel(0x0, priv->smbus->io + MLXBF_I2C_SMBUS_MASTER_PEC);
+       writel(0x0, priv->mst->io + MLXBF_I2C_SMBUS_MASTER_PEC);
        /* Zero byte count. */
-       writel(0x0, priv->smbus->io + MLXBF_I2C_SMBUS_RS_BYTES);
+       writel(0x0, priv->mst->io + priv->chip->smbus_master_rs_bytes_off);
 
        /* GW activation. */
-       writel(command, priv->smbus->io + MLXBF_I2C_SMBUS_MASTER_GW);
+       writel(command, priv->mst->io + MLXBF_I2C_SMBUS_MASTER_GW);
 
        /*
         * Poll master status and check status bits. An ACK is sent when
@@ -705,10 +756,19 @@ mlxbf_i2c_smbus_start_transaction(struct mlxbf_i2c_priv *priv,
        slave = request->slave & GENMASK(6, 0);
        addr = slave << 1;
 
-       /* First of all, check whether the HW is idle. */
-       if (WARN_ON(!mlxbf_smbus_master_wait_for_idle(priv)))
+       /*
+        * Try to acquire the smbus gw lock before any reads of the GW register since
+        * a read sets the lock.
+        */
+       if (WARN_ON(!mlxbf_i2c_smbus_master_lock(priv)))
                return -EBUSY;
 
+       /* Check whether the HW is idle */
+       if (WARN_ON(!mlxbf_i2c_smbus_master_wait_for_idle(priv))) {
+               ret = -EBUSY;
+               goto out_unlock;
+       }
+
        /* Set first byte. */
        data_desc[data_idx++] = addr;
 
@@ -732,8 +792,10 @@ mlxbf_i2c_smbus_start_transaction(struct mlxbf_i2c_priv *priv,
                        write_en = 1;
                        write_len += operation->length;
                        if (data_idx + operation->length >
-                                       MLXBF_I2C_MASTER_DATA_DESC_SIZE)
-                               return -ENOBUFS;
+                                       MLXBF_I2C_MASTER_DATA_DESC_SIZE) {
+                               ret = -ENOBUFS;
+                               goto out_unlock;
+                       }
                        memcpy(data_desc + data_idx,
                               operation->buffer, operation->length);
                        data_idx += operation->length;
@@ -759,25 +821,25 @@ mlxbf_i2c_smbus_start_transaction(struct mlxbf_i2c_priv *priv,
         * must be written to the data registers.
         */
        mlxbf_i2c_smbus_write_data(priv, (const u8 *)data_desc, data_len,
-                                  MLXBF_I2C_MASTER_DATA_DESC_ADDR);
+                                  MLXBF_I2C_MASTER_DATA_DESC_ADDR, true);
 
        if (write_en) {
                ret = mlxbf_i2c_smbus_enable(priv, slave, write_len, block_en,
                                         pec_en, 0);
                if (ret)
-                       return ret;
+                       goto out_unlock;
        }
 
        if (read_en) {
                /* Write slave address to Master GW data descriptor. */
                mlxbf_i2c_smbus_write_data(priv, (const u8 *)&addr, 1,
-                                          MLXBF_I2C_MASTER_DATA_DESC_ADDR);
+                                          MLXBF_I2C_MASTER_DATA_DESC_ADDR, true);
                ret = mlxbf_i2c_smbus_enable(priv, slave, read_len, block_en,
                                         pec_en, 1);
                if (!ret) {
                        /* Get Master GW data descriptor. */
                        mlxbf_i2c_smbus_read_data(priv, data_desc, read_len + 1,
-                                            MLXBF_I2C_MASTER_DATA_DESC_ADDR);
+                                            MLXBF_I2C_MASTER_DATA_DESC_ADDR, true);
 
                        /* Get data from Master GW data descriptor. */
                        memcpy(read_buf, data_desc, read_len + 1);
@@ -789,9 +851,12 @@ mlxbf_i2c_smbus_start_transaction(struct mlxbf_i2c_priv *priv,
                 * next tag integration.
                 */
                writel(MLXBF_I2C_SMBUS_MASTER_FSM_PS_STATE_MASK,
-                       priv->smbus->io + MLXBF_I2C_SMBUS_MASTER_FSM);
+                       priv->mst->io + priv->chip->smbus_master_fsm_off);
        }
 
+out_unlock:
+       mlxbf_i2c_smbus_master_unlock(priv);
+
        return ret;
 }
 
@@ -1078,7 +1143,7 @@ static void mlxbf_i2c_set_timings(struct mlxbf_i2c_priv *priv,
        timer |= mlxbf_i2c_set_timer(priv, timings->scl_low,
                                     false, MLXBF_I2C_MASK_16,
                                     MLXBF_I2C_SHIFT_16);
-       writel(timer, priv->smbus->io +
+       writel(timer, priv->timer->io +
                MLXBF_I2C_SMBUS_TIMER_SCL_LOW_SCL_HIGH);
 
        timer = mlxbf_i2c_set_timer(priv, timings->sda_rise, false,
@@ -1089,34 +1154,34 @@ static void mlxbf_i2c_set_timings(struct mlxbf_i2c_priv *priv,
                                     MLXBF_I2C_MASK_8, MLXBF_I2C_SHIFT_16);
        timer |= mlxbf_i2c_set_timer(priv, timings->scl_fall, false,
                                     MLXBF_I2C_MASK_8, MLXBF_I2C_SHIFT_24);
-       writel(timer, priv->smbus->io +
+       writel(timer, priv->timer->io +
                MLXBF_I2C_SMBUS_TIMER_FALL_RISE_SPIKE);
 
        timer = mlxbf_i2c_set_timer(priv, timings->hold_start, true,
                                    MLXBF_I2C_MASK_16, MLXBF_I2C_SHIFT_0);
        timer |= mlxbf_i2c_set_timer(priv, timings->hold_data, true,
                                     MLXBF_I2C_MASK_16, MLXBF_I2C_SHIFT_16);
-       writel(timer, priv->smbus->io + MLXBF_I2C_SMBUS_TIMER_THOLD);
+       writel(timer, priv->timer->io + MLXBF_I2C_SMBUS_TIMER_THOLD);
 
        timer = mlxbf_i2c_set_timer(priv, timings->setup_start, true,
                                    MLXBF_I2C_MASK_16, MLXBF_I2C_SHIFT_0);
        timer |= mlxbf_i2c_set_timer(priv, timings->setup_stop, true,
                                     MLXBF_I2C_MASK_16, MLXBF_I2C_SHIFT_16);
-       writel(timer, priv->smbus->io +
+       writel(timer, priv->timer->io +
                MLXBF_I2C_SMBUS_TIMER_TSETUP_START_STOP);
 
        timer = mlxbf_i2c_set_timer(priv, timings->setup_data, true,
                                    MLXBF_I2C_MASK_16, MLXBF_I2C_SHIFT_0);
-       writel(timer, priv->smbus->io + MLXBF_I2C_SMBUS_TIMER_TSETUP_DATA);
+       writel(timer, priv->timer->io + MLXBF_I2C_SMBUS_TIMER_TSETUP_DATA);
 
        timer = mlxbf_i2c_set_timer(priv, timings->buf, false,
                                    MLXBF_I2C_MASK_16, MLXBF_I2C_SHIFT_0);
        timer |= mlxbf_i2c_set_timer(priv, timings->thigh_max, false,
                                     MLXBF_I2C_MASK_16, MLXBF_I2C_SHIFT_16);
-       writel(timer, priv->smbus->io + MLXBF_I2C_SMBUS_THIGH_MAX_TBUF);
+       writel(timer, priv->timer->io + MLXBF_I2C_SMBUS_THIGH_MAX_TBUF);
 
        timer = timings->timeout;
-       writel(timer, priv->smbus->io + MLXBF_I2C_SMBUS_SCL_LOW_TIMEOUT);
+       writel(timer, priv->timer->io + MLXBF_I2C_SMBUS_SCL_LOW_TIMEOUT);
 }
 
 enum mlxbf_i2c_timings_config {
@@ -1509,28 +1574,26 @@ static int mlxbf_i2c_calculate_corepll_freq(struct platform_device *pdev,
        return 0;
 }
 
-static int mlxbf_slave_enable(struct mlxbf_i2c_priv *priv, u8 addr)
+static int mlxbf_i2c_slave_enable(struct mlxbf_i2c_priv *priv,
+                             struct i2c_client *slave)
 {
-       u32 slave_reg, slave_reg_tmp, slave_reg_avail, slave_addr_mask;
-       u8 reg, reg_cnt, byte, addr_tmp, reg_avail, byte_avail;
-       bool avail, disabled;
-
-       disabled = false;
-       avail = false;
+       u8 reg, reg_cnt, byte, addr_tmp;
+       u32 slave_reg, slave_reg_tmp;
 
        if (!priv)
                return -EPERM;
 
        reg_cnt = MLXBF_I2C_SMBUS_SLAVE_ADDR_CNT >> 2;
-       slave_addr_mask = MLXBF_I2C_SMBUS_SLAVE_ADDR_MASK;
 
        /*
         * Read the slave registers. There are 4 * 32-bit slave registers.
-        * Each slave register can hold up to 4 * 8-bit slave configuration
-        * (7-bit address, 1 status bit (1 if enabled, 0 if not)).
+        * Each slave register can hold up to 4 * 8-bit slave configuration:
+        * 1) A 7-bit address
+        * 2) And a status bit (1 if enabled, 0 if not).
+        * Look for the next available slave register slot.
         */
        for (reg = 0; reg < reg_cnt; reg++) {
-               slave_reg = readl(priv->smbus->io +
+               slave_reg = readl(priv->slv->io +
                                MLXBF_I2C_SMBUS_SLAVE_ADDR_CFG + reg * 0x4);
                /*
                 * Each register holds 4 slave addresses. So, we have to keep
@@ -1542,121 +1605,87 @@ static int mlxbf_slave_enable(struct mlxbf_i2c_priv *priv, u8 addr)
                        addr_tmp = slave_reg_tmp & GENMASK(7, 0);
 
                        /*
-                        * Mark the first available slave address slot, i.e. its
-                        * enabled bit should be unset. This slot might be used
-                        * later on to register our slave.
-                        */
-                       if (!avail && !MLXBF_I2C_SLAVE_ADDR_ENABLED(addr_tmp)) {
-                               avail = true;
-                               reg_avail = reg;
-                               byte_avail = byte;
-                               slave_reg_avail = slave_reg;
-                       }
-
-                       /*
-                        * Parse slave address bytes and check whether the
-                        * slave address already exists and it's enabled,
-                        * i.e. most significant bit is set.
+                        * If an enable bit is not set in the
+                        * MLXBF_I2C_SMBUS_SLAVE_ADDR_CFG register, then the
+                        * slave address slot associated with that bit is
+                        * free. So set the enable bit and write the
+                        * slave address bits.
                         */
-                       if ((addr_tmp & slave_addr_mask) == addr) {
-                               if (MLXBF_I2C_SLAVE_ADDR_ENABLED(addr_tmp))
-                                       return 0;
-                               disabled = true;
-                               break;
+                       if (!(addr_tmp & MLXBF_I2C_SMBUS_SLAVE_ADDR_EN_BIT)) {
+                               slave_reg &= ~(MLXBF_I2C_SMBUS_SLAVE_ADDR_MASK << (byte * 8));
+                               slave_reg |= (slave->addr << (byte * 8));
+                               slave_reg |= MLXBF_I2C_SMBUS_SLAVE_ADDR_EN_BIT << (byte * 8);
+                               writel(slave_reg, priv->slv->io +
+                                       MLXBF_I2C_SMBUS_SLAVE_ADDR_CFG +
+                                       (reg * 0x4));
+
+                               /*
+                                * Set the slave at the corresponding index.
+                                */
+                               priv->slave[(reg * 4) + byte] = slave;
+
+                               return 0;
                        }
 
                        /* Parse next byte. */
                        slave_reg_tmp >>= 8;
                }
-
-               /* Exit the loop if the slave address is found. */
-               if (disabled)
-                       break;
-       }
-
-       if (!avail && !disabled)
-               return -EINVAL; /* No room for a new slave address. */
-
-       if (avail && !disabled) {
-               reg = reg_avail;
-               byte = byte_avail;
-               /* Set the slave address. */
-               slave_reg_avail &= ~(slave_addr_mask << (byte * 8));
-               slave_reg_avail |= addr << (byte * 8);
-               slave_reg = slave_reg_avail;
        }
 
-       /* Enable the slave address and update the register. */
-       slave_reg |= (1 << MLXBF_I2C_SMBUS_SLAVE_ADDR_EN_BIT) << (byte * 8);
-       writel(slave_reg, priv->smbus->io + MLXBF_I2C_SMBUS_SLAVE_ADDR_CFG +
-               reg * 0x4);
-
-       return 0;
+       return -EBUSY;
 }
 
-static int mlxbf_slave_disable(struct mlxbf_i2c_priv *priv)
+static int mlxbf_i2c_slave_disable(struct mlxbf_i2c_priv *priv, u8 addr)
 {
-       u32 slave_reg, slave_reg_tmp, slave_addr_mask;
-       u8 addr, addr_tmp, reg, reg_cnt, slave_byte;
-       struct i2c_client *client = priv->slave;
-       bool exist;
+       u8 addr_tmp, reg, reg_cnt, byte;
+       u32 slave_reg, slave_reg_tmp;
 
-       exist = false;
-
-       addr = client->addr;
        reg_cnt = MLXBF_I2C_SMBUS_SLAVE_ADDR_CNT >> 2;
-       slave_addr_mask = MLXBF_I2C_SMBUS_SLAVE_ADDR_MASK;
 
        /*
         * Read the slave registers. There are 4 * 32-bit slave registers.
-        * Each slave register can hold up to 4 * 8-bit slave configuration
-        * (7-bit address, 1 status bit (1 if enabled, 0 if not)).
+        * Each slave register can hold up to 4 * 8-bit slave configuration:
+        * 1) A 7-bit address
+        * 2) And a status bit (1 if enabled, 0 if not).
+        * Check if addr is present in the registers.
         */
        for (reg = 0; reg < reg_cnt; reg++) {
-               slave_reg = readl(priv->smbus->io +
+               slave_reg = readl(priv->slv->io +
                                MLXBF_I2C_SMBUS_SLAVE_ADDR_CFG + reg * 0x4);
 
                /* Check whether the address slots are empty. */
-               if (slave_reg == 0)
+               if (!slave_reg)
                        continue;
 
                /*
-                * Each register holds 4 slave addresses. So, we have to keep
-                * the byte order consistent with the value read in order to
-                * update the register correctly, if needed.
+                * Check if addr matches any of the 4 slave addresses
+                * in the register.
                 */
                slave_reg_tmp = slave_reg;
-               slave_byte = 0;
-               while (slave_reg_tmp != 0) {
-                       addr_tmp = slave_reg_tmp & slave_addr_mask;
+               for (byte = 0; byte < 4; byte++) {
+                       addr_tmp = slave_reg_tmp & MLXBF_I2C_SMBUS_SLAVE_ADDR_MASK;
                        /*
                         * Parse slave address bytes and check whether the
                         * slave address already exists.
                         */
                        if (addr_tmp == addr) {
-                               exist = true;
-                               break;
+                               /* Clear the slave address slot. */
+                               slave_reg &= ~(GENMASK(7, 0) << (byte * 8));
+                               writel(slave_reg, priv->slv->io +
+                                       MLXBF_I2C_SMBUS_SLAVE_ADDR_CFG +
+                                       (reg * 0x4));
+                               /* Free slave at the corresponding index */
+                               priv->slave[(reg * 4) + byte] = NULL;
+
+                               return 0;
                        }
 
                        /* Parse next byte. */
                        slave_reg_tmp >>= 8;
-                       slave_byte += 1;
                }
-
-               /* Exit the loop if the slave address is found. */
-               if (exist)
-                       break;
        }
 
-       if (!exist)
-               return 0; /* Slave is not registered, nothing to do. */
-
-       /* Cleanup the slave address slot. */
-       slave_reg &= ~(GENMASK(7, 0) << (slave_byte * 8));
-       writel(slave_reg, priv->smbus->io + MLXBF_I2C_SMBUS_SLAVE_ADDR_CFG +
-               reg * 0x4);
-
-       return 0;
+       return -ENXIO;
 }
 
 static int mlxbf_i2c_init_coalesce(struct platform_device *pdev,
@@ -1746,7 +1775,7 @@ static int mlxbf_i2c_init_slave(struct platform_device *pdev,
        int ret;
 
        /* Reset FSM. */
-       writel(0, priv->smbus->io + MLXBF_I2C_SMBUS_SLAVE_FSM);
+       writel(0, priv->slv->io + MLXBF_I2C_SMBUS_SLAVE_FSM);
 
        /*
         * Enable slave cause interrupt bits. Drive
@@ -1761,7 +1790,7 @@ static int mlxbf_i2c_init_slave(struct platform_device *pdev,
        writel(int_reg, priv->slv_cause->io + MLXBF_I2C_CAUSE_OR_EVTEN0);
 
        /* Finally, set the 'ready' bit to start handling transactions. */
-       writel(0x1, priv->smbus->io + MLXBF_I2C_SMBUS_SLAVE_READY);
+       writel(0x1, priv->slv->io + MLXBF_I2C_SMBUS_SLAVE_READY);
 
        /* Initialize the cause coalesce resource. */
        ret = mlxbf_i2c_init_coalesce(pdev, priv);
@@ -1806,84 +1835,93 @@ static bool mlxbf_i2c_has_coalesce(struct mlxbf_i2c_priv *priv, bool *read,
        return true;
 }
 
-static bool mlxbf_smbus_slave_wait_for_idle(struct mlxbf_i2c_priv *priv,
+static bool mlxbf_i2c_slave_wait_for_idle(struct mlxbf_i2c_priv *priv,
                                            u32 timeout)
 {
        u32 mask = MLXBF_I2C_CAUSE_S_GW_BUSY_FALL;
        u32 addr = MLXBF_I2C_CAUSE_ARBITER;
 
-       if (mlxbf_smbus_poll(priv->slv_cause->io, addr, mask, false, timeout))
+       if (mlxbf_i2c_poll(priv->slv_cause->io, addr, mask, false, timeout))
                return true;
 
        return false;
 }
 
-/* Send byte to 'external' smbus master. */
-static int mlxbf_smbus_irq_send(struct mlxbf_i2c_priv *priv, u8 recv_bytes)
+static struct i2c_client *mlxbf_i2c_get_slave_from_addr(
+                       struct mlxbf_i2c_priv *priv, u8 addr)
 {
-       u8 data_desc[MLXBF_I2C_SLAVE_DATA_DESC_SIZE] = { 0 };
-       u8 write_size, pec_en, addr, byte, value, byte_cnt, desc_size;
-       struct i2c_client *slave = priv->slave;
-       u32 control32, data32;
-       int ret;
+       int i;
 
-       if (!slave)
-               return -EINVAL;
+       for (i = 0; i < MLXBF_I2C_SMBUS_SLAVE_ADDR_CNT; i++) {
+               if (!priv->slave[i])
+                       continue;
+
+               if (priv->slave[i]->addr == addr)
+                       return priv->slave[i];
+       }
 
-       addr = 0;
-       byte = 0;
-       desc_size = MLXBF_I2C_SLAVE_DATA_DESC_SIZE;
+       return NULL;
+}
+
+/*
+ * Send byte to 'external' smbus master. This function is executed when
+ * an external smbus master wants to read data from the BlueField.
+ */
+static int mlxbf_i2c_irq_send(struct mlxbf_i2c_priv *priv, u8 recv_bytes)
+{
+       u8 data_desc[MLXBF_I2C_SLAVE_DATA_DESC_SIZE] = { 0 };
+       u8 write_size, pec_en, addr, value, byte_cnt;
+       struct i2c_client *slave;
+       u32 control32, data32;
+       int ret = 0;
 
        /*
-        * Read bytes received from the external master. These bytes should
-        * be located in the first data descriptor register of the slave GW.
-        * These bytes are the slave address byte and the internal register
-        * address, if supplied.
+        * Read the first byte received from the external master to
+        * determine the slave address. This byte is located in the
+        * first data descriptor register of the slave GW.
         */
-       if (recv_bytes > 0) {
-               data32 = ioread32be(priv->smbus->io +
-                                       MLXBF_I2C_SLAVE_DATA_DESC_ADDR);
-
-               /* Parse the received bytes. */
-               switch (recv_bytes) {
-               case 2:
-                       byte = (data32 >> 8) & GENMASK(7, 0);
-                       fallthrough;
-               case 1:
-                       addr = (data32 & GENMASK(7, 0)) >> 1;
-               }
+       data32 = ioread32be(priv->slv->io +
+                               MLXBF_I2C_SLAVE_DATA_DESC_ADDR);
+       addr = (data32 & GENMASK(7, 0)) >> 1;
 
-               /* Check whether it's our slave address. */
-               if (slave->addr != addr)
-                       return -EINVAL;
+       /*
+        * Check if the slave address received in the data descriptor register
+        * matches any of the slave addresses registered. If there is a match,
+        * set the slave.
+        */
+       slave = mlxbf_i2c_get_slave_from_addr(priv, addr);
+       if (!slave) {
+               ret = -ENXIO;
+               goto clear_csr;
        }
 
        /*
-        * I2C read transactions may start by a WRITE followed by a READ.
-        * Indeed, most slave devices would expect the internal address
-        * following the slave address byte. So, write that byte first,
-        * and then, send the requested data bytes to the master.
+        * An I2C read can consist of a WRITE bit transaction followed by
+        * a READ bit transaction. Indeed, slave devices often expect
+        * the slave address to be followed by the internal address.
+        * So, write the internal address byte first, and then, send the
+        * requested data to the master.
         */
        if (recv_bytes > 1) {
                i2c_slave_event(slave, I2C_SLAVE_WRITE_REQUESTED, &value);
-               value = byte;
+               value = (data32 >> 8) & GENMASK(7, 0);
                ret = i2c_slave_event(slave, I2C_SLAVE_WRITE_RECEIVED,
                                      &value);
                i2c_slave_event(slave, I2C_SLAVE_STOP, &value);
 
                if (ret < 0)
-                       return ret;
+                       goto clear_csr;
        }
 
        /*
-        * Now, send data to the master; currently, the driver supports
-        * READ_BYTE, READ_WORD and BLOCK READ protocols. Note that the
-        * hardware can send up to 128 bytes per transfer. That is the
-        * size of its data registers.
+        * Send data to the master. Currently, the driver supports
+        * READ_BYTE, READ_WORD and BLOCK READ protocols. The
+        * hardware can send up to 128 bytes per transfer which is
+        * the total size of the data registers.
         */
        i2c_slave_event(slave, I2C_SLAVE_READ_REQUESTED, &value);
 
-       for (byte_cnt = 0; byte_cnt < desc_size; byte_cnt++) {
+       for (byte_cnt = 0; byte_cnt < MLXBF_I2C_SLAVE_DATA_DESC_SIZE; byte_cnt++) {
                data_desc[byte_cnt] = value;
                i2c_slave_event(slave, I2C_SLAVE_READ_PROCESSED, &value);
        }
@@ -1891,14 +1929,12 @@ static int mlxbf_smbus_irq_send(struct mlxbf_i2c_priv *priv, u8 recv_bytes)
        /* Send a stop condition to the backend. */
        i2c_slave_event(slave, I2C_SLAVE_STOP, &value);
 
-       /* Handle the actual transfer. */
-
        /* Set the number of bytes to write to master. */
        write_size = (byte_cnt - 1) & 0x7f;
 
        /* Write data to Slave GW data descriptor. */
        mlxbf_i2c_smbus_write_data(priv, data_desc, byte_cnt,
-                                  MLXBF_I2C_SLAVE_DATA_DESC_ADDR);
+                                  MLXBF_I2C_SLAVE_DATA_DESC_ADDR, false);
 
        pec_en = 0; /* Disable PEC since it is not supported. */
 
@@ -1907,46 +1943,52 @@ static int mlxbf_smbus_irq_send(struct mlxbf_i2c_priv *priv, u8 recv_bytes)
        control32 |= rol32(write_size, MLXBF_I2C_SLAVE_WRITE_BYTES_SHIFT);
        control32 |= rol32(pec_en, MLXBF_I2C_SLAVE_SEND_PEC_SHIFT);
 
-       writel(control32, priv->smbus->io + MLXBF_I2C_SMBUS_SLAVE_GW);
+       writel(control32, priv->slv->io + MLXBF_I2C_SMBUS_SLAVE_GW);
 
        /*
         * Wait until the transfer is completed; the driver will wait
         * until the GW is idle, a cause will rise on fall of GW busy.
         */
-       mlxbf_smbus_slave_wait_for_idle(priv, MLXBF_I2C_SMBUS_TIMEOUT);
+       mlxbf_i2c_slave_wait_for_idle(priv, MLXBF_I2C_SMBUS_TIMEOUT);
 
+clear_csr:
        /* Release the Slave GW. */
-       writel(0x0, priv->smbus->io + MLXBF_I2C_SMBUS_SLAVE_RS_MASTER_BYTES);
-       writel(0x0, priv->smbus->io + MLXBF_I2C_SMBUS_SLAVE_PEC);
-       writel(0x1, priv->smbus->io + MLXBF_I2C_SMBUS_SLAVE_READY);
+       writel(0x0, priv->slv->io + MLXBF_I2C_SMBUS_SLAVE_RS_MASTER_BYTES);
+       writel(0x0, priv->slv->io + MLXBF_I2C_SMBUS_SLAVE_PEC);
+       writel(0x1, priv->slv->io + MLXBF_I2C_SMBUS_SLAVE_READY);
 
-       return 0;
+       return ret;
 }
 
-/* Receive bytes from 'external' smbus master. */
-static int mlxbf_smbus_irq_recv(struct mlxbf_i2c_priv *priv, u8 recv_bytes)
+/*
+ * Receive bytes from 'external' smbus master. This function is executed when
+ * an external smbus master wants to write data to the BlueField.
+ */
+static int mlxbf_i2c_irq_recv(struct mlxbf_i2c_priv *priv, u8 recv_bytes)
 {
        u8 data_desc[MLXBF_I2C_SLAVE_DATA_DESC_SIZE] = { 0 };
-       struct i2c_client *slave = priv->slave;
+       struct i2c_client *slave;
        u8 value, byte, addr;
        int ret = 0;
 
-       if (!slave)
-               return -EINVAL;
-
        /* Read data from Slave GW data descriptor. */
        mlxbf_i2c_smbus_read_data(priv, data_desc, recv_bytes,
-                                 MLXBF_I2C_SLAVE_DATA_DESC_ADDR);
-
-       /* Check whether its our slave address. */
+                                 MLXBF_I2C_SLAVE_DATA_DESC_ADDR, false);
        addr = data_desc[0] >> 1;
-       if (slave->addr != addr)
-               return -EINVAL;
 
        /*
-        * Notify the slave backend; another I2C master wants to write data
-        * to us. This event is sent once the slave address and the write bit
-        * is detected.
+        * Check if the slave address received in the data descriptor register
+        * matches any of the slave addresses registered.
+        */
+       slave = mlxbf_i2c_get_slave_from_addr(priv, addr);
+       if (!slave) {
+               ret = -EINVAL;
+               goto clear_csr;
+       }
+
+       /*
+        * Notify the slave backend that an smbus master wants to write data
+        * to the BlueField.
         */
        i2c_slave_event(slave, I2C_SLAVE_WRITE_REQUESTED, &value);
 
@@ -1959,18 +2001,22 @@ static int mlxbf_smbus_irq_recv(struct mlxbf_i2c_priv *priv, u8 recv_bytes)
                        break;
        }
 
-       /* Send a stop condition to the backend. */
+       /*
+        * Send a stop event to the slave backend, to signal
+        * the end of the write transactions.
+        */
        i2c_slave_event(slave, I2C_SLAVE_STOP, &value);
 
+clear_csr:
        /* Release the Slave GW. */
-       writel(0x0, priv->smbus->io + MLXBF_I2C_SMBUS_SLAVE_RS_MASTER_BYTES);
-       writel(0x0, priv->smbus->io + MLXBF_I2C_SMBUS_SLAVE_PEC);
-       writel(0x1, priv->smbus->io + MLXBF_I2C_SMBUS_SLAVE_READY);
+       writel(0x0, priv->slv->io + MLXBF_I2C_SMBUS_SLAVE_RS_MASTER_BYTES);
+       writel(0x0, priv->slv->io + MLXBF_I2C_SMBUS_SLAVE_PEC);
+       writel(0x1, priv->slv->io + MLXBF_I2C_SMBUS_SLAVE_READY);
 
        return ret;
 }
 
-static irqreturn_t mlxbf_smbus_irq(int irq, void *ptr)
+static irqreturn_t mlxbf_i2c_irq(int irq, void *ptr)
 {
        struct mlxbf_i2c_priv *priv = ptr;
        bool read, write, irq_is_set;
@@ -2000,7 +2046,7 @@ static irqreturn_t mlxbf_smbus_irq(int irq, void *ptr)
         * slave, if the higher 8 bits are sent then the slave expect N bytes
         * from the master.
         */
-       rw_bytes_reg = readl(priv->smbus->io +
+       rw_bytes_reg = readl(priv->slv->io +
                                MLXBF_I2C_SMBUS_SLAVE_RS_MASTER_BYTES);
        recv_bytes = (rw_bytes_reg >> 8) & GENMASK(7, 0);
 
@@ -2018,9 +2064,9 @@ static irqreturn_t mlxbf_smbus_irq(int irq, void *ptr)
                MLXBF_I2C_SLAVE_DATA_DESC_SIZE : recv_bytes;
 
        if (read)
-               mlxbf_smbus_irq_send(priv, recv_bytes);
+               mlxbf_i2c_irq_send(priv, recv_bytes);
        else
-               mlxbf_smbus_irq_recv(priv, recv_bytes);
+               mlxbf_i2c_irq_recv(priv, recv_bytes);
 
        return IRQ_HANDLED;
 }
@@ -2115,23 +2161,21 @@ static s32 mlxbf_i2c_smbus_xfer(struct i2c_adapter *adap, u16 addr,
 static int mlxbf_i2c_reg_slave(struct i2c_client *slave)
 {
        struct mlxbf_i2c_priv *priv = i2c_get_adapdata(slave->adapter);
+       struct device *dev = &slave->dev;
        int ret;
 
-       if (priv->slave)
-               return -EBUSY;
-
        /*
         * Do not support ten bit chip address and do not use Packet Error
         * Checking (PEC).
         */
-       if (slave->flags & (I2C_CLIENT_TEN | I2C_CLIENT_PEC))
+       if (slave->flags & (I2C_CLIENT_TEN | I2C_CLIENT_PEC)) {
+               dev_err(dev, "SMBus PEC and 10 bit address not supported\n");
                return -EAFNOSUPPORT;
+       }
 
-       ret = mlxbf_slave_enable(priv, slave->addr);
-       if (ret < 0)
-               return ret;
-
-       priv->slave = slave;
+       ret = mlxbf_i2c_slave_enable(priv, slave);
+       if (ret)
+               dev_err(dev, "Surpassed max number of registered slaves allowed\n");
 
        return 0;
 }
@@ -2139,18 +2183,19 @@ static int mlxbf_i2c_reg_slave(struct i2c_client *slave)
 static int mlxbf_i2c_unreg_slave(struct i2c_client *slave)
 {
        struct mlxbf_i2c_priv *priv = i2c_get_adapdata(slave->adapter);
+       struct device *dev = &slave->dev;
        int ret;
 
-       WARN_ON(!priv->slave);
-
-       /* Unregister slave, i.e. disable the slave address in hardware. */
-       ret = mlxbf_slave_disable(priv);
-       if (ret < 0)
-               return ret;
-
-       priv->slave = NULL;
+       /*
+        * Unregister slave by:
+        * 1) Disabling the slave address in hardware
+        * 2) Freeing priv->slave at the corresponding index
+        */
+       ret = mlxbf_i2c_slave_disable(priv, slave->addr);
+       if (ret)
+               dev_err(dev, "Unable to find slave 0x%x\n", slave->addr);
 
-       return 0;
+       return ret;
 }
 
 static u32 mlxbf_i2c_functionality(struct i2c_adapter *adap)
@@ -2166,14 +2211,27 @@ static struct mlxbf_i2c_chip_info mlxbf_i2c_chip[] = {
                        [1] = &mlxbf_i2c_corepll_res[MLXBF_I2C_CHIP_TYPE_1],
                        [2] = &mlxbf_i2c_gpio_res[MLXBF_I2C_CHIP_TYPE_1]
                },
-               .calculate_freq = mlxbf_i2c_calculate_freq_from_tyu
+               .calculate_freq = mlxbf_i2c_calculate_freq_from_tyu,
+               .smbus_master_rs_bytes_off = MLXBF_I2C_YU_SMBUS_RS_BYTES,
+               .smbus_master_fsm_off = MLXBF_I2C_YU_SMBUS_MASTER_FSM
        },
        [MLXBF_I2C_CHIP_TYPE_2] = {
                .type = MLXBF_I2C_CHIP_TYPE_2,
                .shared_res = {
                        [0] = &mlxbf_i2c_corepll_res[MLXBF_I2C_CHIP_TYPE_2]
                },
-               .calculate_freq = mlxbf_i2c_calculate_freq_from_yu
+               .calculate_freq = mlxbf_i2c_calculate_freq_from_yu,
+               .smbus_master_rs_bytes_off = MLXBF_I2C_YU_SMBUS_RS_BYTES,
+               .smbus_master_fsm_off = MLXBF_I2C_YU_SMBUS_MASTER_FSM
+       },
+       [MLXBF_I2C_CHIP_TYPE_3] = {
+               .type = MLXBF_I2C_CHIP_TYPE_3,
+               .shared_res = {
+                       [0] = &mlxbf_i2c_corepll_res[MLXBF_I2C_CHIP_TYPE_3]
+               },
+               .calculate_freq = mlxbf_i2c_calculate_freq_from_yu,
+               .smbus_master_rs_bytes_off = MLXBF_I2C_RSH_YU_SMBUS_RS_BYTES,
+               .smbus_master_fsm_off = MLXBF_I2C_RSH_YU_SMBUS_MASTER_FSM
        }
 };
 
@@ -2189,24 +2247,11 @@ static struct i2c_adapter_quirks mlxbf_i2c_quirks = {
        .max_write_len = MLXBF_I2C_MASTER_DATA_W_LENGTH,
 };
 
-static const struct of_device_id mlxbf_i2c_dt_ids[] = {
-       {
-               .compatible = "mellanox,i2c-mlxbf1",
-               .data = &mlxbf_i2c_chip[MLXBF_I2C_CHIP_TYPE_1]
-       },
-       {
-               .compatible = "mellanox,i2c-mlxbf2",
-               .data = &mlxbf_i2c_chip[MLXBF_I2C_CHIP_TYPE_2]
-       },
-       {},
-};
-
-MODULE_DEVICE_TABLE(of, mlxbf_i2c_dt_ids);
-
 #ifdef CONFIG_ACPI
 static const struct acpi_device_id mlxbf_i2c_acpi_ids[] = {
        { "MLNXBF03", (kernel_ulong_t)&mlxbf_i2c_chip[MLXBF_I2C_CHIP_TYPE_1] },
        { "MLNXBF23", (kernel_ulong_t)&mlxbf_i2c_chip[MLXBF_I2C_CHIP_TYPE_2] },
+       { "MLNXBF31", (kernel_ulong_t)&mlxbf_i2c_chip[MLXBF_I2C_CHIP_TYPE_3] },
        {},
 };
 
@@ -2215,35 +2260,27 @@ MODULE_DEVICE_TABLE(acpi, mlxbf_i2c_acpi_ids);
 static int mlxbf_i2c_acpi_probe(struct device *dev, struct mlxbf_i2c_priv *priv)
 {
        const struct acpi_device_id *aid;
-       struct acpi_device *adev;
-       unsigned long bus_id = 0;
-       const char *uid;
+       u64 bus_id;
        int ret;
 
        if (acpi_disabled)
                return -ENOENT;
 
-       adev = ACPI_COMPANION(dev);
-       if (!adev)
-               return -ENXIO;
-
        aid = acpi_match_device(mlxbf_i2c_acpi_ids, dev);
        if (!aid)
                return -ENODEV;
 
        priv->chip = (struct mlxbf_i2c_chip_info *)aid->driver_data;
 
-       uid = acpi_device_uid(adev);
-       if (!uid || !(*uid)) {
+       ret = acpi_dev_uid_to_integer(ACPI_COMPANION(dev), &bus_id);
+       if (ret) {
                dev_err(dev, "Cannot retrieve UID\n");
-               return -ENODEV;
+               return ret;
        }
 
-       ret = kstrtoul(uid, 0, &bus_id);
-       if (!ret)
-               priv->bus = bus_id;
+       priv->bus = bus_id;
 
-       return ret;
+       return 0;
 }
 #else
 static int mlxbf_i2c_acpi_probe(struct device *dev, struct mlxbf_i2c_priv *priv)
@@ -2252,36 +2289,12 @@ static int mlxbf_i2c_acpi_probe(struct device *dev, struct mlxbf_i2c_priv *priv)
 }
 #endif /* CONFIG_ACPI */
 
-static int mlxbf_i2c_of_probe(struct device *dev, struct mlxbf_i2c_priv *priv)
-{
-       const struct of_device_id *oid;
-       int bus_id = -1;
-
-       if (IS_ENABLED(CONFIG_OF) && dev->of_node) {
-               oid = of_match_node(mlxbf_i2c_dt_ids, dev->of_node);
-               if (!oid)
-                       return -ENODEV;
-
-               priv->chip = oid->data;
-
-               bus_id = of_alias_get_id(dev->of_node, "i2c");
-               if (bus_id >= 0)
-                       priv->bus = bus_id;
-       }
-
-       if (bus_id < 0) {
-               dev_err(dev, "Cannot get bus id");
-               return bus_id;
-       }
-
-       return 0;
-}
-
 static int mlxbf_i2c_probe(struct platform_device *pdev)
 {
        struct device *dev = &pdev->dev;
        struct mlxbf_i2c_priv *priv;
        struct i2c_adapter *adap;
+       u32 resource_version;
        int irq, ret;
 
        priv = devm_kzalloc(dev, sizeof(struct mlxbf_i2c_priv), GFP_KERNEL);
@@ -2289,17 +2302,63 @@ static int mlxbf_i2c_probe(struct platform_device *pdev)
                return -ENOMEM;
 
        ret = mlxbf_i2c_acpi_probe(dev, priv);
-       if (ret < 0 && ret != -ENOENT && ret != -ENXIO)
-               ret = mlxbf_i2c_of_probe(dev, priv);
-
        if (ret < 0)
                return ret;
 
-       ret = mlxbf_i2c_init_resource(pdev, &priv->smbus,
-                                     MLXBF_I2C_SMBUS_RES);
-       if (ret < 0) {
-               dev_err(dev, "Cannot fetch smbus resource info");
-               return ret;
+       /* This property allows the driver to stay backward compatible with older
+        * ACPI tables.
+        * Starting BlueField-3 SoC, the "smbus" resource was broken down into 3
+        * separate resources "timer", "master" and "slave".
+        */
+       if (device_property_read_u32(dev, "resource_version", &resource_version))
+               resource_version = 0;
+
+       priv->resource_version = resource_version;
+
+       if (priv->chip->type < MLXBF_I2C_CHIP_TYPE_3 && resource_version == 0) {
+               priv->timer = devm_kzalloc(dev, sizeof(struct mlxbf_i2c_resource), GFP_KERNEL);
+               if (!priv->timer)
+                       return -ENOMEM;
+
+               priv->mst = devm_kzalloc(dev, sizeof(struct mlxbf_i2c_resource), GFP_KERNEL);
+               if (!priv->mst)
+                       return -ENOMEM;
+
+               priv->slv = devm_kzalloc(dev, sizeof(struct mlxbf_i2c_resource), GFP_KERNEL);
+               if (!priv->slv)
+                       return -ENOMEM;
+
+               ret = mlxbf_i2c_init_resource(pdev, &priv->smbus,
+                                             MLXBF_I2C_SMBUS_RES);
+               if (ret < 0) {
+                       dev_err(dev, "Cannot fetch smbus resource info");
+                       return ret;
+               }
+
+               priv->timer->io = priv->smbus->io;
+               priv->mst->io = priv->smbus->io + MLXBF_I2C_MST_ADDR_OFFSET;
+               priv->slv->io = priv->smbus->io + MLXBF_I2C_SLV_ADDR_OFFSET;
+       } else {
+               ret = mlxbf_i2c_init_resource(pdev, &priv->timer,
+                                             MLXBF_I2C_SMBUS_TIMER_RES);
+               if (ret < 0) {
+                       dev_err(dev, "Cannot fetch timer resource info");
+                       return ret;
+               }
+
+               ret = mlxbf_i2c_init_resource(pdev, &priv->mst,
+                                             MLXBF_I2C_SMBUS_MST_RES);
+               if (ret < 0) {
+                       dev_err(dev, "Cannot fetch master resource info");
+                       return ret;
+               }
+
+               ret = mlxbf_i2c_init_resource(pdev, &priv->slv,
+                                             MLXBF_I2C_SMBUS_SLV_RES);
+               if (ret < 0) {
+                       dev_err(dev, "Cannot fetch slave resource info");
+                       return ret;
+               }
        }
 
        ret = mlxbf_i2c_init_resource(pdev, &priv->mst_cause,
@@ -2358,8 +2417,8 @@ static int mlxbf_i2c_probe(struct platform_device *pdev)
        irq = platform_get_irq(pdev, 0);
        if (irq < 0)
                return irq;
-       ret = devm_request_irq(dev, irq, mlxbf_smbus_irq,
-                              IRQF_ONESHOT | IRQF_SHARED | IRQF_PROBE_SHARED,
+       ret = devm_request_irq(dev, irq, mlxbf_i2c_irq,
+                              IRQF_SHARED | IRQF_PROBE_SHARED,
                               dev_name(dev), priv);
        if (ret < 0) {
                dev_err(dev, "Cannot get irq %d\n", irq);
@@ -2387,8 +2446,19 @@ static int mlxbf_i2c_remove(struct platform_device *pdev)
        struct device *dev = &pdev->dev;
        struct resource *params;
 
-       params = priv->smbus->params;
-       devm_release_mem_region(dev, params->start, resource_size(params));
+       if (priv->chip->type < MLXBF_I2C_CHIP_TYPE_3 && priv->resource_version == 0) {
+               params = priv->smbus->params;
+               devm_release_mem_region(dev, params->start, resource_size(params));
+       } else {
+               params = priv->timer->params;
+               devm_release_mem_region(dev, params->start, resource_size(params));
+
+               params = priv->mst->params;
+               devm_release_mem_region(dev, params->start, resource_size(params));
+
+               params = priv->slv->params;
+               devm_release_mem_region(dev, params->start, resource_size(params));
+       }
 
        params = priv->mst_cause->params;
        devm_release_mem_region(dev, params->start, resource_size(params));
@@ -2420,7 +2490,6 @@ static struct platform_driver mlxbf_i2c_driver = {
        .remove = mlxbf_i2c_remove,
        .driver = {
                .name = "i2c-mlxbf",
-               .of_match_table = mlxbf_i2c_dt_ids,
 #ifdef CONFIG_ACPI
                .acpi_match_table = ACPI_PTR(mlxbf_i2c_acpi_ids),
 #endif /* CONFIG_ACPI  */
@@ -2453,4 +2522,5 @@ module_exit(mlxbf_i2c_exit);
 
 MODULE_DESCRIPTION("Mellanox BlueField I2C bus driver");
 MODULE_AUTHOR("Khalil Blaiech <kblaiech@nvidia.com>");
+MODULE_AUTHOR("Asmaa Mnebhi <asmaa@nvidia.com>");
 MODULE_LICENSE("GPL v2");