#include <linux/slab.h>
 
 #define I2C_RS_TRANSFER                        (1 << 4)
+#define I2C_ARB_LOST                   (1 << 3)
 #define I2C_HS_NACKERR                 (1 << 2)
 #define I2C_ACKERR                     (1 << 1)
 #define I2C_TRANSAC_COMP               (1 << 0)
        struct clk *clk_main;           /* main clock for i2c bus */
        struct clk *clk_dma;            /* DMA clock for i2c via DMA */
        struct clk *clk_pmic;           /* PMIC clock for i2c from PMIC */
+       struct clk *clk_arb;            /* Arbitrator clock for i2c */
        bool have_pmic;                 /* can use i2c pins from PMIC */
        bool use_push_pull;             /* IO config push-pull mode */
 
                if (ret)
                        goto err_pmic;
        }
+
+       if (i2c->clk_arb) {
+               ret = clk_prepare_enable(i2c->clk_arb);
+               if (ret)
+                       goto err_arb;
+       }
+
        return 0;
 
+err_arb:
+       if (i2c->have_pmic)
+               clk_disable_unprepare(i2c->clk_pmic);
 err_pmic:
        clk_disable_unprepare(i2c->clk_main);
 err_main:
 
 static void mtk_i2c_clock_disable(struct mtk_i2c *i2c)
 {
+       if (i2c->clk_arb)
+               clk_disable_unprepare(i2c->clk_arb);
+
        if (i2c->have_pmic)
                clk_disable_unprepare(i2c->clk_pmic);
 
 
        /* Clear interrupt status */
        mtk_i2c_writew(i2c, restart_flag | I2C_HS_NACKERR | I2C_ACKERR |
-              I2C_TRANSAC_COMP, OFFSET_INTR_STAT);
+                           I2C_ARB_LOST | I2C_TRANSAC_COMP, OFFSET_INTR_STAT);
 
        mtk_i2c_writew(i2c, I2C_FIFO_ADDR_CLR, OFFSET_FIFO_ADDR_CLR);
 
        /* Enable interrupt */
        mtk_i2c_writew(i2c, restart_flag | I2C_HS_NACKERR | I2C_ACKERR |
-              I2C_TRANSAC_COMP, OFFSET_INTR_MASK);
+                           I2C_ARB_LOST | I2C_TRANSAC_COMP, OFFSET_INTR_MASK);
 
        /* Set transfer and transaction len */
        if (i2c->op == I2C_MASTER_WRRD) {
 
        /* Clear interrupt mask */
        mtk_i2c_writew(i2c, ~(restart_flag | I2C_HS_NACKERR | I2C_ACKERR |
-              I2C_TRANSAC_COMP), OFFSET_INTR_MASK);
+                           I2C_ARB_LOST | I2C_TRANSAC_COMP), OFFSET_INTR_MASK);
 
        if (i2c->op == I2C_MASTER_WR) {
                dma_unmap_single(i2c->dev, wpaddr,
                return PTR_ERR(i2c->clk_dma);
        }
 
+       i2c->clk_arb = devm_clk_get(&pdev->dev, "arb");
+       if (IS_ERR(i2c->clk_arb))
+               i2c->clk_arb = NULL;
+
        clk = i2c->clk_main;
        if (i2c->have_pmic) {
                i2c->clk_pmic = devm_clk_get(&pdev->dev, "pmic");